mirror of
https://github.com/OpenSolo/OpenSolo.git
synced 2025-05-02 15:44:24 +02:00
140 lines
3.2 KiB
Python
Executable File
140 lines
3.2 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
import sys
|
|
import time
|
|
import pwd
|
|
import subprocess
|
|
import os, pwd, grp
|
|
|
|
def get_username():
|
|
return pwd.getpwuid( os.getuid() )[ 0 ]
|
|
|
|
me = get_username()
|
|
if ( me != 'root' ) :
|
|
print(" must be run as root, sorry, aborting. \n");
|
|
exit(1);
|
|
|
|
# open logfile:
|
|
logfile = "/vagrant/progress.log"
|
|
log = open(logfile, "w+")
|
|
|
|
|
|
# we display results as-it-happens, but also capture them
|
|
def run_command_capture_results(cmd='true', doLog=True):
|
|
if doLog:
|
|
print >>log, "cmd: "+cmd+"\n"
|
|
print "cmd: "+cmd+"\n"
|
|
p = subprocess.Popen(cmd, shell=True, stderr=subprocess.PIPE,stdout=subprocess.PIPE)
|
|
ret = ''
|
|
|
|
dobreak = False
|
|
doNext = False
|
|
while True:
|
|
#print p.poll()
|
|
|
|
#out = p.stdout.readline()
|
|
out = p.stdout.read(1)
|
|
# poll() returns None if process is still running, or exit code if it's done
|
|
if out == "":
|
|
if p.poll() is not None:
|
|
#print "OUT BREAK : " + str(p.poll())
|
|
dobreak = True
|
|
# otherwise handle the data we got prom the process:
|
|
while out != '' and doNext == False:
|
|
#if doNext == True:
|
|
# out = out + "\n"
|
|
# doNext = False
|
|
sys.stdout.write(out)
|
|
sys.stdout.flush()
|
|
if doLog:
|
|
log.write(out)
|
|
log.flush()
|
|
ret += out
|
|
out = p.stdout.read(1)
|
|
#if out == "\n":
|
|
# doNext = True
|
|
|
|
#err = p.stderr.readline()
|
|
err = p.stderr.read(1)
|
|
if err == "":
|
|
if p.poll() is not None:
|
|
#print "ERR BREAK" + str(p.poll())
|
|
dobreak = True
|
|
while err != '' and doNext == False:
|
|
#if doNext == True:
|
|
# out = out + "\n"
|
|
# doNext = False
|
|
sys.stdout.write(err)
|
|
sys.stdout.flush()
|
|
if doLog:
|
|
log.write(err)
|
|
log.flush()
|
|
ret += err
|
|
err = p.stderr.read(1)
|
|
#if err == "\n":
|
|
# doNext = True
|
|
|
|
if dobreak == True:
|
|
break
|
|
return ret
|
|
|
|
|
|
def drop_privileges(uid_name='nobody', gid_name='nogroup'):
|
|
if os.getuid() != 0:
|
|
# We're not root so, like, whatever dude
|
|
return
|
|
|
|
# Get the uid/gid from the name
|
|
running_uid = pwd.getpwnam(uid_name).pw_uid
|
|
running_gid = grp.getgrnam(gid_name).gr_gid
|
|
|
|
# Remove group privileges
|
|
os.setgroups([])
|
|
|
|
# Try setting the new uid/gid
|
|
os.setgid(running_gid)
|
|
os.setuid(running_uid)
|
|
|
|
# Ensure a very conservative umask
|
|
old_umask = os.umask(002)
|
|
|
|
# tasks potentially needing root:
|
|
|
|
#run_command_capture_results("chown -R vagrant /vagrant/.*");
|
|
#run_command_capture_results("chown -R vagrant /vagrant/*");
|
|
|
|
# now actually do ita:
|
|
#print >>log, "dropping privs to 'vagrant'\n";
|
|
drop_privileges('vagrant','vagrant');
|
|
|
|
me = get_username()
|
|
if ( me != 'vagrant' ) :
|
|
print >>log, "error while trying to drop root and running as vagrant user, aborting. \n";
|
|
exit(1);
|
|
|
|
|
|
|
|
print >>log, "now running as 'vagrant' user.\nstarting run at:\n";
|
|
run_command_capture_results("date");
|
|
|
|
|
|
print >>log, "chdir /vagrant";
|
|
os.chdir("/vagrant");
|
|
|
|
run_command_capture_results("git pull");
|
|
|
|
run_command_capture_results("echo \"y\" |/vagrant/builder.sh");
|
|
|
|
print >>log, "finished run at:\n";
|
|
run_command_capture_results("date");
|
|
|
|
print "log closed. \n";
|
|
log.close();
|
|
|
|
print "log relocated. \n";
|
|
run_command_capture_results("mv /vagrant/progress.log /vagrant/www/latest_run.txt", False);
|
|
|
|
run_command_capture_results("chmod 664 /vagrant/www/latest_run.txt", False);
|
|
|
|
|