from __future__ import with_statement from couchpotato.api import addApiView from couchpotato.core.event import addEvent from flask.helpers import jsonify 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 = 'global', default = ''): try: value = self.p.get(section, option) return self.cleanValue(value) except: return default def cleanValue(self, value): if(self.is_int(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] = 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 is_int(self, value): try: int(value) return True except ValueError: return False def addOptions(self, section_name, options): self.options[section_name] = options def getOptions(self): return self.options def view(self): return jsonify({ 'options': self.getOptions(), 'values': self.getValues() }) def saveView(self): a = Env.getParams() section = a.get('section') option = a.get('name') value = a.get('value') self.set(option, section, value) self.save() return jsonify({ 'success': True, })