Add initial implementation of CodeItem.number_of_uses

this feature is still unfinished and under active implementation.

Author: albertas

deadcode/tests/nested_scope/test_nested_scope.py

@@ -1,3 +1,5 @@
+from unittest import skip
+
 from deadcode.cli import main
 from deadcode.tests.base import BaseTestCase
 
@@ -120,6 +122,7 @@ class Eggs(Spam):
 
         self.assertIsNone(result)
 
+    @skip("This feature is being implemented")
     def test_type_tracking_for_function_arguments(self):
         self.files = {
             "foo.py": """
@@ -160,3 +163,18 @@ def spam(eggs):
         )
 
         self.assertIsNone(result)
+
+    @skip(">> Observed types in scope definition is still being implemented")
+    def test_scope_update_for_method_call_expression(self):
+        # >>> TODO: Mark types in scope usage correctly for method invocation
+        self.files = {
+            "foo.py": """
+                Bar().spam()
+                """
+        }
+
+        result = main(["foo.py", "--no-color", "--fix"])
+
+        self.assertFiles({"foo.py": """"""})
+
+        self.assertIsNone(result)

deadcode/utils/nested_scopes.py

@@ -31,7 +31,7 @@ def add(self, code_item: CodeItem) -> None:
                 current_scope[scope_part] = {}  # Could use None if type cannot have scope
             current_scope = current_scope[scope_part]
 
-        # > TODO: leef should be replaced. Is it replaced with new code item?
+        # > TODO: leaf should be replaced. Is it replaced with new code item?
         current_scope[code_item] = {}
 
     def get(self, name: str, scope: str) -> Optional[Union[CodeItem, str]]:
@@ -60,3 +60,24 @@ def get(self, name: str, scope: str) -> Optional[Union[CodeItem, str]]:
                 return current_scope_keys[current_scope_keys.index(name)]
 
         return None
+
+    def mark_as_used(self, name: str, scope: str) -> None:
+        # >TODO: This method does not work, when methods are being invoked.
+        # The scope is global, name is method name, but the instance and
+        # class names are not being taken into account.
+
+        # Solution: parsing of a method invocation should be handled differently.
+        # More precicely. The parsing should be in the right order.
+
+        # Next step to solve this issue:
+        # > Investigate how usage statement is being handled and what can be done about it.
+        # Write tests for Class and instance creations.
+
+        # In this place I should get a list of names, which are being used in the invocation.
+
+        # if name == "spam":
+        #     breakpoint()
+
+        code_item = self.get(name, scope)
+        if isinstance(code_item, CodeItem):
+            code_item.number_of_uses += 1

deadcode/visitor/code_item.py

@@ -52,6 +52,7 @@ class CodeItem:  # TODO: This should also be a dataclass, because hash and tuple
         "name_column",
         "message",
         "error_code",
+        "number_of_uses",
     )
 
     def __init__(
@@ -70,12 +71,14 @@ def __init__(
         name_line: Optional[int] = None,
         name_column: Optional[int] = None,
         message: str = "",
+        number_of_uses: int = 0,
     ):
         self.name = name
         self.type_ = type_
         self.filename = filename
         self.scope = scope
         self.inherits_from = inherits_from
+        self.number_of_uses = number_of_uses
 
         if code_parts is None:
             self.code_parts = []