Small path-related changes to scripts

seart-group · Feb 2, 2024 · a4be97f · a4be97f
1 parent 31097b2
commit a4be97f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/generate.py b/generate.py
@@ -17,12 +17,13 @@
     parser.add_argument(
         "-o",
         "--output",
-        default=f"{__location__}/TreeSitter.java",
+        default=path(__location__, "TreeSitter.java"),
         help="Output file path.",
     )
     args = parser.parse_args()
-    path = args.output
-    cmd = ["git", "submodule", "status", f"{__location__}/tree-sitter"]
+    output = args.output
+    submodule = path(__location__, "tree-sitter")
+    cmd = ["git", "submodule", "status", submodule]
     status = run(cmd, capture_output=True, text=True)
     sha, tag = match(pattern, status.stdout).groups()
     content = f"""/*
@@ -76,5 +77,5 @@
     }}
 }}
 """
-    with open(path, "w") as file:
+    with open(output, "w") as file:
         file.write(content)
diff --git a/properties.py b/properties.py
@@ -34,7 +34,7 @@ def write(self, outfile):
     parser.add_argument(
         "-o",
         "--output",
-        default=f"{__location__}/language.properties",
+        default=path(__location__, "language.properties"),
         help="Output file path.",
     )
     args = parser.parse_args()