add python2 interface

Author: kguzek

README.md

@@ -70,7 +70,7 @@ python -m lpg --help
 
 Planned available languages:
 
-- [ ] python
+- [x] python
 - [x] python3 (thanks [@dancikmad](https://github.com/dancikmad))
 - [ ] javascript
 - [ ] typescript

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "leetcode-project-generator"
-version = "1.3.0"
+version = "1.3.1"
 authors = [
   { name = "Konrad Guzek", email = "[email protected]" },
 ]

src/lpg/interfaces/file.py

@@ -6,10 +6,12 @@
 
 from .lang.base import BaseLanguageInterface
 from .lang.c import CLanguageInterface
+from .lang.python import PythonLanguageInterface
 from .lang.python3 import Python3LanguageInterface
 
 LANGUAGE_INTERFACES: dict[str, BaseLanguageInterface] = {
     "c": CLanguageInterface(),
+    "python": PythonLanguageInterface(),
     "python3": Python3LanguageInterface(),
 }
 

src/lpg/interfaces/lang/python.py

@@ -0,0 +1,56 @@
+"""Project generator for the Python 2 language."""
+
+import re
+
+from .base import BaseLanguageInterface
+
+FUNCTION_SIGNATURE_PATTERN = re.compile(
+    r"^\s*def (?P<name>\w+)\((?P<params>[^)]*)\):$",
+    flags=re.MULTILINE,
+)
+
+TEST_FILE_TEMPLATE = """\
+from solution import Solution
+
+
+{supplemental_code}if __name__ == "__main__":
+    {params_setup}
+    result = Solution().{name}({params_call})
+    print("{OUTPUT_RESULT_PREFIX}", result)
+"""
+
+
+class PythonLanguageInterface(BaseLanguageInterface):
+    """Implementation of the Python 2 language project template interface."""
+
+    function_signature_pattern = FUNCTION_SIGNATURE_PATTERN
+    compile_command = None
+    test_command = ["python", "test.py"]
+    default_output = "None"
+
+    def prepare_project_files(self, template: str):
+        params = (
+            [
+                param
+                for param in self.groups["params"].split(", ")
+                if param and param != "self"
+            ]
+            if self.groups["params"]
+            else []
+        )
+        self.groups["params_setup"] = "\n    ".join(
+            param if "=" in param else f"{param} = None" for param in params
+        )
+        self.groups["params_call"] = ", ".join(
+            param.split("=")[0].split(":")[0].strip() for param in params
+        )
+
+        supplemental_code = self.get_supplemental_code(template)
+        supplemental_code = (
+            "" if supplemental_code is None else f"{supplemental_code}\n\n\n"
+        )
+        self.groups["supplemental_code"] = supplemental_code
+        return {
+            "solution.py": f"{supplemental_code}{template}pass\n",
+            "test.py": TEST_FILE_TEMPLATE.format(**self.groups),
+        }

src/lpg/interfaces/lang/python3.py

@@ -2,58 +2,26 @@
 
 import re
 
-from .base import BaseLanguageInterface
+from .python import PythonLanguageInterface
 
 FUNCTION_SIGNATURE_PATTERN = re.compile(
-    r"^(class Solution:\n)?\s*def (?P<name>\w+)\((?P<params>[^)]*)\) -> (?P<returnType>[^:]+):$",
+    r"^\s*def (?P<name>\w+)\((?P<params>[^)]*)\) -> (?P<returnType>[^:]+):$",
     flags=re.MULTILINE,
 )
 
+
 # LeetCode uses Typing types, not Python 3.9+ types
 TYPING_IMPORT_TEMPLATE = "from typing import *\n\n"
 
-TEST_FILE_TEMPLATE = """\
-from solution import Solution
-
-
-{supplemental_code}if __name__ == "__main__":
-    {params_setup}
-    result = Solution().{name}({params_call})
-    print("{OUTPUT_RESULT_PREFIX}", result)
-"""
 
-
-class Python3LanguageInterface(BaseLanguageInterface):
+class Python3LanguageInterface(PythonLanguageInterface):
     """Implementation of the Python 3 language project template interface."""
 
     function_signature_pattern = FUNCTION_SIGNATURE_PATTERN
-    compile_command = None
-    test_command = ["python", "test.py"]
-    default_output = "None"
-
-    def prepare_project_files(self, template: str):
-        params = (
-            [
-                param
-                for param in self.groups["params"].split(", ")
-                if param and param != "self"
-            ]
-            if self.groups["params"]
-            else []
-        )
-        self.groups["params_setup"] = "\n    ".join(
-            param if "=" in param else f"{param} = None" for param in params
-        )
-        self.groups["params_call"] = ", ".join(
-            param.split("=")[0].split(":")[0].strip() for param in params
-        )
-
-        supplemental_code = self.get_supplemental_code(template)
-        supplemental_code = (
-            "" if supplemental_code is None else f"{supplemental_code}\n\n\n"
-        )
-        self.groups["supplemental_code"] = supplemental_code
+
+    def prepare_project_files(self, template):
+        project_files = super().prepare_project_files(template)
         return {
-            "solution.py": f"{TYPING_IMPORT_TEMPLATE}\n{supplemental_code}{template}pass\n",
-            "test.py": f"{TYPING_IMPORT_TEMPLATE}{TEST_FILE_TEMPLATE.format(**self.groups)}",
+            "solution.py": f"{TYPING_IMPORT_TEMPLATE}\n{project_files['solution.py']}",
+            "test.py": f"{TYPING_IMPORT_TEMPLATE}{project_files['test.py']}",
         }