File: Config.py

package info (click to toggle)
disk-manager 1.0.1-5
  • links: PTS, VCS
  • area: main
  • in suites: squeeze
  • size: 1,764 kB
  • ctags: 439
  • sloc: python: 3,373; sh: 3,282; makefile: 119
file content (103 lines) | stat: -rw-r--r-- 3,200 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
# -*- coding: UTF-8 -*-
#
#  Config.py : Manage configuration file
#  Copyright (C) 2007 Mertens Florent <flomertens@gmail.com>
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#

import os
import pwd
import logging
import ConfigParser
from config import CONF_FILE
from Utility import get_user

# Default configuration :

general = {
"notify"       : "True" ,
"ntfs_driver"  : "ntfs" ,
"backend"      : "auto" ,
"fstab_naming" : "auto" ,
}

gui = {
"main_width"     : "700" ,
"main_height"    : "450" ,
"selected"       : "2" ,
"selected_order" : "ascending" 
}

detected = {
"detected" : "" ,
"known" : ""
}

categorie = {
"General"         : general ,
"Gui Config"      : gui ,
"Detected Device" : detected
} 


class Config(ConfigParser.SafeConfigParser) :

    def __init__(self) :
    
        ConfigParser.SafeConfigParser.__init__(self)
        try :
            self.read_config()
        except ConfigParser.ParsingError :
            logging.warning("Parsing error while loading conf file. Using default configuration...")
        for section in categorie.keys() :
            if not self.has_section(section) :
                self.add_section(section)
            for option in categorie[section].keys() :
                if not self.has_option(section,option) :
                    self.set(section,option,categorie[section][option])   
        self.apply_change()
    
    def set(self, section, option, value) :
    
        ConfigParser.SafeConfigParser.set(self, section, option, str(value))
        self.apply_change()
        
    def set_default(self, section, option) :
    
        ConfigParser.SafeConfigParser.set(self, section, option, categorie[section][option])
        self.apply_change()
        return ConfigParser.SafeConfigParser.get(self, section, option)
        
    def read_config(self) :
        ''' Ugly stuff. We actually store config file in the user home, to be able to
            read it when checking if there is new device '''

        self.user_config = get_user("dir") + CONF_FILE
        self.root_config = pwd.getpwuid(0).pw_dir + CONF_FILE
        if not self.user_config in self.read([self.root_config, self.user_config]) :
            conffile = open(self.user_config, "w")
            conffile.close()
            os.chown(self.user_config, get_user("uid"), get_user("gid"))
            

    def apply_change(self) :
    
        conffile = open(self.user_config, "w")
        self.write(conffile)
        conffile.close()
        self.read([self.root_config, self.user_config])