Upload pyimageserverrun.py

pyimageserverrun.py

@@ -1,7 +1,15 @@
+# import sys
+# import ctypes
+# if not ctypes.windll.shell32.IsUserAnAdmin():
+#     ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, " ".join(sys.argv), None, 1)
+#     sys.exit()
+
+
 from argparse import ArgumentParser
 from zipfile import ZipFile
 from subprocess import Popen, PIPE
 from shutil import rmtree
+import atexit
 
 argparser = ArgumentParser()
 argparser.add_argument("-i", "--image", type=str, help="app image", required=True)
@@ -11,6 +19,8 @@ argparser.add_argument("-a", "--address", type=str, default="0.0.0.0", help="add
 args = argparser.parse_args()
 folder = args.folder
 rmtree(folder, ignore_errors=True)
+atexit.register(lambda: rmtree(folder, ignore_errors=True))
+
 ZipFile(args.image).extractall(folder)
 try:
     process = Popen([folder + "\\main", "-p", str(args.port), "-a", str(args.address)], stdout=PIPE, stderr=PIPE, shell=True)
@@ -25,4 +35,4 @@ try:
 except Exception as e:
     print(f"An error occurred: {e}")
 finally:
-    rmtree(folder)
+    rmtree(folder)