CG-10899: file.add_import (#673)

`file.add_import_from_string` and `file.add_symbol_import` has now been
consolidated into one method `file.import` which takes in `Symbol | str`
either the string import or the Symbol that needs to be imported and
will handle the logic

---------

Author: tawsifkamal

codegen-examples/examples/dict_to_schema/run.py

@@ -84,7 +84,7 @@ def run(codebase: Codebase):
 
         # Add imports if needed
         if needs_imports:
-            file.add_import_from_import_string("from pydantic import BaseModel")
+            file.add_import("from pydantic import BaseModel")
 
         if file_modified:
             files_modified += 1

codegen-examples/examples/flask_to_fastapi_migration/run.py

@@ -57,7 +57,7 @@ def setup_static_files(file):
     print(f"📁 Processing file: {file.filepath}")
 
     # Add import for StaticFiles
-    file.add_import_from_import_string("from fastapi.staticfiles import StaticFiles")
+    file.add_import("from fastapi.staticfiles import StaticFiles")
     print("✅ Added import: from fastapi.staticfiles import StaticFiles")
 
     # Add app.mount for static file handling

codegen-examples/examples/sqlalchemy_soft_delete/README.md

@@ -58,7 +58,7 @@ The codemod processes your codebase in several steps:
    ```python
    def ensure_and_import(file):
        if not any("and_" in imp.name for imp in file.imports):
-           file.add_import_from_import_string("from sqlalchemy import and_")
+           file.add_import("from sqlalchemy import and_")
    ```
 
    - Automatically adds required SQLAlchemy imports (`and_`)

codegen-examples/examples/sqlalchemy_soft_delete/run.py

@@ -51,7 +51,7 @@ def ensure_and_import(file):
     """Ensure the file has the necessary and_ import."""
     if not any("and_" in imp.name for imp in file.imports):
         print(f"File {file.filepath} does not import and_. Adding import.")
-        file.add_import_from_import_string("from sqlalchemy import and_")
+        file.add_import("from sqlalchemy import and_")
 
 
 def clone_repo(repo_url: str, repo_path: Path) -> None:

codegen-examples/examples/sqlalchemy_type_annotations/run.py

@@ -100,16 +100,16 @@ def run(codebase: Codebase):
 
             # Add necessary imports
             if not cls.file.has_import("Mapped"):
-                cls.file.add_import_from_import_string("from sqlalchemy.orm import Mapped\n")
+                cls.file.add_import("from sqlalchemy.orm import Mapped\n")
 
             if "Optional" in new_type and not cls.file.has_import("Optional"):
-                cls.file.add_import_from_import_string("from typing import Optional\n")
+                cls.file.add_import("from typing import Optional\n")
 
             if "Decimal" in new_type and not cls.file.has_import("Decimal"):
-                cls.file.add_import_from_import_string("from decimal import Decimal\n")
+                cls.file.add_import("from decimal import Decimal\n")
 
             if "datetime" in new_type and not cls.file.has_import("datetime"):
-                cls.file.add_import_from_import_string("from datetime import datetime\n")
+                cls.file.add_import("from datetime import datetime\n")
 
         if class_modified:
             classes_modified += 1

codegen-examples/examples/unittest_to_pytest/run.py

@@ -24,7 +24,7 @@ def convert_to_pytest_fixtures(file):
     print(f"🔍 Processing file: {file.filepath}")
 
     if not any(imp.name == "pytest" for imp in file.imports):
-        file.add_import_from_import_string("import pytest")
+        file.add_import("import pytest")
         print(f"➕ Added pytest import to {file.filepath}")
 
     for cls in file.classes:

codegen-examples/examples/usesuspensequery_to_usesuspensequeries/README.md

@@ -25,7 +25,7 @@ The script automates the entire migration process in a few key steps:
 
    ```python
    import_str = "import { useQuery, useSuspenseQueries } from '@tanstack/react-query'"
-   file.add_import_from_import_string(import_str)
+   file.add_import(import_str)
    ```
 
    - Uses Codegen's import analysis to add required imports

codegen-examples/examples/usesuspensequery_to_usesuspensequeries/run.py

@@ -26,7 +26,7 @@ def run(codebase: Codebase):
 
         print(f"Processing {file.filepath}")
         # Add the import statement
-        file.add_import_from_import_string(import_str)
+        file.add_import(import_str)
         file_modified = False
 
         # Iterate through all functions in the file

docs/building-with-codegen/imports.mdx

@@ -120,7 +120,7 @@ for module, imports in module_imports.items():
     if len(imports) > 1:
         # Create combined import
         symbols = [imp.name for imp in imports]
-        file.add_import_from_import_string(
+        file.add_import(
             f"import {{ {', '.join(symbols)} }} from '{module}'"
         )
         # Remove old imports

docs/building-with-codegen/react-and-jsx.mdx

@@ -136,5 +136,5 @@ for function in codebase.functions:
 
                 # Add import if needed
                 if not file.has_import("NewComponent"):
-                    file.add_symbol_import(new_component)
+                    file.add_import(new_component)
 ```