Merge pull request #13 from DSACMS/convert-to-poetry

Convert to poetry

Author: sachin-panayil

.gitignore

@@ -1,4 +1,6 @@
 #pycache
 __pycache__/
 *.pyc
-
+.venv/
+poetry.lock
+dist/

codejson_index_generator/__init__.py

@@ -0,0 +1 @@
+from .parsers import *

codejson_index_generator/parsers.py

@@ -0,0 +1,73 @@
+import json
+import base64
+import argparse
+import os
+
+from typing import Dict, Optional
+from github import Github, Repository, GithubException
+
+class IndexGenerator:
+    def __init__(self, agency: str, verison: str, token: Optional[str] = None,):
+        self.github = Github(token) if token else Github()
+
+        # user can change agency and version depending on paramters
+        self.index = {
+            "agency": agency,
+            "version": verison,
+            "measurementType": {
+                "method": "projects"
+            },
+            "releases": []
+        }
+
+    def get_code_json(self, repo: Repository) -> Optional[Dict]:
+        try:
+            content = repo.get_contents("code.json", ref = "main")
+        except GithubException as e:
+            print(f"GitHub Error: {e.data.get('message', 'No message available')}")
+            return None
+
+        try:
+            decoded_content = base64.b64decode(content.content)
+            return json.loads(decoded_content)
+        except (json.JSONDecodeError, ValueError) as e:
+            print(f"JSON Error: {str(e)}")
+            return None
+
+    def update_index(self, index: Dict, code_json: Dict, org_name: str, repo_name: str) -> None:
+        baseline = {
+            'organization': org_name,
+            'name': repo_name
+        }
+
+        baseline.update(code_json)
+    
+        index['releases'].append(baseline)
+
+    def process_organization(self, org_name: str) -> None:
+        try:
+            org = self.github.get_organization(org_name)
+            print(f"\nProcessing organization: {org_name}")
+
+            total_repos = org.public_repos
+            print(f"Found {total_repos} public repositories")
+            
+            for id, repo in enumerate(org.get_repos(type='public'), 1):
+                print(f"\nChecking {repo.name} [{id}/{total_repos}]")
+                
+                code_json = self.get_code_json(repo)
+                if code_json:
+                    print(f"✅ Found code.json in {repo.name}")
+                    self.update_index(self.index, code_json, org_name, repo.name)
+                else:
+                    print(f"❌ No code.json found in {repo.name}")
+                    
+        except GithubException as e:
+            print(f"Error processing organization {org_name}: {str(e)}")
+
+    def save_index(self, output_path: str) -> None:
+        # sorts index by organizaiton then by name
+        self.index['releases'].sort(key=lambda x: (x.get('organization', ''), x.get('name', '')))
+
+        with open(output_path, 'w') as f:
+            json.dump(self.index, f, indent=2)

main.py

@@ -3,74 +3,7 @@
 import argparse
 import os
 
-from typing import Dict, Optional
-from github import Github, Repository, GithubException
-
-class IndexGenerator:
-    def __init__(self, agency: str, verison: str, token: Optional[str] = None,):
-        self.github = Github(token) if token else Github()
-
-        # user can change agency and version depending on paramters
-        self.index = {
-            "agency": agency,
-            "version": verison,
-            "measurementType": {
-                "method": "projects"
-            },
-            "releases": []
-        }
-
-    def get_code_json(self, repo: Repository) -> Optional[Dict]:
-        try:
-            content = repo.get_contents("code.json", ref = "main")
-        except GithubException as e:
-            print(f"GitHub Error: {e.data.get('message', 'No message available')}")
-            return None
-
-        try:
-            decoded_content = base64.b64decode(content.content)
-            return json.loads(decoded_content)
-        except (json.JSONDecodeError, ValueError) as e:
-            print(f"JSON Error: {str(e)}")
-            return None
-
-    def update_index(self, index: Dict, code_json: Dict, org_name: str, repo_name: str) -> None:
-        baseline = {
-            'organization': org_name,
-            'name': repo_name
-        }
-
-        baseline.update(code_json)
-    
-        index['releases'].append(baseline)
-
-    def process_organization(self, org_name: str) -> None:
-        try:
-            org = self.github.get_organization(org_name)
-            print(f"\nProcessing organization: {org_name}")
-
-            total_repos = org.public_repos
-            print(f"Found {total_repos} public repositories")
-            
-            for id, repo in enumerate(org.get_repos(type='public'), 1):
-                print(f"\nChecking {repo.name} [{id}/{total_repos}]")
-                
-                code_json = self.get_code_json(repo)
-                if code_json:
-                    print(f"✅ Found code.json in {repo.name}")
-                    self.update_index(self.index, code_json, org_name, repo.name)
-                else:
-                    print(f"❌ No code.json found in {repo.name}")
-                    
-        except GithubException as e:
-            print(f"Error processing organization {org_name}: {str(e)}")
-
-    def save_index(self, output_path: str) -> None:
-        # sorts index by organizaiton then by name
-        self.index['releases'].sort(key=lambda x: (x.get('organization', ''), x.get('name', '')))
-
-        with open(output_path, 'w') as f:
-            json.dump(self.index, f, indent=2)
+from codejson_index_generator import IndexGenerator
 
 def main():
     parser = argparse.ArgumentParser(

pyproject.toml

@@ -0,0 +1,16 @@
+[tool.poetry]
+name = "codejson-index-generator"
+version = "1.0.0"
+description = "Script to create an indexed code.json for agencies."
+authors = ["Sachin Panyil <[email protected]>"]
+license = "CC0 1.0"
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.13"
+pygithub = ">=1.59,<2.0"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"