Exemplo n.º 1
0
def _read_config(path, try_backup=False):
    """ Read the complete INI file and check its version number
        if OK, pass values to config-database
    """
    global CFG, database, modified

    if try_backup or not os.path.exists(path):
        # Not found, try backup
        try:
            shutil.copyfile(path + '.bak', path)
            try_backup = True
        except IOError:
            pass

    if not os.path.exists(path):
        # No file found, create default INI file
        try:
            if not sabnzbd.WIN32:
                prev = os.umask(077)
            fp = open(path, "w")
            fp.write("__version__=%s\n[misc]\n[logging]\n" % CONFIG_VERSION)
            fp.close()
            if not sabnzbd.WIN32:
                os.umask(prev)
        except IOError:
            return False, 'Cannot create INI file %s' % path

    try:
        fp = open(path, 'rb')
        lines = fp.read().split('\n')
        if len(lines) == 1:
            fp.seek(0)
            lines = fp.read().split('\r')
        lines = [line.rstrip('\r\n') for line in lines]
        fp.close()

        try:
            # First try UTF-8 encoding
            CFG = configobj.ConfigObj(lines,
                                      default_encoding='utf-8',
                                      encoding='utf-8')
        except UnicodeDecodeError:
            # Failed, enable retry
            CFG = {}

        if not re.search(r'utf[ -]*8', CFG.get('__encoding__', ''), re.I):
            # INI file is still in 8bit ASCII encoding, so try Latin-1 instead
            CFG = configobj.ConfigObj(lines,
                                      default_encoding='cp1252',
                                      encoding='cp1252')

    except (IOError, configobj.ConfigObjError, UnicodeEncodeError), strerror:
        if try_backup:
            if isinstance(strerror, UnicodeEncodeError):
                strerror = 'Character encoding of the file is inconsistent'
            return False, '"%s" is not a valid configuration file<br>Error message: %s' % (
                path, strerror)
        else:
            # Try backup file
            return _read_config(path, True)
Exemplo n.º 2
0
def read_config(path):
    """ Read the complete INI file and check its version number
        if OK, pass values to config-database
    """
    global CFG, database, modified

    if not os.path.exists(path):
        # No file found, create default INI file
        try:
            if not sabnzbd.WIN32:
                prev = os.umask(077)
            fp = open(path, "w")
            fp.write("__version__=%s\n[misc]\n[logging]\n" % CONFIG_VERSION)
            fp.close()
            if not sabnzbd.WIN32:
                os.umask(prev)
        except IOError:
            return False, 'Cannot create INI file %s' % path

    try:
        CFG = configobj.ConfigObj(path)
        try:
            if int(CFG['__version__']) > int(CONFIG_VERSION):
                return False, "Incorrect version number %s in %s" % (
                    CFG['__version__'], path)
        except (KeyError, ValueError):
            CFG['__version__'] = CONFIG_VERSION
    except configobj.ConfigObjError, strerror:
        return False, '"%s" is not a valid configuration file<br>Error message: %s' % (
            path, strerror)
Exemplo n.º 3
0
#!/usr/bin/env python2
import sys
sys.path.append('/opt/sabnzbd')

from os import path
from sabnzbd import config
from sabnzbd.utils import configobj

print("Reading sabnzbd config file")
config.read_config('/opt/sabnzbd-data/sabnzbd.ini')

print("Reading sabnzbd_delta config file")
configDelta = configobj.ConfigObj('/tmp/config/sabnzbd_delta.ini')

print("Merging the sabnzbd configs")
config.CFG.merge(configDelta)

print("Writing resulting sabnzbd config file")
with open('/opt/sabnzbd-data/sabnzbd.ini', 'w') as configfile:
    config.CFG.write(configfile)



if path.exists('/opt/sabnzbd-data/nzbtomedia/autoProcessMedia.cfg'):
    print("Reading autoProcessMedia config file from existing file")
    autoProcessMediaPath='/opt/sabnzbd-data/nzbtomedia/autoProcessMedia.cfg'
else:
    print("Reading autoProcessMedia config file from default file")
    autoProcessMediaPath='/opt/nzbToMedia/autoProcessMedia.cfg.spec'
config = configobj.ConfigObj(autoProcessMediaPath)