Newer
Older
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',
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
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/javascript/extjs/charts.js.bak'):
cprint(colors.NORMAL, 'Reverting charts.js replacement..')
os.remove('/usr/share/javascript/extjs/charts.js')
shutil.copyfile('/usr/share/javascript/extjs/charts.js.bak', '/usr/share/javascript/extjs/charts.js')
os.remove('/usr/share/javascript/extjs/charts.js.bak')
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
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__":