75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
import os
|
|
import shutil
|
|
import sys
|
|
import subprocess
|
|
import random
|
|
import string
|
|
import upgrade2dot2
|
|
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)
|
|
# TODO : fix this, at least print the error
|
|
except Exception, e:
|
|
pass
|
|
|
|
def get_rand_string(length=10):
|
|
return ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(length))
|
|
|
|
PATH_INI_FILE = '/etc/airtime/media-monitor.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/../media-monitor.cfg'%current_script_dir, PATH_INI_FILE)
|
|
|
|
# load config file
|
|
try:
|
|
config = ConfigObj(PATH_INI_FILE)
|
|
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)
|
|
shutil.copy('%s/../monit-airtime-media-monitor.cfg'%current_script_dir, '/etc/monit/conf.d/')
|
|
|
|
#create log dir
|
|
create_dir(config['log_dir'])
|
|
|
|
#copy python files
|
|
copy_dir("%s/.."%current_script_dir, config["bin_dir"])
|
|
# mm2
|
|
mm2_source = os.path.realpath(os.path.join(current_script_dir,
|
|
"../../media-monitor2"))
|
|
copy_dir(mm2_source, os.path.join( config["bin_dir"], "mm2" ))
|
|
# upgrade mm2 config
|
|
upgrade2dot2.upgrade(('%s/../media-monitor.cfg'%current_script_dir),
|
|
PATH_INI_FILE)
|
|
|
|
#copy init.d script
|
|
shutil.copy(config["bin_dir"]+"/airtime-media-monitor-init-d", "/etc/init.d/airtime-media-monitor")
|
|
|
|
except Exception, e:
|
|
print e
|