From 692442eb964d473c2d37de30d387a51699a827ed Mon Sep 17 00:00:00 2001
From: Weilbyte <git@weilbyte.dev>
Date: Sat, 29 Aug 2020 05:01:21 +0200
Subject: [PATCH] Add unattended mode, and change google.com ping to github.com

---
 PVEDiscordDark.py | 48 ++++++++++++++++++++++++++++++++---------------
 1 file changed, 33 insertions(+), 15 deletions(-)

diff --git a/PVEDiscordDark.py b/PVEDiscordDark.py
index 4cc6c33..bdd4efa 100644
--- a/PVEDiscordDark.py
+++ b/PVEDiscordDark.py
@@ -3,9 +3,12 @@ import subprocess
 import urllib.request
 import os.path
 import shutil
+import argparse
 import os
 import time
 
+ACTION = None 
+
 images = [
     'dd_cephblurp.png', 'dd_cephwhite.png',
     'dd_icon-cpu.png', 'dd_icon-hdd.png',
@@ -30,7 +33,8 @@ class colors:
     UNDERLINE = '\033[4m'
 
 def clear():
-    print("\033c", end="")
+    if ACTION == None:
+        print("\033c", end="")
 
 def cprint(color, text, bold=False, inline=False):
     endc = '\n'
@@ -59,7 +63,7 @@ def checkPVE():
 
 def checkConn():
     try:
-        urllib.request.urlopen('http://google.com')
+        urllib.request.urlopen('http://github.com')
         return
     except:
         cprint(colors.FAIL, 'An Internet connection is required to install PVEDiscordDark.', True)
@@ -95,9 +99,10 @@ def installTheme():
         cprint(colors.NORMAL, f'Downloading images [{index + 1}/{len(images)}]..\r', False, True)
         urllib.request.urlretrieve(f'{baseURL}/PVEDiscordDark/images/{image}', f'/usr/share/pve-manager/images/{image}')
     cprint(colors.OKGREEN, '\nTheme installed successfully!', True)
-    cprint(colors.NORMAL, 'Press [ENTER] to go back.')
-    input('')
-    doMainMenu()
+    if ACTION == None:
+        cprint(colors.NORMAL, 'Press [ENTER] to go back.')
+        input('')
+        doMainMenu()
 
 def uninstallTheme():
     clear()
@@ -126,9 +131,10 @@ def uninstallTheme():
         if asset.startswith('dd_'):
             os.remove(f'/usr/share/pve-manager/images/{asset}')
     cprint(colors.OKGREEN, '\n\nTheme uninstalled successfully!', True)
-    cprint(colors.NORMAL, 'Press [ENTER] to go back.')
-    input('')
-    doMainMenu()
+    if ACTION == None:
+        cprint(colors.NORMAL, 'Press [ENTER] to go back.')
+        input('')
+        doMainMenu()
 
 def doHeader():
     cprint(colors.HEADER, '[~]', True, True)
@@ -157,15 +163,27 @@ def doMainMenu():
         doMainMenu()
 
 def main():
+    parser = argparse.ArgumentParser(description='PVEDiscordDark Theme Utility')
+    parser.add_argument('--action', '-a', choices=['install', 'uninstall'], help='action for unattended mode')
+    args = parser.parse_args()
+    global ACTION
+    ACTION = args.action
     checkPVE()
     checkConn()
     time.sleep(0.5)
-    try:
-        doMainMenu()
-    except KeyboardInterrupt:
-        print('\n')
-        exit(0)
-
+    if ACTION == None:
+        try:
+            print('mm')
+            doMainMenu()
+        except KeyboardInterrupt:
+            print('\n')
+            exit(0)
+    else:
+        if ACTION == 'install': 
+            installTheme()
+        else:
+            if themeIsInstalled():
+                uninstallTheme()
 
 if __name__ == "__main__":
-    main()
+    main()
\ No newline at end of file