ChangesMadeSecondTime

This commit is contained in:
sarthakshishodia20
2025-01-22 22:03:10 +05:30
parent 663f4c351e
commit e0d5ac11c6

View File

@@ -1,8 +1,9 @@
from jinja2 import Environment, FileSystemLoader
import json import json
import os
import logging import logging
from jinja2 import Environment, FileSystemLoader # type: ignore
# Setting up logging for better debugging and traceability # Configuring logging
logging.basicConfig(level=logging.INFO) logging.basicConfig(level=logging.INFO)
DATAFILE = "./data.json" DATAFILE = "./data.json"
@@ -11,53 +12,64 @@ TEMPLATEFILE = "README-template.j2"
TARGETFILE = "./README.md" TARGETFILE = "./README.md"
def new_technology_dict(repo_technology): def new_technology_dict(repo_technology):
"""Creates a dictionary for new technology with an empty entries list."""
return {"link_id": repo_technology.lower(), "entries": []} return {"link_id": repo_technology.lower(), "entries": []}
def load_data(datafile): # Function to log warnings for missing data
"""Loads data from the JSON file and handles errors in case of missing file or invalid JSON.""" def log_warning(message):
try: logging.warning(message)
with open(datafile, "r") as file:
return json.loads(file.read())
except FileNotFoundError:
logging.error(f"Data file {datafile} not found.")
raise
except json.JSONDecodeError:
logging.error(f"Error decoding JSON from {datafile}. Please check the file format.")
raise
def process_technologies(data): # Check if the data file exists
"""Process technology data into a structured format.""" if not os.path.exists(DATAFILE):
log_warning(f"Data file {DATAFILE} does not exist.")
exit(1)
# Load data from the JSON file
try:
with open(DATAFILE, "r") as datafile:
data = json.loads(datafile.read())
except json.JSONDecodeError:
log_warning("Error: Failed to parse JSON data in the file.")
exit(1)
# Initialize technologies dictionary
technologies = {} technologies = {}
for technology in data["technologies"]:
# Processing technologies
for technology in data.get("technologies", {}):
technologies[technology] = { technologies[technology] = {
"link_id": data["technologies"][technology], "link_id": data["technologies"].get(technology),
"entries": [], "entries": [],
} }
return technologies
def process_repositories(data, technologies): # Processing repositories
"""Process repository data and associate technologies.""" for repository in data.get("repositories", []):
for repository in data["repositories"]: repo_technologies = repository.get("technologies", [])
repo_technologies = repository["technologies"] if not repo_technologies:
log_warning(f"Repository {repository['name']} has no technologies listed.")
for repo_technology in repo_technologies: for repo_technology in repo_technologies:
if repo_technology not in technologies: if repo_technology not in technologies:
technologies[repo_technology] = new_technology_dict(repo_technology) technologies[repo_technology] = new_technology_dict(repo_technology)
log_warning(f"Technology {repo_technology} is newly added.")
technologies[repo_technology]["entries"].append(repository) technologies[repo_technology]["entries"].append(repository)
return technologies
def organize_categories(technologies): # Create Jinja2 environment and load the template
"""Organize categories and entries for templating.""" env = Environment(loader=FileSystemLoader(TEMPLATEPATH))
if not os.path.exists(os.path.join(TEMPLATEPATH, TEMPLATEFILE)):
log_warning(f"Template file {TEMPLATEFILE} does not exist in the provided path.")
exit(1)
template = env.get_template(TEMPLATEFILE)
# Create categories from the technologies
categories = [] categories = []
for key, value in zip(technologies.keys(), technologies.values()): for key, value in technologies.items():
categories.append( categories.append(
{"title": key, "link_id": value["link_id"], "entries": value["entries"]} {"title": key, "link_id": value["link_id"], "entries": value["entries"]}
) )
# Sorting categories and entries
categories = sorted(categories, key=lambda x: x["title"].upper()) categories = sorted(categories, key=lambda x: x["title"].upper())
category_groups = {"Misc": []}
# Group categories by first character (A-Z), and handle miscellaneous ones separately category_groups = {"Misc": []}
for category in categories: for category in categories:
category["entries"] = sorted(category["entries"], key=lambda x: x["name"].upper()) category["entries"] = sorted(category["entries"], key=lambda x: x["name"].upper())
first_char = category["title"][0].upper() first_char = category["title"][0].upper()
@@ -68,43 +80,28 @@ def organize_categories(technologies):
else: else:
category_groups["Misc"].append(category) category_groups["Misc"].append(category)
return categories, category_groups # Process sponsors
sponsors = data.get("sponsors", [])
def generate_readme(template, categories, category_groups, sponsors): # Generate Table of Contents (TOC)
"""Render the README file using the Jinja2 template.""" toc = []
output = template.render(category_groups=category_groups, categories=categories, sponsors=sponsors) for category in categories:
toc.append(f"- [{category['title']}]({category['link_id']})")
# Prepare context for rendering the template
context = {
"category_groups": category_groups,
"categories": categories,
"sponsors": sponsors,
"toc": toc # Adding TOC to context
}
# Rendering the README file
try:
output = template.render(context)
with open(TARGETFILE, "w") as targetfile: with open(TARGETFILE, "w") as targetfile:
targetfile.write(output) targetfile.write(output)
logging.info(f"README generated at {TARGETFILE}") logging.info("README file generated successfully.")
def main():
"""Main function to load data, process repositories and generate the README."""
try:
# Load the data from the JSON file
data = load_data(DATAFILE)
# Process the technology data
technologies = process_technologies(data)
# Process repositories and associate them with technologies
technologies = process_repositories(data, technologies)
# Set up the Jinja2 environment and load the template
env = Environment(loader=FileSystemLoader(TEMPLATEPATH))
template = env.get_template(TEMPLATEFILE)
# Organize categories based on technologies
categories, category_groups = organize_categories(technologies)
# Extract sponsors data
sponsors = data["sponsors"]
# Generate the README file using the template
generate_readme(template, categories, category_groups, sponsors)
except Exception as e: except Exception as e:
logging.error("An error occurred while generating the README.") log_warning(f"Error while rendering template: {e}")
logging.exception(e) exit(1)
if __name__ == "__main__":
main()