Wrapper comming together

This commit is contained in:
neersighted 2012-02-29 09:44:05 -08:00
parent d4a930dffb
commit 743dbb6acf

View file

@ -3,41 +3,90 @@
import sys import sys
import os import os
import subprocess import subprocess
import re
usage = "usage: ./cloudbot {start|stop|restart|status}"
pwd = os.getcwd() pwd = os.getcwd()
clearlog = ": >./bot.log && " clearlog = ": >./bot.log && "
start = "daemon -n cloudbot -O " + pwd + "/bot.log -r python " + pwd + "/bot.py"
stop = clearlog + "daemon -n cloudbot --stop"
restart = clearlog + "daemon -n cloudbot --restart"
command = ":" command = ":"
print " ______ __ ______ __ __ _______ .______ ______ .___________." daemoncheck = subprocess.check_output("locate /usr/bin/daemon", shell=True)
print " / || | / __ \ | | | | | \ | _ \ / __ \ | |" daemon = re.match(r'^/usr/bin/daemon$', daemoncheck)
print "| ,----'| | | | | | | | | | | .--. || |_) | | | | | `---| |----`" screencheck = subprocess.check_output("locate /usr/bin/screen", shell=True)
print "| | | | | | | | | | | | | | | || _ < | | | | | | " daemon = re.match(r'^/usr/bin/screen$', screencheck)
print "| `----.| `----.| `--' | | `--' | | '--' || |_) | | `--' | | | "
print " \______||_______| \______/ \______/ |_______/ |______/ \______/ |__| " error1 = "Neither screen nor daemon is installed! "\
print "http://git.io/cloudbot by lukeroge" "This script cannot run! {ERROR 1}"
error2 = "Could not find bot.py! Are you in the wrong folder? "\
(" + pwd + ") {ERROR 2}"
error3 = "Invalid choice, exiting! {ERROR 3}"
start = "echo " + error1
stop = "echo " + error1
restart = "echo " + error1
status = "ps aux|grep cloudbot"
if daemon:
start = "daemon -n cloudbot -O " + pwd + \
"/bot.log -r python " + pwd + "/bot.py"
stop = clearlog + "daemon -n cloudbot --stop"
restart = stop + " && " + start
elif screen:
start = "screen -S cloudbot -dm python" + pwd +\
"/bot.py >>" + pwd + "/bot.log 2>&1"
stop = clearlog + "kill `pidof /usr/bin/screen`"
restart = stop + " && " + start
else:
print error1
print " ______ __ ______ __ __ "\
" _______ .______ ______ .___________."
print " / || | / __ \ | | | | "\
"| \ | _ \ / __ \ | |"
print "| ,----'| | | | | | | | | | "\
"| .--. || |_) | | | | | `---| |----`"
print "| | | | | | | | | | | | "\
"| | | || _ < | | | | | | "
print "| `----.| `----.| `--' | | `--' | "\
"| '--' || |_) | | `--' | | | "
print " \______||_______| \______/ \______/ "\
"|_______/ |______/ \______/ |__| "
print "http://git.io/cloudbot "\
" by lukeroge"
if (len(sys.argv) > 1): if (len(sys.argv) > 1):
if( sys.argv[1] == 'start' ): if (sys.argv[1] == 'start'):
command = start command = start
elif( sys.argv[1] == 'stop' ): elif (sys.argv[1] == 'stop'):
command = stop command = stop
elif( sys.argv[1] == 'restart' ): elif (sys.argv[1] == 'restart'):
command = restart command = restart
elif (sys.argv[1] == 'status'):
command = status
else: else:
print "usage: ./cloudbot {start|stop|restart}" print usage
sys.exit() exit
else: else:
print "usage: ./cloudbot {start|stop|restart}" print "{1|start} {2|stop} {3|restart} {4|status} {5|exit}"
sys.exit() read = int(raw_input('Please choose a option: '))
if read == 1:
command = start
elif read == 2:
command = stop
elif read == 3:
command = restart
elif read == 4:
command = status
elif read == 5:
exit
else:
print error3
exit
if os.path.isfile("./bot.py"): if os.path.isfile("./bot.py"):
command = command command = command
else: else:
print "Could not find bot.py! Are you in the wrong folder? (" + pwd + ")" print error2
sys.exit() exit
subprocess.call(command, shell=True) subprocess.call(command, shell=True)
sys.exit() exit