pygtktalog/src/models/m_config.py

# This Python file uses the following encoding: utf-8
import utils._importer
import utils.globals
from gtkmvc import Model

import sys
import os

import gtk
import gobject

from ConfigParser import ConfigParser

class Ini(object):
    def __init__(self):
        self.ini = []
        
    def add_section(self, section):
        self.ini.append("[%s]" % section)
        
    def add_key(self, key, value):
        self.ini.append("%s=%s" % (key, value))
        
    def add_comment(self, comment):
        self.ini.append(";%s" % comment)
        
    def add_verb(self, verb):
        self.ini.append(verb)

    def show(self):
        return "\n".join(self.ini)

class ConfigModel(Model):
    ini = Ini()
    
    __properties__ = {}
    
    confd = {
        'savewin' : True,
        'savepan' : True,
        'wx' : 800,
        'wy' : 600,
        'h' : 200,
        'v' : 300,
        'exportxls' : False,
        'cd' : '/cdrom',
        'ejectapp' : 'eject -r',
        'eject' : True,
        'pil': False,
        'gthumb':False,
        'exif':False,
        'confirmquit':True,
        'mntwarn':True,
        'confirmabandon':True,
        'showtoolbar':True,
        'showstatusbar':True,
        'delwarn':True,
    }
    
    dictconf = {
        "save main window size" : "savewin",
        "save panes size" : "savepan",
        "main window width" : "wx",
        "main window height": "wy",
        "horizontal panes": "h",
        "vertical panes":"v",
        "export xls":"exportxls",
        "cd drive":"cd",
        "eject command":"ejectapp",
        "eject":"eject",
        "image support":"pil",
        'confirm quit':'confirmquit',
        'warn mount/umount errors':'mntwarn',
        'warn on delete':'delwarn',
        'confirm abandon current catalog':'confirmabandon',
        'show toolbar':'showtoolbar',
        'show statusbar and progress bar':'showstatusbar',
    }
    
    dbool = (
             'exportxls',
             'pil',
             'savewin',
             'savepan',
             'eject',
             'gthumb',
             'exif',
             'confirmquit',
             'mntwarn',
             'delwarn',
             'confirmabandon',
             'showtoolbar',
             'showstatusbar',
    )
    
    recent = []
    RECENT_MAX = 10
    
    dstring = ('cd','ejectapp')
    
    try:
        path = os.environ['HOME']
    except:
        path = "/tmp"
    
    def __init__(self):
        Model.__init__(self)
        self.category_tree = gtk.ListStore(gobject.TYPE_STRING)
        return
        
    def save(self):
        try:
            os.lstat("%s/.pygtktalog" % self.path)
        except:
            if __debug__:
                print "m_config.py: save() Saving preferences to %s/.pygtktalog" % self.path
        newIni = Ini()
        newIni.add_section("pyGTKtalog conf")
        for opt in self.dictconf:
            newIni.add_key(opt,self.confd[self.dictconf[opt]])
        newIni.add_section("pyGTKtalog recent")
        
        count = 1
        max_count = self.RECENT_MAX + 1
        
        for opt in self.recent:
            if count < max_count:
                newIni.add_key(count, opt)
            else:
                break
            count+=1
        try:
            f = open("%s/.pygtktalog" % self.path,"w")
            success = True
        except:
            if __debug__:
                print "m_config.py: save() Cannot open config file %s for writing." % (self.path, "/.pygtktalog")
            success = False
        f.write(newIni.show())
        f.close()
        return success
        
    def load(self):
        try:
        # try to read config file
            parser = ConfigParser()
            parser.read("%s/.pygtktalog" % self.path)
            r = {}
            self.recent = []
            for sec in parser.sections():
                if sec == 'pyGTKtalog conf':
                    for opt in parser.options(sec):
                        try:
                            if self.dictconf[opt] in self.dbool:
                                self.confd[self.dictconf[opt]] = parser.getboolean(sec,opt)
                            elif self.dictconf[opt] in self.dstring:
                                self.confd[self.dictconf[opt]] = parser.get(sec,opt)
                            else:
                                self.confd[self.dictconf[opt]] = parser.getint(sec,opt)
                        except:
                            if __debug__:
                                print "m_config.py: load() failed to parse option:", opt
                            pass
                elif sec == 'pyGTKtalog recent':
                    
                    for opt in parser.options(sec):
                        try:
                            r[int(opt)] = parser.get(sec,opt)
                        except:
                            if __debug__:
                                print "m_config.py: load() failed to parse option:", opt
                            pass
            for i in range(1, self.RECENT_MAX + 1):
                if r.has_key(i):
                    self.recent.append(r[i])
                
        except:
            if __debug__:
                print "m_config.py: load() load config file failed"
            pass

    def add_recent(self, path):
        if not path:
            return
            
        if  path in self.recent:
            self.recent.remove(path)
            self.recent.insert(0,path)
            return
            
        self.recent.insert(0,path)
        if len(self.recent) > self.RECENT_MAX:
            self.recent = self.recent[:self.RECENT_MAX]
        return
        
    def __str__(self):
        """show prefs in string way"""
        string = "[varname]\tvalue\n"
        for i in self.confd:
            string+="%s\t%s\n" % (i,self.confd[i])
        return string