review feedback

pixee · Oct 25, 2023 · 66f3c48 · 66f3c48
1 parent e9cc4d7
commit 66f3c48
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/src/codemodder/context.py b/src/codemodder/context.py
@@ -33,6 +33,7 @@ class CodemodExecutionContext:  # pylint: disable=too-many-instance-attributes
     dry_run: bool = False
     verbose: bool = False
     registry: CodemodRegistry
+    repo_manager: PythonRepoManager
 
     def __init__(
         self,

diff --git a/src/codemodder/project_analysis/file_parsers/setup_cfg_file_parser.py b/src/codemodder/project_analysis/file_parsers/setup_cfg_file_parser.py
@@ -10,7 +10,7 @@ class SetupCfgParser(BaseParser):
     def file_name(self):
         return "setup.cfg"
 
-    def _parse_dependencies_from_toml(self, config: configparser.ConfigParser):
+    def _parse_dependencies_from_cfg(self, config: configparser.ConfigParser):
         # todo: handle cases for
         # 1. no dependencies, no options dict
         # setup_requires, tests_require, extras_require
@@ -32,6 +32,6 @@ def _parse_file(self, file: Path):
         return PackageStore(
             type=self.file_name,
             file=str(file),
-            dependencies=self._parse_dependencies_from_toml(config),
+            dependencies=self._parse_dependencies_from_cfg(config),
             py_versions=self._parse_py_versions(config),
         )
diff --git a/src/codemodder/project_analysis/file_parsers/setup_py_file_parser.py b/src/codemodder/project_analysis/file_parsers/setup_py_file_parser.py
@@ -31,7 +31,7 @@ def _parse_py_versions(self, version_str):
         return [clean_simplestring(version_str)]
 
     def _parse_file(self, file: Path):
-        visitor = SetupCallVisotor()
+        visitor = SetupCallVisitor()
         with open(str(file), "r", encoding="utf-8") as f:
             # todo: handle failure in parsing
             module = cst.parse_module(f.read())
@@ -47,7 +47,7 @@ def _parse_file(self, file: Path):
         )
 
 
-class SetupCallVisotor(cst.CSTVisitor):
+class SetupCallVisitor(cst.CSTVisitor):
     def __init__(self):
         self.python_requires = None
         self.install_requires = None

diff --git a/src/codemodder/project_analysis/file_parsers/utils.py b/src/codemodder/project_analysis/file_parsers/utils.py
@@ -4,4 +4,4 @@
 def clean_simplestring(node: cst.SimpleString | str) -> str:
     if isinstance(node, str):
         return node.strip('"')
-    return node.value.strip('"')
+    return node.raw_value