Ruud
128 lines
3.4 KiB
Python
128 lines
3.4 KiB
Python
from couchpotato.api import addApiView
|
|
from couchpotato.core.event import addEvent
|
|
from couchpotato.core.helpers.encoding import isInt
|
|
from couchpotato.core.helpers.request import getParams, jsonified
|
|
from couchpotato.core.helpers.variable import mergeDicts
|
|
import ConfigParser
|
|
import os.path
|
|
import time
|
|
|
|
|
|
class Settings():
|
|
|
|
bool = {'true':True, 'false':False}
|
|
options = {}
|
|
|
|
def __init__(self):
|
|
|
|
addApiView('settings', self.view)
|
|
addApiView('settings.save', self.saveView)
|
|
|
|
def setFile(self, file):
|
|
self.file = file
|
|
|
|
self.p = ConfigParser.RawConfigParser()
|
|
self.p.read(file)
|
|
|
|
from couchpotato.core.logger import CPLog
|
|
self.log = CPLog(__name__)
|
|
|
|
self.connectEvents()
|
|
|
|
def parser(self):
|
|
return self.p
|
|
|
|
def sections(self):
|
|
return self.p.sections()
|
|
|
|
def connectEvents(self):
|
|
addEvent('settings.options', self.addOptions)
|
|
addEvent('settings.register', self.registerDefaults)
|
|
addEvent('settings.save', self.save)
|
|
|
|
def registerDefaults(self, section_name, options = {}, save = True):
|
|
self.addSection(section_name)
|
|
for option, value in options.iteritems():
|
|
self.setDefault(section_name, option, value)
|
|
|
|
#self.log.debug('Defaults for "%s": %s' % (section_name, options))
|
|
|
|
if save:
|
|
self.save(self)
|
|
|
|
def set(self, section, option, value):
|
|
return self.p.set(section, option, self.cleanValue(value))
|
|
|
|
def get(self, option = '', section = 'core', default = ''):
|
|
try:
|
|
value = self.p.get(section, option)
|
|
return self.cleanValue(value)
|
|
except:
|
|
return default
|
|
|
|
def cleanValue(self, value):
|
|
if(isInt(value)):
|
|
return int(value)
|
|
|
|
if str(value).lower() in self.bool:
|
|
return self.bool.get(str(value).lower())
|
|
|
|
return value.strip()
|
|
|
|
def getValues(self):
|
|
values = {}
|
|
for section in self.sections():
|
|
values[section] = {}
|
|
for option in self.p.items(section):
|
|
(option_name, option_value) = option
|
|
values[section][option_name] = self.cleanValue(option_value)
|
|
return values
|
|
|
|
def save(self):
|
|
with open(self.file, 'wb') as configfile:
|
|
self.p.write(configfile)
|
|
|
|
self.log.debug('Saved settings')
|
|
|
|
def addSection(self, section):
|
|
if not self.p.has_section(section):
|
|
self.p.add_section(section)
|
|
|
|
def setDefault(self, section, option, value):
|
|
if not self.p.has_option(section, option):
|
|
self.p.set(section, option, value)
|
|
|
|
def addOptions(self, section_name, options):
|
|
|
|
if not self.options.get(section_name):
|
|
self.options[section_name] = options
|
|
else:
|
|
options['groups'] = self.options[section_name].get('groups') + options.get('groups')
|
|
self.options[section_name] = mergeDicts(self.options[section_name], options)
|
|
|
|
def getOptions(self):
|
|
return self.options
|
|
|
|
|
|
def view(self):
|
|
|
|
return jsonified({
|
|
'options': self.getOptions(),
|
|
'values': self.getValues()
|
|
})
|
|
|
|
def saveView(self):
|
|
|
|
params = getParams()
|
|
|
|
section = params.get('section')
|
|
option = params.get('name')
|
|
value = params.get('value')
|
|
|
|
self.set(section, option, value)
|
|
self.save()
|
|
|
|
return jsonified({
|
|
'success': True,
|
|
})
|