utils.py

import ConfigParser
import os
import subprocess
import sys
import time

runningdir = os.path.split(os.path.abspath(os.path.realpath(sys.argv[0])))[0]


def restart():
    time.sleep(4)
    execfile(os.path.join(runningdir, 'premiumizer.py'), globals(), globals())


def update():
    del sys.argv[1:]
    time.sleep(2)
    subprocess.call(['git', '-C', os.path.join(runningdir, 'nzbtomedia'), 'pull'])
    subprocess.call(['git', '-C', runningdir, 'pull'])

    prem_config = ConfigParser.RawConfigParser()
    default_config = ConfigParser.RawConfigParser()
    prem_config.read(os.path.join(runningdir, 'settings.cfg'))
    default_config.read(os.path.join(runningdir, 'settings.cfg.tpl'))

    prem_config.set('update', 'updated', '1')
    with open(os.path.join(runningdir, 'settings.cfg'), 'w') as configfile:
        prem_config.write(configfile)
    if prem_config.getfloat('update', 'req_version') < default_config.getfloat('update', 'req_version'):
        import pip
        pip.main(['install', '-r', os.path.join(runningdir, 'requirements.txt')])
        prem_config.set('update', 'req_version', (default_config.getfloat('update', 'req_version')))
        with open(os.path.join(runningdir, 'settings.cfg'), 'w') as configfile:
            prem_config.write(configfile)
    if prem_config.getfloat('update', 'config_version') < default_config.getfloat('update', 'config_version'):
        import shutil
        shutil.copy(os.path.join(runningdir, 'settings.cfg'), os.path.join(runningdir, 'settings.cfg.old2'))
        shutil.copy(os.path.join(runningdir, 'settings.cfg.tpl'), os.path.join(runningdir, 'settings.cfg'))
        prem_config.read(os.path.join(runningdir, 'settings.cfg.old2'))
        default_config.read(os.path.join(runningdir, 'settings.cfg'))
        for section in prem_config.sections():
            if section in default_config.sections() and section != 'update':
                for key in prem_config.options(section):
                    if key in default_config.options(section):
                        default_config.set(section, key, (prem_config.get(section, key)))
        with open(os.path.join(runningdir, 'settings.cfg'), 'w') as configfile:
            default_config.write(configfile)

    if os_arg == '--windows':
        pass
    else:
        time.sleep(3)
        execfile(os.path.join(runningdir, 'premiumizer.py'), globals(), globals())


if len(sys.argv) == 3:
    option_arg = sys.argv[1]
    os_arg = sys.argv[2]
    if os_arg == '--windows':
        rootdir = runningdir[:-12]
    else:
        os_arg = ''
    if option_arg == '--restart':
        restart()
    elif option_arg == '--update':
        update()
else:
    sys.exit()