chore: import segregation

scripts/export_data_graph.py

@@ -1,6 +1,7 @@
-from neo4j import GraphDatabase
 import json
 
+from neo4j import GraphDatabase
+
 # Replace these credentials with your Neo4j database info
 uri = "bolt://localhost:7687"
 username = "neo4j"

unoplat-code-confluence/.isort.cfg

@@ -0,0 +1,7 @@
+[settings]
+known_third_party = marko,pydantic,ruff,loguru,pygithub,pypdf,pydantic_settings,litellm,pytest,dspy_ai,packaging,progiter,sentence_transformers,einops,rich,neo4j,neomodel,requirements_parser,tomlkit,stdlib_list,pytest_cov,gitpython
+import_heading_stdlib = Standard Library
+import_heading_thirdparty = Third Party
+import_heading_firstparty = First Party
+import_heading_localfolder = Local 
+py_version = 311  # For Python 3.12

unoplat-code-confluence/unoplat_code_confluence/__main__.py

@@ -1,26 +1,41 @@
+# Standard Library
 import argparse
 import asyncio
-import os
-from loguru import logger
 import datetime
-from unoplat_code_confluence.codebaseparser.arc_guard_handler import ArchGuardHandler
+import os
 import re
+import warnings
+
+# Third Party
+from loguru import logger
+from packaging import version
+
+# First Party
+from unoplat_code_confluence.codebaseparser.arc_guard_handler import \
+    ArchGuardHandler
 from unoplat_code_confluence.configuration.external_config import AppConfig
-from unoplat_code_confluence.data_models.chapi_unoplat_codebase import UnoplatCodebase
-from unoplat_code_confluence.data_models.dspy.dspy_unoplat_codebase_summary import DspyUnoplatCodebaseSummary
-from unoplat_code_confluence.database.graph.unoplat_graph_processing import UnoplatGraphProcessing
+from unoplat_code_confluence.data_models.chapi_unoplat_codebase import \
+    UnoplatCodebase
+from unoplat_code_confluence.data_models.dspy.dspy_unoplat_codebase_summary import \
+    DspyUnoplatCodebaseSummary
+from unoplat_code_confluence.database.graph.unoplat_graph_processing import \
+    UnoplatGraphProcessing
 from unoplat_code_confluence.downloader.downloader import Downloader
-from unoplat_code_confluence.dspy_class_summary import CodeConfluenceClassModule
-from unoplat_code_confluence.dspy_codebase_summary import CodeConfluenceCodebaseModule
-from unoplat_code_confluence.dspy_function_summary import CodeConfluenceFunctionModule
-from unoplat_code_confluence.dspy_package_summary import CodeConfluencePackageModule
+from unoplat_code_confluence.dspy_class_summary import \
+    CodeConfluenceClassModule
+from unoplat_code_confluence.dspy_codebase_summary import \
+    CodeConfluenceCodebaseModule
+from unoplat_code_confluence.dspy_function_summary import \
+    CodeConfluenceFunctionModule
+from unoplat_code_confluence.dspy_package_summary import \
+    CodeConfluencePackageModule
 from unoplat_code_confluence.loader import iload_json, iparse_json
 from unoplat_code_confluence.loader.json_loader import JsonLoader
 from unoplat_code_confluence.loader.parse_json import JsonParser
-from unoplat_code_confluence.markdownparser.markdownsummariser import MarkdownSummariser
-from unoplat_code_confluence.summary_parser.codebase_summary import CodebaseSummaryParser
-import warnings
-from packaging import version
+from unoplat_code_confluence.markdownparser.markdownsummariser import \
+    MarkdownSummariser
+from unoplat_code_confluence.summary_parser.codebase_summary import \
+    CodebaseSummaryParser
 
 
 async def start_pipeline():

unoplat-code-confluence/unoplat_code_confluence/codebaseparser/arc_guard_handler.py

@@ -1,8 +1,14 @@
+# Standard Library
 import os
 import subprocess
-from unoplat_code_confluence.utility.total_file_count import TotalFileCount
+
+# Third Party
 from loguru import logger
 
+# First Party
+from unoplat_code_confluence.utility.total_file_count import TotalFileCount
+
+
 class ArchGuardHandler:
     def __init__(self, jar_path, language, codebase_path,codebase_name, output_path,extension:str):
         self.jar_path = jar_path

unoplat-code-confluence/unoplat_code_confluence/configuration/external_config.py

@@ -1,13 +1,11 @@
+# Standard Library
 from enum import Enum
 from typing import Any, Dict, List
-from pydantic import BaseModel, ValidationInfo, field_validator, Field
 
+# Third Party
+from pydantic import BaseModel, Field, ValidationInfo, field_validator
 
 
-
-
-
-
 class ProgrammingLanguage(Enum):
     PYTHON = 'python'
     JAVA = 'java'

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_annotation.py

@@ -1,8 +1,13 @@
-from unoplat_code_confluence.data_models.chapi_unoplat_annotation_key_val import ChapiUnoplatAnnotationKeyVal
-from unoplat_code_confluence.data_models.chapi_unoplat_position import Position
-from pydantic import  BaseModel, Field
+# Standard Library
 from typing import Optional
 
+# Third Party
+from pydantic import BaseModel, Field
+
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_annotation_key_val import \
+    ChapiUnoplatAnnotationKeyVal
+from unoplat_code_confluence.data_models.chapi_unoplat_position import Position
 
 
 class Annotation(BaseModel):

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_annotation_key_val.py

@@ -1,8 +1,10 @@
 
 
 
+# Standard Library
 from typing import Optional
 
+# Third Party
 from pydantic import BaseModel, Field
 
 

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_class_fieldmodel.py

@@ -1,8 +1,12 @@
-from pydantic import BaseModel, Field
+# Standard Library
 from typing import List, Optional
 
-from unoplat_code_confluence.data_models.chapi_unoplat_annotation import Annotation
+# Third Party
+from pydantic import BaseModel, Field
 
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_annotation import \
+    Annotation
 
 
 class ClassFieldModel(BaseModel):

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_codebase.py

@@ -1,8 +1,12 @@
+# Standard Library
 from typing import List, Optional
 
+# Third Party
 from pydantic import BaseModel, Field
 
-from unoplat_code_confluence.data_models.chapi_unoplat_package import UnoplatPackage
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_package import \
+    UnoplatPackage
 
 
 class UnoplatCodebase(BaseModel):

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_function.py

@@ -1,10 +1,18 @@
+# Standard Library
+from typing import List, Optional
+
+# Third Party
 from pydantic import BaseModel, Field
-from typing import Optional, List
 
-from unoplat_code_confluence.data_models.chapi_unoplat_functioncall import FunctionCall
-from unoplat_code_confluence.data_models.unoplat_function_field_model import UnoplatFunctionFieldModel
-from unoplat_code_confluence.data_models.chapi_unoplat_annotation import Annotation
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_annotation import \
+    Annotation
+from unoplat_code_confluence.data_models.chapi_unoplat_functioncall import \
+    FunctionCall
 from unoplat_code_confluence.data_models.chapi_unoplat_position import Position
+from unoplat_code_confluence.data_models.unoplat_function_field_model import \
+    UnoplatFunctionFieldModel
+
 
 class ChapiUnoplatFunction(BaseModel):
     name: Optional[str] = Field(default=None, alias="Name")

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_functioncall.py

@@ -1,12 +1,17 @@
 
 
 
+# Standard Library
 from typing import List, Optional
 
-from unoplat_code_confluence.data_models.chapi_unoplat_position import Position
-from unoplat_code_confluence.data_models.chapi_unoplat_parameter import Parameter
+# Third Party
 from pydantic import BaseModel, Field
 
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_parameter import \
+    Parameter
+from unoplat_code_confluence.data_models.chapi_unoplat_position import Position
+
 
 class FunctionCall(BaseModel):
     """

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_import.py

@@ -1,6 +1,8 @@
-from pydantic import BaseModel, Field
-from typing import Optional, List
+# Standard Library
+from typing import List, Optional
 
+# Third Party
+from pydantic import BaseModel, Field
 
 
 class Import(BaseModel):

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_node.py

@@ -1,13 +1,20 @@
+# Standard Library
+from typing import List, Optional
+
+# Third Party
 from pydantic import BaseModel, Field
-from typing import Optional, List
-from unoplat_code_confluence.data_models.chapi_unoplat_annotation import Annotation
-from unoplat_code_confluence.data_models.chapi_unoplat_class_fieldmodel import ClassFieldModel
+
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_annotation import \
+    Annotation
+from unoplat_code_confluence.data_models.chapi_unoplat_class_fieldmodel import \
+    ClassFieldModel
+from unoplat_code_confluence.data_models.chapi_unoplat_function import \
+    ChapiUnoplatFunction
 from unoplat_code_confluence.data_models.chapi_unoplat_import import Import
-from unoplat_code_confluence.data_models.chapi_unoplat_function import ChapiUnoplatFunction
 from unoplat_code_confluence.data_models.chapi_unoplat_position import Position
 
 
-
 class ChapiUnoplatNode(BaseModel):
     node_name: Optional[str] = Field(default=None, alias="NodeName")
     type: Optional[str] = Field(default=None, alias="Type")

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_package.py

@@ -1,7 +1,12 @@
+# Standard Library
 from typing import Dict, List, Optional
+
+# Third Party
 from pydantic import BaseModel, Field
 
-from unoplat_code_confluence.data_models.chapi_unoplat_node import ChapiUnoplatNode
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_node import \
+    ChapiUnoplatNode
 
 
 class UnoplatPackage(BaseModel):

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_parameter.py

@@ -1,4 +1,7 @@
+# Standard Library
 from typing import Optional
+
+# Third Party
 from pydantic import BaseModel, Field
 
 

unoplat-code-confluence/unoplat_code_confluence/data_models/chapi_unoplat_position.py

@@ -1,5 +1,8 @@
 
+# Standard Library
 from typing import Optional
+
+# Third Party
 from pydantic import BaseModel, Field
 
 

unoplat-code-confluence/unoplat_code_confluence/data_models/dspy/dspy_unoplat_codebase_summary.py

@@ -1,7 +1,13 @@
+# Standard Library
 from typing import Dict, Optional
-from pydantic import BaseModel,Field
 
-from unoplat_code_confluence.data_models.dspy.dspy_unoplat_package_summary import DspyUnoplatPackageSummary
+# Third Party
+from pydantic import BaseModel, Field
+
+# First Party
+from unoplat_code_confluence.data_models.dspy.dspy_unoplat_package_summary import \
+    DspyUnoplatPackageSummary
+
 
 class DspyUnoplatCodebaseSummary(BaseModel):
     codebase_summary: Optional[str] = Field(default=None, description="A summary of the codebase")

unoplat-code-confluence/unoplat_code_confluence/data_models/dspy/dspy_unoplat_function_summary.py

@@ -1,6 +1,13 @@
-from typing import Optional, List
-from pydantic import BaseModel,Field
-from unoplat_code_confluence.data_models.chapi_unoplat_function import ChapiUnoplatFunction
+# Standard Library
+from typing import List, Optional
+
+# Third Party
+from pydantic import BaseModel, Field
+
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_function import \
+    ChapiUnoplatFunction
+
 
 class DspyUnoplatFunctionSummary(ChapiUnoplatFunction):
     qualified_name: str = Field(required=True, alias="QualifiedName",description="The qualified name of the function that contains the entire hierarchy of the class")

unoplat-code-confluence/unoplat_code_confluence/data_models/dspy/dspy_unoplat_node_summary.py

@@ -1,9 +1,17 @@
+# Standard Library
 from typing import Any, Dict, List, Optional
-from pydantic import BaseModel,Field
 
-from unoplat_code_confluence.data_models.chapi_unoplat_class_fieldmodel import ClassFieldModel
-from unoplat_code_confluence.data_models.chapi_unoplat_node import ChapiUnoplatNode
-from unoplat_code_confluence.data_models.dspy.dspy_unoplat_function_summary import DspyUnoplatFunctionSummary
+# Third Party
+from pydantic import BaseModel, Field
+
+# First Party
+from unoplat_code_confluence.data_models.chapi_unoplat_class_fieldmodel import \
+    ClassFieldModel
+from unoplat_code_confluence.data_models.chapi_unoplat_node import \
+    ChapiUnoplatNode
+from unoplat_code_confluence.data_models.dspy.dspy_unoplat_function_summary import \
+    DspyUnoplatFunctionSummary
+
 
 class DspyUnoplatNodeSummary(ChapiUnoplatNode):
     qualified_name: str = Field(required=True, alias="QualifiedName",description="The qualified name of the class that contains the entire hierarchy of the class")

unoplat-code-confluence/unoplat_code_confluence/data_models/dspy/dspy_unoplat_package_summary.py

@@ -1,9 +1,12 @@
+# Standard Library
 from typing import Dict, List, Optional
-from pydantic import BaseModel,Field
-
-from unoplat_code_confluence.data_models.dspy.dspy_unoplat_node_summary import DspyUnoplatNodeSummary
 
+# Third Party
+from pydantic import BaseModel, Field
 
+# First Party
+from unoplat_code_confluence.data_models.dspy.dspy_unoplat_node_summary import \
+    DspyUnoplatNodeSummary
 
 
 class DspyUnoplatPackageSummary(BaseModel):

unoplat-code-confluence/unoplat_code_confluence/data_models/unoplat_function_field_model.py

@@ -1,4 +1,7 @@
+# Standard Library
 from typing import Optional
+
+# Third Party
 from pydantic import BaseModel, Field
 
 

unoplat-code-confluence/unoplat_code_confluence/database/graph/__init__.py

@@ -1,5 +1,9 @@
+# Standard Library
+from typing import Any, Dict
+
+# Third Party
 from neo4j import GraphDatabase
-from typing import Dict, Any
+
 
 class Neo4jConnector:
     def __init__(self, uri: str, username: str, password: str):

unoplat-code-confluence/unoplat_code_confluence/database/graph/unoplat_graph_ingestion.py

@@ -1,7 +1,13 @@
+# Standard Library
+from typing import Any, Dict
+
+# Third Party
 from neo4j import GraphDatabase
-from typing import Dict, Any
+from neomodel import config, db
+
+# First Party
 from unoplat_code_confluence.configuration.external_config import AppConfig
-from neomodel import config,db
+
 
 class UnoplatGraphIngestion:
     def __init__(self, app_config: AppConfig):