simplify py code

Author: tkucar

src/codegen/sdk/python/file.py

@@ -190,43 +190,13 @@ def remove_unused_imports(self) -> None:
         - Future imports even if unused
         - Type hints and annotations
         """
-        # Track processed imports to avoid duplicates
-        processed_imports = set()
-
-        # Group imports by module for more efficient processing
-        module_imports = {}
-
-        # First pass - group imports by module
+        # Process each import statement
         for import_stmt in self.imports:
-            if import_stmt in processed_imports:
-                continue
-
             # Always preserve __future__ and star imports since we can't track their usage
-            if import_stmt.is_future_import or import_stmt.is_star_import:
-                continue
-
-            module = import_stmt.module_name
-            if module not in module_imports:
-                module_imports[module] = []
-            module_imports[module].append(import_stmt)
-
-        # Second pass - process each module's imports
-        for module, imports in module_imports.items():
-            # Skip if any import from this module is used
-            if any(imp.usages for imp in imports):
-                # Remove individual unused imports if it's a from-style import
-                if len(imports) > 1 and imports[0].is_from_import():
-                    for imp in imports:
-                        if not imp.usages and imp not in processed_imports:
-                            processed_imports.add(imp)
-                            imp.remove()
+            if import_stmt.is_future_import or import_stmt.is_wildcard_import():
                 continue
 
-            # If no imports from module are used, remove them all
-            for imp in imports:
-                if imp not in processed_imports:
-                    processed_imports.add(imp)
-                    imp.remove()
+            import_stmt.remove_if_unused()
 
         self.G.commit_transactions()
 

src/codegen/sdk/python/import_resolution.py

@@ -319,21 +319,7 @@ def is_from_import(self) -> bool:
         Returns:
             bool: True if this is a from-style import, False otherwise.
         """
-        return self.ts_node.type == "import_from_statement"
-
-    @property
-    def is_star_import(self) -> bool:
-        """Determines if this is a star import (from x import *).
-
-        Returns:
-            bool: True if this is a star import, False otherwise
-        """
-        if self.ts_node.type != "import_from_statement":
-            return False
-
-        # Look for wildcard_import node among children
-        wildcard_import = next((node for node in self.ts_node.children if node.type == "wildcard_import"), None)
-        return wildcard_import is not None
+        return self.import_type in [ImportType.NAMED_EXPORT, ImportType.WILDCARD]
 
     @property
     def is_future_import(self) -> bool:

tests/unit/codegen/sdk/python/import_resolution/test_import_properties.py

@@ -48,25 +48,6 @@ def test_is_from_import(tmpdir) -> None:
         assert imports[5].is_from_import()
 
 
-def test_is_star_import(tmpdir) -> None:
-    # language=python
-    content = """
-import module1
-from module2 import symbol
-from module3 import *
-from module4 import (a, b, c)
-"""
-    with get_codebase_session(tmpdir=tmpdir, files={"test.py": content}) as codebase:
-        file = codebase.get_file("test.py")
-        imports = file.imports
-
-        # Only star import should return True
-        assert not imports[0].is_star_import
-        assert not imports[1].is_star_import
-        assert imports[2].is_star_import
-        assert not imports[3].is_star_import
-
-
 def test_is_future_import(tmpdir) -> None:
     # language=python
     content = """