145 lines
4.9 KiB
Python
145 lines
4.9 KiB
Python
import os
|
|
import grp
|
|
import stat
|
|
import shutil
|
|
import sys
|
|
import subprocess
|
|
import random
|
|
import string
|
|
import re
|
|
from configobj import ConfigObj
|
|
|
|
if os.geteuid() != 0:
|
|
print "Please run this as root."
|
|
sys.exit(1)
|
|
|
|
def get_current_script_dir():
|
|
current_script_dir = os.path.realpath(__file__)
|
|
index = current_script_dir.rindex('/')
|
|
return current_script_dir[0:index]
|
|
|
|
def copy_dir(src_dir, dest_dir):
|
|
if (os.path.exists(dest_dir)) and (dest_dir != "/"):
|
|
shutil.rmtree(dest_dir)
|
|
if not (os.path.exists(dest_dir)):
|
|
#print "Copying directory "+os.path.realpath(src_dir)+" to "+os.path.realpath(dest_dir)
|
|
shutil.copytree(src_dir, dest_dir)
|
|
|
|
def create_dir(path):
|
|
try:
|
|
os.makedirs(path)
|
|
except Exception, e:
|
|
pass
|
|
|
|
def get_rand_string(length=10):
|
|
return ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(length))
|
|
|
|
def get_rand_string(length=10):
|
|
return ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(length))
|
|
|
|
def get_monit_version():
|
|
version = 0
|
|
try:
|
|
p = subprocess.Popen(['monit', '-V'], stdout=subprocess.PIPE)
|
|
out = p.communicate()[0].strip()
|
|
search = re.search(r'This is Monit version (.*)\n', out, re.IGNORECASE)
|
|
|
|
if search:
|
|
matches = search.groups()
|
|
if len(matches) == 1:
|
|
version = matches[0]
|
|
except Exception:
|
|
print "Could not get monit version"
|
|
|
|
return version
|
|
|
|
#return 1 if version1 > version2
|
|
#return 0 if version1 == version2
|
|
#return -1 if version1 < version2
|
|
def version_compare(version1, version2):
|
|
def normalize(v):
|
|
return [int(x) for x in re.sub(r'(\.0+)*$','', v).split(".")]
|
|
return cmp(normalize(version1), normalize(version2))
|
|
|
|
PATH_INI_FILE = '/etc/airtime/pypo.cfg'
|
|
|
|
try:
|
|
# Absolute path this script is in
|
|
current_script_dir = get_current_script_dir()
|
|
|
|
if not os.path.exists(PATH_INI_FILE):
|
|
shutil.copy('%s/../pypo.cfg'%current_script_dir, PATH_INI_FILE)
|
|
|
|
try:
|
|
os.remove("/etc/airtime/liquidsoap.cfg")
|
|
except Exception, e:
|
|
pass
|
|
gid = grp.getgrnam("pypo").gr_gid
|
|
os.chown("/etc/airtime", -1, gid)
|
|
os.chmod("/etc/airtime", stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR | stat.S_IRGRP | stat.S_IWGRP | stat.S_IXGRP | stat.S_IROTH | stat.S_IXOTH)
|
|
|
|
# load config file
|
|
try:
|
|
config = ConfigObj(PATH_INI_FILE)
|
|
config['rabbitmq_user'] = os.environ['RABBITMQ_USER']
|
|
config['rabbitmq_password'] = os.environ['RABBITMQ_PASSWORD']
|
|
config['rabbitmq_vhost'] = os.environ['RABBITMQ_VHOST']
|
|
config.write()
|
|
except Exception, e:
|
|
print 'Error loading config file: ', e
|
|
sys.exit(1)
|
|
|
|
#copy monit files
|
|
shutil.copy('%s/../../monit/monit-airtime-generic.cfg'%current_script_dir, '/etc/monit/conf.d/')
|
|
subprocess.call('sed -i "s/\$admin_pass/%s/g" /etc/monit/conf.d/monit-airtime-generic.cfg' % get_rand_string(), shell=True)
|
|
|
|
monit_version = get_monit_version()
|
|
if version_compare(monit_version, "5.3.0") >= 0:
|
|
shutil.copy('%s/../monit-airtime-liquidsoap.cfg' % current_script_dir, \
|
|
'/etc/monit/conf.d/monit-airtime-liquidsoap.cfg')
|
|
else:
|
|
shutil.copy('%s/../monit-pre530-airtime-liquidsoap.cfg' % current_script_dir, \
|
|
'/etc/monit/conf.d/monit-airtime-liquidsoap.cfg')
|
|
|
|
shutil.copy('%s/../monit-airtime-playout.cfg'%current_script_dir, '/etc/monit/conf.d/')
|
|
|
|
#create pypo log dir
|
|
create_dir(config['pypo_log_dir'])
|
|
os.system("chown -R pypo:pypo "+config["pypo_log_dir"])
|
|
|
|
#create liquidsoap log dir
|
|
create_dir(config['liquidsoap_log_dir'])
|
|
os.system("chown -R pypo:pypo "+config["liquidsoap_log_dir"])
|
|
|
|
#create cache, tmp, file dirs
|
|
create_dir(config['cache_dir'])
|
|
create_dir(config['file_dir'])
|
|
create_dir(config['tmp_dir'])
|
|
|
|
create_dir(config["base_recorded_files"])
|
|
|
|
#copy files to bin dir
|
|
copy_dir("%s/.."%current_script_dir, config["bin_dir"]+"/bin/")
|
|
|
|
# delete /usr/lib/airtime/pypo/bin/liquidsoap_scripts/liquidsoap.cfg
|
|
# as we don't use it anymore.(CC-2552)
|
|
os.remove(config["bin_dir"]+"/bin/liquidsoap_scripts/liquidsoap.cfg")
|
|
|
|
#set permissions in bin dir and cache dir
|
|
os.system("chmod 755 "+os.path.join(config["bin_dir"], "bin/liquidsoap_scripts/notify.sh"))
|
|
os.system("chown -R pypo:pypo "+config["bin_dir"])
|
|
os.system("chown -R pypo:pypo "+config["cache_base_dir"])
|
|
os.system("chown -R pypo:pypo "+config["base_recorded_files"])
|
|
|
|
#copy init.d script
|
|
shutil.copy(config["bin_dir"]+"/bin/airtime-playout-init-d", "/etc/init.d/airtime-playout")
|
|
shutil.copy(config["bin_dir"]+"/bin/airtime-liquidsoap-init-d", "/etc/init.d/airtime-liquidsoap")
|
|
|
|
#copy log rotate script
|
|
shutil.copy(config["bin_dir"]+"/bin/liquidsoap_scripts/airtime-liquidsoap.logrotate", "/etc/logrotate.d/airtime-liquidsoap")
|
|
|
|
except Exception, e:
|
|
print e
|
|
|
|
|