Create update_solution.py

Author: Hunterdii

src/update_solution.py

@@ -0,0 +1,52 @@
+import requests
+import json
+import sys
+import re
+from datetime import datetime
+
+if __name__ == "__main__":
+    assert(len(sys.argv) == 4)
+    handle = sys.argv[1]
+    token = sys.argv[2]
+    readmePath = sys.argv[3]
+
+    headers = {
+        "Authorization": f"token {token}"
+    }
+
+    # Get the latest commit details
+    commit_url = f"https://api.github.com/repos/{handle}/{sys.argv[1]}/commits?sha=main"
+    response = requests.get(commit_url, headers=headers)
+
+    if response.status_code != 200:
+        print(f"Error fetching commit details: {response.text}")
+        sys.exit(1)
+
+    commit_data = response.json()[0]
+    commit_sha = commit_data['sha']
+    commit_message = commit_data['commit']['message']
+    commit_date = commit_data['commit']['committer']['date']
+    
+    # Extract the question name or solution identifier from the commit message (example: "01(Nov) Solution Name")
+    solution_identifier = commit_message.split(":")[0]  # Assuming commit message starts with the identifier
+    
+    # Generate the badge URL dynamically based on the solution
+    badge_url = f"https://img.shields.io/badge/Solution-{solution_identifier}-blue"
+    badge_link = f"[![Today's Solution]({badge_url})](https://github.com/{handle}/{sys.argv[1]}/commit/{commit_sha})"
+    
+    # Prepare the commit link
+    commit_link = f"Commit URL: https://github.com/{handle}/{sys.argv[1]}/commit/{commit_sha}"
+
+    # Update README with the new commit and badge
+    with open(readmePath, "r") as readme:
+        content = readme.read()
+
+    # Update the commit link and the badge in the README file
+    new_content = re.sub(r"(?<=<!--START_SECTION:latest-commit-->)[\s\S]*(?=<!--END_SECTION:latest-commit-->)", commit_link, content)
+    new_content = re.sub(r"(?<=<!--START_SECTION:potd-badge-->)[\s\S]*(?=<!--END_SECTION:potd-badge-->)", badge_link, new_content)
+
+    # Write the updated content back to README
+    with open(readmePath, "w") as readme:
+        readme.write(new_content)
+
+    print("Successfully updated README with the latest commit and badge.")