From 35a7b15eecbb60e19ef6411c1c54e70cb34363ca Mon Sep 17 00:00:00 2001
From: Weilbyte <git@weilbyte.dev>
Date: Sun, 5 Jul 2020 21:11:48 +0200
Subject: [PATCH] Utility
---
PVEDiscordDark.py | 170 ++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 170 insertions(+)
create mode 100644 PVEDiscordDark.py
diff --git a/PVEDiscordDark.py b/PVEDiscordDark.py
new file mode 100644
index 0000000..60eb95c
--- /dev/null
+++ b/PVEDiscordDark.py
@@ -0,0 +1,170 @@
+import sys,os
+import subprocess
+import urllib.request
+import os.path
+import shutil
+import os
+import time
+
+images = [
+ 'dd_cephblurp.png', 'dd_cephwhite.png',
+ 'dd_icon-cpu.png', 'dd_icon-hdd.png',
+ 'dd_icon-ram.png', 'dd_icon-swap.png',
+ 'dd_icon-cd.png', 'dd_icon-display.png',
+ 'dd_icon-network.png', 'dd_icon-cloud.png',
+ 'dd_icon-serial.png', 'dd_icon-usb.png',
+ 'dd_icon-pci.png', 'dd_logo.png',
+ 'dd_mini-bottom.png', 'dd_mini-top.png',
+ 'dd_readme', '/dd_tool-sprites.png',
+ 'dd_trigger.png', 'dd_loading.svg']
+
+class colors:
+ HEADER = '\033[95m'
+ OKBLUE = '\033[94m'
+ OKGREEN = '\033[92m'
+ WARNING = '\033[93m'
+ FAIL = '\033[91m'
+ NORMAL = '\033[0m'
+ BOLD = '\033[1m'
+ UNDERLINE = '\033[4m'
+
+def clear():
+ print("\033c", end="")
+
+def cprint(color, text, bold=False, inline=False):
+ endc = '\n'
+ if inline:
+ endc = ''
+ if bold:
+ print(colors.BOLD)
+ print(f'{color}{text}{colors.NORMAL}', end=endc)
+
+def getPVEVersion():
+ pv = subprocess.check_output('pveversion --verbose | grep pve-manager | cut -c 14- | cut -c -6', shell=True, stderr=open(os.devnull, 'w'))
+ pv = pv.decode('ascii').replace(" ", "").replace("\n","")
+ if (('/bin/' in pv) or (len(pv) > 7)):
+ return ""
+ else:
+ return pv
+
+def checkPVE():
+ ver = getPVEVersion()
+ if (len(ver) < 1):
+ cprint(colors.FAIL, 'Unable to detect Proxmox.', True)
+ cprint(colors.NORMAL, 'A Proxmox installation could not be detected.')
+ exit(1)
+ else:
+ cprint(colors.OKGREEN, f'\nPVE {ver} detected.')
+
+def checkConn():
+ try:
+ urllib.request.urlopen('http://google.com')
+ return
+ except:
+ cprint(colors.FAIL, 'An Internet connection is required to install PVEDiscordDark.', True)
+ cprint(colors.NORMAL, 'Connect to the Internet and try again.')
+ exit(1)
+
+def themeIsInstalled():
+ tplUsesTheme = False
+ tplFile = open('/usr/share/pve-manager/index.html.tpl')
+ tplLines = tplFile.readlines()
+ for line in tplLines:
+ if "<link rel='stylesheet' type='text/css' href='/pve2/css/dd_style.css'>" in line:
+ tplUsesTheme = True
+ if (os.path.isfile('/usr/share/pve-manager/css/dd_style.css') or tplUsesTheme):
+ return True
+ return False
+
+def installTheme():
+ clear()
+ doHeader()
+ baseURL = os.getenv('BASEURL', f'https://github.com/Weilbyte/PVEDiscordDark/raw/{os.getenv("BRANCH", "master")}')
+ cprint(colors.NORMAL, '\nBacking up index template file..')
+ shutil.copyfile('/usr/share/pve-manager/index.html.tpl', '/usr/share/pve-manager/index.html.tpl.bak')
+ cprint(colors.NORMAL, 'Downloading stylesheet..')
+ urllib.request.urlretrieve(f'{baseURL}/PVEDiscordDark/sass/PVEDiscordDark.css', '/usr/share/pve-manager/css/dd_style.css')
+ cprint(colors.NORMAL, 'Downloading patcher..')
+ urllib.request.urlretrieve(f'{baseURL}/PVEDiscordDark/js/PVEDiscordDark.js', '/usr/share/pve-manager/js/dd_patcher.js')
+ cprint(colors.NORMAL, 'Applying stylesheet and patcher..')
+ with open('/usr/share/pve-manager/index.html.tpl', 'a') as tplFile:
+ tplFile.write("<link rel='stylesheet' type='text/css' href='/pve2/css/dd_style.css'>")
+ tplFile.write("<script type='text/javascript' src='/pve2/js/dd_patcher.js'></script>")
+ cprint(colors.NORMAL, 'Replacing charts.js..')
+ if os.path.exists('/usr/share/javascript/extjs/charts.js'):
+ shutil.copyfile('/usr/share/javascript/extjs/charts.js', '/usr/share/javascript/extjs/charts.js.bak')
+ os.remove('/usr/share/javascript/extjs/charts.js')
+ urllib.request.urlretrieve(f'{baseURL}/PVEDiscordDark/js/charts.js', '/usr/share/javascript/extjs/charts.js')
+ for index, image in enumerate(images):
+ 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()
+
+def uninstallTheme():
+ clear()
+ doHeader()
+ cprint(colors.NORMAL, '\nCleaning up index template file..')
+ with open('/usr/share/pve-manager/index.html.tpl', 'r+') as tplFile:
+ tplLines = tplFile.readlines()
+ tplFile.seek(0)
+ for line in tplLines:
+ if ("<link rel='stylesheet' type='text/css' href='/pve2/css/dd_style.css'>" not in line or "<script type='text/javascript' src='/pve2/js/dd_patcher.js'></script>" not in line):
+ tplFile.write(line)
+ tplFile.truncate()
+ if os.path.exists('/usr/share/pve-manager/css/dd_style.css'):
+ cprint(colors.NORMAL, 'Removing stylesheet..')
+ os.remove('/usr/share/pve-manager/css/dd_style.css')
+ if os.path.exists('/usr/share/pve-manager/js/dd_patcher.js'):
+ cprint(colors.NORMAL, 'Removing patcher..')
+ os.remove('/usr/share/pve-manager/js/dd_patcher.js')
+ cprint(colors.NORMAL, 'Removing images..')
+ for asset in os.listdir('/usr/share/pve-manager/images/'):
+ 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()
+
+def doHeader():
+ cprint(colors.HEADER, '[~]', True, True)
+ cprint(colors.NORMAL, ' PVEDiscordDark Utility\n', False, True)
+
+def doMainMenu():
+ clear()
+ doHeader()
+ isInstalled = themeIsInstalled()
+ cprint(colors.NORMAL, '[I]', True, True)
+ cprint(colors.NORMAL, ' Install theme', False, True)
+ if isInstalled:
+ cprint(colors.NORMAL, '[U]', True, True)
+ cprint(colors.NORMAL, ' Uninstall theme', False, True)
+ cprint(colors.NORMAL, '[Q]', True, True)
+ cprint(colors.NORMAL, ' Exit', False, True)
+ choice = input('\n\n>? ')
+ choice = choice.upper().replace(" ", "")
+ if choice == 'I':
+ installTheme()
+ elif (choice == 'U' and isInstalled):
+ uninstallTheme()
+ elif choice == 'Q':
+ exit(0)
+ else:
+ doMainMenu()
+
+def main():
+ checkPVE()
+ checkConn()
+ time.sleep(0.5)
+ try:
+ doMainMenu()
+ except KeyboardInterrupt:
+ print('\n')
+ exit(0)
+
+
+if __name__ == "__main__":
+ main()
\ No newline at end of file
--
GitLab