Easy python script to add sites
Information: Usage: - siteinstaller.pyc -f <filename> - only *.tar.gz compressed files Credits: spreedated
This commit is contained in:
deltax
GitHub
parent
7b6a015955
commit
03796c11f4
|
|
@ -0,0 +1,242 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
import sys
|
||||
import getopt
|
||||
import os
|
||||
import subprocess
|
||||
|
||||
site_name = ''
|
||||
site_language = ''
|
||||
installed_sites = 0
|
||||
flux_cont = ''
|
||||
|
||||
|
||||
def main(argv):
|
||||
global site_name, site_language
|
||||
if os.geteuid() != 0:
|
||||
exit("You need to have root privileges to run this script.\nPlease try again, this time using 'sudo'. Exiting....")
|
||||
sys.exit()
|
||||
usage = '>>>\tUsage:\n' +'\t\tsiteinstaller.pyc -f <filename>' +'\n\t\tsiteinstaller.pyc --file <file>\n\n>>>\tOnly *.tar.gz compressed files!'
|
||||
try:
|
||||
opts, args = getopt.getopt(sys.argv[1:], 'f:h', ['file=', 'help'])
|
||||
if(len(opts) <= 0):
|
||||
print(usage)
|
||||
sys.exit()
|
||||
except getopt.GetoptError:
|
||||
print(usage)
|
||||
sys.exit(2)
|
||||
|
||||
for opt, arg in opts:
|
||||
if opt in ('-h', '--help'):
|
||||
print(usage)
|
||||
sys.exit(2)
|
||||
elif opt in ('-f', '--file'):
|
||||
if not('.tar.gz' in arg):
|
||||
print('ONLY *.tar.gz files supported. Sorry bro')
|
||||
sys.exit()
|
||||
if(os.path.isfile(arg) == False):
|
||||
print('Your file does not exist, maybe a typo?')
|
||||
sys.exit()
|
||||
site_name = arg[:arg.rfind('.tar.gz')]
|
||||
site_language = site_name[site_name.rfind('_')+1:]
|
||||
else:
|
||||
print(usage)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main(sys.argv[1:])
|
||||
|
||||
def check_fluxion():
|
||||
global installed_sites, flux_cont
|
||||
try:
|
||||
fl = open('fluxion','rw')
|
||||
flux_cont = fl.read()
|
||||
fl.close()
|
||||
except:
|
||||
print('Could not open fluxion, check permissions. Exiting...')
|
||||
sys.exit()
|
||||
if(flux_cont <= 100):
|
||||
print('No fluxion installation found.\nPlease use this installer INSIDE the fluxion folder. Exiting...')
|
||||
sys.exit()
|
||||
flv = ''
|
||||
flr = ''
|
||||
fld = ''
|
||||
if('version=' in flux_cont) and ('revision=' in flux_cont):
|
||||
flv = flux_cont[flux_cont.find('version=')+8:]
|
||||
flv = flv[:flv.find('\n')]
|
||||
flr = flux_cont[flux_cont.find('revision=')+9:]
|
||||
flr = flr[:flr.find('\n')]
|
||||
else:
|
||||
print('Corrupted fluxion installation. Exiting...')
|
||||
sys.exit()
|
||||
if(flv != '0.23'):
|
||||
print('Your fluxion version "' + flv + '". \nSupported versions are 0.23')
|
||||
sys.exit()
|
||||
|
||||
installed_sites = int(flux_cont.count('elif [ "$fluxass" ='))
|
||||
fls = str(installed_sites + 1)
|
||||
|
||||
# Check on double installation!
|
||||
fld = flux_cont[flux_cont.find('$DUMP_PATH/data/index.htm'):]
|
||||
if(site_name in fld):
|
||||
usdc = raw_input('Seems like there is already a site with the name "' + site_name + '" Do you want to continue anyway? [Y\\n]')
|
||||
if(len(usdc) <= 0) or (usdc == 'y') or (usdc == 'yes'):
|
||||
pass
|
||||
else:
|
||||
sys.exit()
|
||||
return flv + '#' + flr + '#' + fls
|
||||
|
||||
def welcome():
|
||||
flc = check_fluxion().split('#')
|
||||
wsn = int(22 - len(site_name))
|
||||
whitespacen = ''
|
||||
wsl = int((22+6) - len(site_language))
|
||||
whitespacel = ''
|
||||
for i in xrange(wsn):
|
||||
whitespacen += ' '
|
||||
if(i+1 == wsn) and (i+1 <= wsn):
|
||||
whitespacen += '#'
|
||||
for i in xrange(wsl):
|
||||
whitespacel += ' '
|
||||
if(i+1 == wsl) and (i+1 <= wsl):
|
||||
whitespacel += '#'
|
||||
|
||||
print(' # ############################################')
|
||||
print(' # FluxIon - Site Installer v0.10 #')
|
||||
print(' # ############################################')
|
||||
print('')
|
||||
print(' # ############## Fluxion found! ##############')
|
||||
print(' # Version: '+ flc[0] +' #')
|
||||
print(' # Revision: '+ flc[1] +' #')
|
||||
print(' # Installed Sites: '+ flc[2] +' #')
|
||||
print(' # ############################################')
|
||||
print('')
|
||||
print(' # ############################################')
|
||||
print(' # SiteName to install: ' + site_name + whitespacen)
|
||||
print(' # Language flag: ' + site_language + whitespacel)
|
||||
print(' # ############################################')
|
||||
print('')
|
||||
print(' # ########### Everything correct? ############')
|
||||
print('')
|
||||
usc = raw_input(' # Begin installation? [Y\\n]').lower()
|
||||
|
||||
if(len(usc) <= 0) or (usc == 'y') or (usc == 'yes'):
|
||||
pass
|
||||
else:
|
||||
print('\n\t # Nothing changed, your choice. Exiting...')
|
||||
sys.exit()
|
||||
|
||||
welcome()
|
||||
|
||||
def insert_at_secondlast_pos1():
|
||||
global flux_cont, site_name, site_language
|
||||
whitespaces = ''
|
||||
search_string = 'echo -e " "$red"["$yellow"$n"$red"]"$transparent"\e'
|
||||
ws = int(12 - len(site_name))
|
||||
for i in xrange(ws):
|
||||
whitespaces+=' '
|
||||
insert_site = 'echo -e " "$red"["$yellow"$n"$red"]"$transparent" ' + site_name + whitespaces + '[' + site_language + '] ";n=` expr $n + 1`\n'
|
||||
before = flux_cont[:int(flux_cont.rfind(search_string))]
|
||||
after = '\t\t\t' + flux_cont[int(flux_cont.rfind(search_string)):]
|
||||
flux_cont = before + insert_site + after
|
||||
|
||||
def insert_at_secondlast_pos2():
|
||||
global installed_sites, flux_cont, site_name
|
||||
site_number = str(installed_sites +1)
|
||||
insert_site = 'elif [ "$fluxass" = "'+ site_number +'" ]; then\n\t\t\t\t' + site_name + '\n\t\t\t\tbreak\n\n\t\t\t'
|
||||
flux_cont = flux_cont[:int(flux_cont.rfind('elif [ "$fluxass" ='))] + insert_site + flux_cont[int(flux_cont.rfind('elif [ "$fluxass" =')):]
|
||||
|
||||
def last_option_correct_number2():
|
||||
global installed_sites, flux_cont
|
||||
acc = int(flux_cont.rfind('elif [ "$fluxass" ='))
|
||||
acc0 = int(flux_cont[acc:].find('" = "')+5)
|
||||
before = flux_cont[:acc+acc0]
|
||||
after = flux_cont[acc+acc0+2:]
|
||||
flux_cont = before + str(installed_sites+2) + after
|
||||
|
||||
def insert_at_last_pos3():
|
||||
global flux_cont, site_name
|
||||
insert_site = '\n\nfunction ' + site_name + ' {\n\tmkdir $DUMP_PATH/data &>$flux_output_device\n\tcp $WORK_DIR/Sites/' + site_name + '/* $DUMP_PATH/data\n\t}'
|
||||
before = flux_cont[:int(flux_cont.rfind('}'))+1]
|
||||
after = flux_cont[int(flux_cont.rfind('}'))+1:]
|
||||
flux_cont = before + insert_site + after
|
||||
|
||||
print('')
|
||||
print(' # Creating backup...')
|
||||
try:
|
||||
subprocess.Popen(['cp','fluxion', 'bckp_fluxion'])
|
||||
except:
|
||||
print('No Permission')
|
||||
sys.exit()
|
||||
print(' # Done!')
|
||||
|
||||
print('')
|
||||
print(' # Copying files...')
|
||||
try:
|
||||
subprocess.Popen(['tar','xfz', site_name+'.tar.gz', '-C', 'Sites/'])
|
||||
except:
|
||||
print('Could not copy files...soz')
|
||||
sys.exit()
|
||||
print(' # Done!')
|
||||
|
||||
print('')
|
||||
print(' # Reconfiguring fluxion bash...')
|
||||
try:
|
||||
insert_at_secondlast_pos1()
|
||||
insert_at_secondlast_pos2()
|
||||
last_option_correct_number2()
|
||||
insert_at_last_pos3()
|
||||
except:
|
||||
print('Internal failure... we fucked it up')
|
||||
e = sys.exc_info()[0]
|
||||
print("Error: %s" % e )
|
||||
sys.exit()
|
||||
|
||||
print(' # Done!')
|
||||
|
||||
print('')
|
||||
print(' Rewriting fluxion bash...')
|
||||
try:
|
||||
wflux = open('fluxion','w')
|
||||
wflux.write(flux_cont)
|
||||
wflux.close()
|
||||
except:
|
||||
print('FATAL ERROR[501]...')
|
||||
print('Trying to restore from backup')
|
||||
try:
|
||||
subprocess.Popen(['mv','bckp_fluxion', 'fluxion'])
|
||||
print('FluxIon Backup restored...')
|
||||
except:
|
||||
print('FATAL ERROR[502]... sorry bro something went wrong, fluxion may be broken now...')
|
||||
sys.exit()
|
||||
print(' # Done!')
|
||||
|
||||
print('')
|
||||
print(' # Setting modes...')
|
||||
try:
|
||||
subprocess.Popen(['chmod','755', 'Sites/'+site_name])
|
||||
subprocess.Popen(['chmod','644', '-R', 'Sites/'+site_name])
|
||||
print(' # Done!')
|
||||
except:
|
||||
print('ERROR[506]... continue...')
|
||||
|
||||
print('')
|
||||
print(' # Verifying integrity of fluxion...')
|
||||
try:
|
||||
fluxit = open('fluxion','r')
|
||||
integ_fluxion = fluxit.read()
|
||||
fluxit.close()
|
||||
if(len(integ_fluxion) == len(flux_cont)):
|
||||
print(' # Done!')
|
||||
print(' # Deleting backup...')
|
||||
subprocess.Popen(['rm','bckp_fluxion'])
|
||||
else:
|
||||
print('FATAL ERROR[509]... your fluxion might be messed up')
|
||||
sys.exit()
|
||||
except:
|
||||
print('FATAL ERROR[503]... sorry bro something went wrong, fluxion may be broken now...')
|
||||
sys.exit()
|
||||
|
||||
print('\n # Finished, successfully installed "' + site_name + '" to your fluxion!')
|
||||
|
||||
|
||||
Loading…
Reference in New Issue