模板: 优化

SimonKoh93 · Nov 10, 2022 · 6a660b1 · 6a660b1
1 parent db3f2dd
commit 6a660b1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/templet/produce.py b/templet/produce.py
@@ -208,7 +208,7 @@ def routine_cmd(clone: str, config: str):
         subprocess.run(cmd, shell=True)
 
 
-if __name__ == '__main__':
+def main():
     destdir = (rp1 := os.getenv('REPO_PATH').rstrip('/')) + '/' + (dd1 := os.getenv('DEPLOYDIR').rstrip('/'))
     ba1 = f'{destdir}/backups'
     he1 = f'{destdir}/headers'
@@ -257,3 +257,7 @@ def routine_cmd(clone: str, config: str):
     subprocess.run(f"sed -i 's/name: xxxxxx/name: device {serial}/' {by1}", shell=True)
     subprocess.run(f"sed -i 's/SERIAL_NU: xxxxxx/SERIAL_NU: {serial}/' {by1}", shell=True)
     subprocess.run(f"sed -i 's/DEPLOYDIR: xxxxxx/DEPLOYDIR: {dd1}/' {by1}", shell=True)
+
+
+if __name__ == '__main__':
+    main()
diff --git a/templet/refresh.py b/templet/refresh.py
@@ -7,5 +7,7 @@
     filedir = os.getenv('FILEDIR').strip().rstrip('/')
     fconfig = os.getenv('FILE').strip()
     fclone = fconfig.split('.')[0] + '.clone.sh'
-    produce.routine_cmd(f'{filedir}/{fclone}', c1 := f'{filedir}/{fconfig}')
-    produce.simplify_config(c1)
+    if not os.path.exists(fc1 := f'{filedir}/{fclone}'):
+        fc1 = f'{filedir}/1.clone.sh'
+    produce.routine_cmd(fc1, fc2 := f'{filedir}/{fconfig}')
+    produce.simplify_config(fc2)