added updating submodules to setup.py

Author: tanjeffreyz

.idea/vcs.xml

@@ -1,39 +1,52 @@
 """Creates a desktop shortcut that can run Auto Maple from anywhere."""
 
 import os
+import git
 import argparse
 import win32com.client as client
 
 
-parser = argparse.ArgumentParser()
-parser.add_argument('--stay', action='store_true')
-args = parser.parse_args()
+def create_desktop_shortcut():
+    """Creates and saves a desktop shortcut using absolute paths"""
+    print('\n[~] Creating a desktop shortcut for Auto Maple:')
+    CWD = os.getcwd()
+    TARGET = os.path.join(os.environ['WINDIR'], 'System32', 'cmd.exe')
+    PATH = os.path.join(os.environ['USERPROFILE'], 'Desktop', 'Auto Maple.lnk')
+    flag = "/c"
+    if args.stay:
+        flag = "/k"
+        print(" -  Leaving command prompt open after program finishes")
 
-# Create and save the shortcut using absolute paths
-print('[~] Creating a desktop shortcut for Auto Maple:')
-CWD = os.getcwd()
-TARGET = os.path.join(os.environ['WINDIR'], 'System32', 'cmd.exe')
-PATH = os.path.join(os.environ['USERPROFILE'], 'Desktop', 'Auto Maple.lnk')
-flag = "/c"
-if args.stay:
-    flag = "/k"
-    print(" ~  Leaving command prompt open after program finishes")
+    shell = client.Dispatch('WScript.Shell')
+    shortcut = shell.CreateShortCut(PATH)
+    shortcut.Targetpath = TARGET
+    shortcut.Arguments = flag + f' \"cd {CWD} & python main.py\"'
+    shortcut.IconLocation = os.path.join(CWD, 'assets', 'icon.ico')
+    shortcut.save()
 
-shell = client.Dispatch('WScript.Shell')
-shortcut = shell.CreateShortCut(PATH)
-shortcut.Targetpath = TARGET
-shortcut.Arguments = flag + f' \"cd {CWD} & python main.py\"'
-shortcut.IconLocation = os.path.join(CWD, 'assets', 'icon.ico')
-shortcut.save()
+    # Enable "run as administrator"
+    with open(PATH, 'rb') as lnk:
+        arr = bytearray(lnk.read())
 
-# Enable "run as administrator"
-with open(PATH, 'rb') as lnk:
-    arr = bytearray(lnk.read())
+    arr[0x15] = arr[0x15] | 0x20        # Set the 6th bit of 21st byte to 1
 
-arr[0x15] = arr[0x15] | 0x20        # Set the 6th bit of 21st byte to 1
+    with open(PATH, 'wb') as lnk:
+        lnk.write(arr)
+        print(' -  Enabled the "Run as Administrator" option')
+    print(' ~  Successfully created Auto Maple shortcut')
 
-with open(PATH, 'wb') as lnk:
-    lnk.write(arr)
-    print(' ~  Enabled the "Run as Administrator" option')
 
-print('[~] Successfully created Auto Maple shortcut')
+def update_submodules():
+    print('\n[~] Updating submodules:')
+    repo = git.Repo()
+    repo.git.submodule('update', '--init', '--recursive')
+    print(' ~  Finished updating submodules')
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--stay', action='store_true')
+    args = parser.parse_args()
+
+    create_desktop_shortcut()
+    update_submodules()