linting

Author: codegen-bot

src/codegen/sdk/core/codebase.py

@@ -39,11 +39,11 @@
 from codegen.sdk.core.detached_symbols.code_block import CodeBlock
 from codegen.sdk.core.detached_symbols.parameter import Parameter
 from codegen.sdk.core.directory import Directory
+from codegen.sdk.core.export import Export
 from codegen.sdk.core.external_module import ExternalModule
 from codegen.sdk.core.file import File, SourceFile
 from codegen.sdk.core.function import Function
 from codegen.sdk.core.import_resolution import Import
-from codegen.sdk.core.export import Export
 from codegen.sdk.core.interface import Interface
 from codegen.sdk.core.interfaces.editable import Editable
 from codegen.sdk.core.interfaces.has_name import HasName
@@ -66,12 +66,12 @@
 from codegen.sdk.typescript.class_definition import TSClass
 from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock
 from codegen.sdk.typescript.detached_symbols.parameter import TSParameter
+from codegen.sdk.typescript.export import TSExport
 from codegen.sdk.typescript.file import TSFile
 from codegen.sdk.typescript.function import TSFunction
 from codegen.sdk.typescript.import_resolution import TSImport
-from codegen.sdk.typescript.statements.import_statement import TSImportStatement
-from codegen.sdk.typescript.export import TSExport
 from codegen.sdk.typescript.interface import TSInterface
+from codegen.sdk.typescript.statements.import_statement import TSImportStatement
 from codegen.sdk.typescript.symbol import TSSymbol
 from codegen.sdk.typescript.type_alias import TSTypeAlias
 from codegen.sdk.utils import determine_project_language
@@ -102,10 +102,6 @@
 PyDirectory = Directory[PyFile, PySymbol, PyImportStatement, PyGlobalVar, PyClass, PyFunction, PyImport]
 
 
-
-
-
-
 @apidoc
 class Codebase(Generic[TSourceFile, TDirectory, TSymbol, TClass, TFunction, TImport, TGlobalVar, TInterface, TTypeAlias, TParameter, TCodeBlock]):
     """This class provides the main entrypoint for most programs to analyzing and manipulating codebases.
@@ -275,11 +271,10 @@ def imports(self) -> list[TImport]:
             TImport can be PyImport for Python codebases or TSImport for TypeScript codebases.
         """
         return self.G.get_nodes(NodeType.IMPORT)
-        
 
     @property
     @py_noapidoc
-    def exports(self: 'TSCodebaseType') -> list[TSExport]:
+    def exports(self: "TSCodebaseType") -> list[TSExport]:
         """Returns a list of all Export nodes in the codebase.
 
         Retrieves all Export nodes from the codebase graph. These exports represent all export statements across all files in the codebase,
@@ -291,9 +286,8 @@ def exports(self: 'TSCodebaseType') -> list[TSExport]:
         Returns:
             list[TSExport]: A list of Export nodes representing all exports in the codebase.
             TExport can only be a  TSExport for TypeScript codebases.
-        
-        """
 
+        """
         if self.language == ProgrammingLanguage.PYTHON:
             raise NotImplementedError("Exports are not supported for Python codebases since Python does not have an export mechanism.")
 
@@ -1177,6 +1171,7 @@ def from_repo(cls, repo_name: str, *, tmp_dir: str | None = None, commit: str |
             logger.error(f"Failed to initialize codebase: {e}")
             raise
 
+
 # The last 2 lines of code are added to the runner. See codegen-backend/cli/generate/utils.py
 # Type Aliases
 CodebaseType = Codebase[SourceFile, Directory, Symbol, Class, Function, Import, Assignment, Interface, TypeAlias, Parameter, CodeBlock]

src/codegen/sdk/core/directory.py

@@ -10,16 +10,15 @@
     from codegen.sdk.core.class_definition import Class
     from codegen.sdk.core.file import File
     from codegen.sdk.core.function import Function
-    from codegen.sdk.core.import_resolution import ImportStatement
+    from codegen.sdk.core.import_resolution import Import, ImportStatement
     from codegen.sdk.core.symbol import Symbol
+    from codegen.sdk.typescript.class_definition import TSClass
     from codegen.sdk.typescript.export import TSExport
-    from codegen.sdk.core.import_resolution import Import
     from codegen.sdk.typescript.file import TSFile
     from codegen.sdk.typescript.function import TSFunction
     from codegen.sdk.typescript.import_resolution import TSImport
     from codegen.sdk.typescript.statements.import_statement import TSImportStatement
     from codegen.sdk.typescript.symbol import TSSymbol
-    from codegen.sdk.typescript.class_definition import TSClass
 
 import logging
 
@@ -36,6 +35,7 @@
 
 TSGlobalVar = TypeVar("TSGlobalVar", bound="Assignment")
 
+
 @apidoc
 class Directory(Generic[TFile, TSymbol, TImportStatement, TGlobalVar, TClass, TFunction, TImport]):
     """Directory representation for codebase.

tests/unit/codegen/sdk/python/codebase/test_codebase_raise_error.py

@@ -1,4 +1,5 @@
 import pytest
+
 from codegen.sdk.codebase.factory.get_session import get_codebase_session
 from codegen.sdk.enums import ProgrammingLanguage
 
@@ -14,4 +15,4 @@ def hello():
     with get_codebase_session(tmpdir=tmpdir, files={"test.py": content}, programming_language=ProgrammingLanguage.PYTHON) as codebase:
         # Verify that accessing exports raises NotImplementedError
         with pytest.raises(NotImplementedError):
-            _ = codebase.exports
+            _ = codebase.exports

tests/unit/codegen/sdk/typescript/codebase/test_codebase_exports.py

@@ -30,11 +30,7 @@ def test_codebase_reexports(tmpdir) -> None:
     export { x } from './file1';
     export { y as z } from './file1';
     """
-    with get_codebase_session(
-        tmpdir=tmpdir,
-        files={"file1.ts": content1, "file2.ts": content2},
-        programming_language=ProgrammingLanguage.TYPESCRIPT
-    ) as codebase:
+    with get_codebase_session(tmpdir=tmpdir, files={"file1.ts": content1, "file2.ts": content2}, programming_language=ProgrammingLanguage.TYPESCRIPT) as codebase:
         assert len(codebase.exports) == 4
         export_names = {exp.name for exp in codebase.exports}
         assert export_names == {"x", "y", "z"}
@@ -49,4 +45,4 @@ def test_codebase_default_exports(tmpdir) -> None:
     with get_codebase_session(tmpdir=tmpdir, files={"file.ts": content}, programming_language=ProgrammingLanguage.TYPESCRIPT) as codebase:
         assert len(codebase.exports) == 1
         export = codebase.exports[0]
-        assert export.name == "value"
+        assert export.name == "value"

tests/unit/codegen/sdk/typescript/export/test_directory_exports.py

@@ -30,11 +30,7 @@ def test_codebase_reexports(tmpdir) -> None:
     export { x } from './file1';
     export { y as z } from './file1';
     """
-    with get_codebase_session(
-        tmpdir=tmpdir,
-        files={"file1.ts": content1, "file2.ts": content2},
-        programming_language=ProgrammingLanguage.TYPESCRIPT
-    ) as codebase:
+    with get_codebase_session(tmpdir=tmpdir, files={"file1.ts": content1, "file2.ts": content2}, programming_language=ProgrammingLanguage.TYPESCRIPT) as codebase:
         assert len(codebase.exports) == 4
         export_names = {exp.name for exp in codebase.exports}
         assert export_names == {"x", "y", "z"}
@@ -49,4 +45,4 @@ def test_codebase_default_exports(tmpdir) -> None:
     with get_codebase_session(tmpdir=tmpdir, files={"file.ts": content}, programming_language=ProgrammingLanguage.TYPESCRIPT) as codebase:
         assert len(codebase.exports) == 1
         export = codebase.exports[0]
-        assert export.name == "value"
+        assert export.name == "value"

tests/unit/codegen/sdk/typescript/import_resolution/test_directory_import.py

@@ -1,5 +1,5 @@
 from codegen.sdk.codebase.factory.get_session import get_codebase_session
-from codegen.sdk.enums import ImportType, ProgrammingLanguage
+from codegen.sdk.enums import ProgrammingLanguage
 
 
 def test_directory_imports(tmpdir) -> None:
@@ -13,26 +13,20 @@ def test_directory_imports(tmpdir) -> None:
     import defaultExport from './module';
     """
     with get_codebase_session(
-        tmpdir=tmpdir,
-        files={
-            "dir1/file1.ts": content1,
-            "dir1/file2.ts": content2,
-            "dir2/file3.ts": "import { d } from '../shared';"
-        },
-        programming_language=ProgrammingLanguage.TYPESCRIPT
+        tmpdir=tmpdir, files={"dir1/file1.ts": content1, "dir1/file2.ts": content2, "dir2/file3.ts": "import { d } from '../shared';"}, programming_language=ProgrammingLanguage.TYPESCRIPT
     ) as codebase:
         dir1 = codebase.get_directory("dir1")
         dir2 = codebase.get_directory("dir2")
-        
+
         # Test dir1 imports
         assert len(dir1.imports) == 5
         dir1_import_names = {imp.name for imp in dir1.imports}
         assert dir1_import_names == {"a", "b", "IFoo", "c", "defaultExport"}
-        
+
         # Test dir2 imports
         assert len(dir2.imports) == 1
         assert dir2.imports[0].name == "d"
-        
+
         # Test get_import method
         assert dir1.get_import("a") is not None
         assert dir1.get_import("d") is None
@@ -51,19 +45,13 @@ def test_directory_nested_imports(tmpdir) -> None:
     import { c } from '../../module3';
     """
     with get_codebase_session(
-        tmpdir=tmpdir,
-        files={
-            "dir1/file1.ts": content1,
-            "dir1/subdir/file2.ts": content2,
-            "dir1/subdir/deepdir/file3.ts": content3
-        },
-        programming_language=ProgrammingLanguage.TYPESCRIPT
+        tmpdir=tmpdir, files={"dir1/file1.ts": content1, "dir1/subdir/file2.ts": content2, "dir1/subdir/deepdir/file3.ts": content3}, programming_language=ProgrammingLanguage.TYPESCRIPT
     ) as codebase:
         dir1 = codebase.get_directory("dir1")
         subdir = codebase.get_directory("dir1/subdir")
         deepdir = codebase.get_directory("dir1/subdir/deepdir")
-        
+
         # Test imports at each directory level
         assert len(dir1.imports) == 3  # Should include all nested imports
         assert len(subdir.imports) == 2  # Should include its own and deeper imports
-        assert len(deepdir.imports) == 1  # Should only include its own imports
+        assert len(deepdir.imports) == 1  # Should only include its own imports