mirror of
https://github.com/krateng/maloja.git
synced 2023-08-10 21:12:55 +03:00
116 lines
2.1 KiB
Python
Executable File
116 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import subprocess
|
|
import sys
|
|
import signal
|
|
import os
|
|
import stat
|
|
|
|
|
|
neededmodules = [
|
|
"bottle",
|
|
"waitress",
|
|
"setproctitle"
|
|
]
|
|
|
|
SOURCE_URL = "https://github.com/krateng/maloja/archive/master.zip"
|
|
|
|
|
|
|
|
def install():
|
|
toinstall = []
|
|
for m in neededmodules:
|
|
try:
|
|
exec("import " + m) #I'm sorry
|
|
except:
|
|
toinstall.append(m)
|
|
|
|
if toinstall != []:
|
|
print("The following python modules need to be installed:")
|
|
for m in toinstall:
|
|
print("\t" + m)
|
|
else:
|
|
return
|
|
|
|
def getInstance():
|
|
try:
|
|
output = subprocess.check_output(["pidof","Maloja"])
|
|
pid = int(output)
|
|
return pid
|
|
except:
|
|
return None
|
|
|
|
|
|
|
|
|
|
def start():
|
|
install()
|
|
|
|
try:
|
|
p = subprocess.Popen(["python3","server.py"],stdout=subprocess.DEVNULL,stderr=subprocess.DEVNULL)
|
|
print("Maloja started! PID: " + str(p.pid))
|
|
return True
|
|
except:
|
|
print("Error while starting Maloja. Are you sure it's not already running?")
|
|
return False
|
|
|
|
|
|
def restart():
|
|
#pid = getInstance()
|
|
#if pid == None:
|
|
# print("Server is not running.")
|
|
#else:
|
|
# stop()
|
|
#start()
|
|
|
|
wasrunning = stop()
|
|
start()
|
|
return wasrunning
|
|
|
|
def stop():
|
|
pid = getInstance()
|
|
if pid == None:
|
|
print("Server is not running")
|
|
return False
|
|
else:
|
|
print("Stopping " + str(pid))
|
|
os.kill(pid,signal.SIGTERM)
|
|
return True
|
|
|
|
def update():
|
|
|
|
import urllib.request
|
|
import shutil
|
|
import tempfile
|
|
import zipfile
|
|
import distutils.dir_util
|
|
|
|
print("Updating Maloja...")
|
|
with urllib.request.urlopen(SOURCE_URL) as response:
|
|
with tempfile.NamedTemporaryFile(delete=True) as tmpfile:
|
|
shutil.copyfileobj(response,tmpfile)
|
|
|
|
with zipfile.ZipFile(tmpfile.name,"r") as z:
|
|
|
|
for f in z.namelist():
|
|
print("extracting " + f)
|
|
z.extract(f)
|
|
|
|
|
|
|
|
distutils.dir_util.copy_tree("./maloja-master/","./",verbose=2)
|
|
shutil.rmtree("./maloja-master")
|
|
print("Done!")
|
|
|
|
os.chmod("./maloja",os.stat("./maloja").st_mode | stat.S_IXUSR)
|
|
|
|
if stop(): start() #stop returns whether it was running before, in which case we restart it
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if sys.argv[1] == "start": start()
|
|
elif sys.argv[1] == "restart": restart()
|
|
elif sys.argv[1] == "stop": stop()
|
|
elif sys.argv[1] == "update": update()
|
|
|