Fix: mypy now creates a temp file to locate and test the pkg

scripts/__init__.py

@@ -1,15 +1,8 @@
 import sys
 
 from loguru import logger
-from pathlib import Path
-
-pkg_path = [x for x in sys.path if x.endswith('site-packages')]
-
-if not Path(fr'{pkg_path[0]}/my_paths.pth').exists():
-    with open(fr'{pkg_path[0]}/my_paths.pth', 'w') as file:
-        file.write(str(Path.cwd()))
-
 
+# Config the format of log message
 config = {
     "handlers": [
         {

scripts/test/procedures.py

@@ -1,13 +1,29 @@
 import shutil
 import subprocess
 from pathlib import Path
+import sys
 
-from scripts._job import Step, run_job
+
+def create_mypy_pkg_file():
+    pkg_path = [x for x in sys.path if x.endswith('site-packages')]
+
+    if not Path(fr'{pkg_path[0]}/my_path.pth').exists():
+        with open(fr'{pkg_path[0]}/my_path.pth', 'w') as file:
+            file.write(str(Path.cwd()))
+
+
+def destroy_mypy_pkg_file():
+    pkg_path = [x for x in sys.path if x.endswith('site-packages')]
+
+    if Path(fr'{pkg_path[0]}/my_path.pth').exists():
+        Path(fr'{pkg_path[0]}/my_path.pth').unlink()
 
 
 def run_mypy_src():
+    create_mypy_pkg_file()
     cmd = ["mypy", "pandas-stubs", "tests", "--no-incremental"]
     subprocess.run(cmd, check=True)
+    destroy_mypy_pkg_file()
 
 
 def run_pyright_src():
@@ -31,24 +47,49 @@ def install_dist():
     subprocess.run(cmd, check=True)
 
 
+def add_last_changes():
+    cmd = ["git", "add", "."]
+    subprocess.run(cmd, check=True)
+
+
+def commit_last_changes():
+    cmd = ["git", "commit", "-am", "\"temp commit\""]
+    subprocess.run(cmd, check=True)
+
+
 def remove_src():
     shutil.rmtree(r"pandas-stubs")
 
 
 def run_mypy_dist():
+    create_mypy_pkg_file()
     cmd = ["mypy", "tests", "--no-incremental"]
     subprocess.run(cmd, check=True)
+    destroy_mypy_pkg_file()
 
 
 def run_pyright_dist():
     cmd = ["pyright", "tests"]
     subprocess.run(cmd, check=True)
 
+
 def uninstall_dist():
     cmd = ["pip", "uninstall", "-y", "pandas-stubs"]
     subprocess.run(cmd, check=True)
 
 
+def restore_last_changes():
+    cmd = ["git", "show", "-s", "--format=%s"]
+    process = subprocess.Popen(cmd, stdout=subprocess.PIPE)
+    last_commit_name = process.communicate()[0]
+
+    if last_commit_name == b'"temp commit"\n':
+        cmd = ["git", "reset", "--soft", "HEAD~1"]
+        subprocess.run(cmd, check=True)
+    else:
+        print("There is not temp commit to restore.")
+
+
 def restore_src():
     cmd = ["git", "checkout", "HEAD", "pandas-stubs"]
     subprocess.run(cmd, check=True)
@@ -74,3 +115,6 @@ def create_new_venv():
     cmd = ["poetry", "shell"]
     subprocess.run(cmd, check=True)
 
+
+if __name__ == '__main__':
+    restore_last_changes()