rewrite/src/Config.py

import ConfigParser

# Process the configuration options
class Config:
    # Initialize these vars
    __mirror = {}
    __dists = {}
    __suggests = ""
    __recomends = ""
    
    def __init__(self, filename):
        cnf = ConfigParser.ConfigParser()
        cnf.read(filename)
        for section in cnf.sections():
            for option in cnf.options(section):
                if section == "mirror":
                    if option == "archs" or option == "files":
                        self.__mirror[option] = cnf.get(section, option).split(" ")
                    else:
                        self.__mirror[option] = cnf.get(section, option)
                else:
                    if not self.__dists.has_key(section):
                        self.__dists[section] = {}

                    if option == "filter" or option == "packages":
                        filters = cnf.get(section, option).split(" ")

                        for rule in filters:
                            rule = rule.split(":")

                            if not self.__dists[section].has_key(rule[0]):
                                self.__dists[section][rule[0]] = []

                            if option == "filter":
                                self.__dists[section][rule[0]].append({"section": rule[1], "priority": rule[2], "package": "*"})
                            else:
                                self.__dists[section][rule[0]].append({"section": "*", "priority": "*", "package": rule[1]})

    def getValue(self, section, option):
        if self.__options.has_key(section):
            if self.__option[section].has_key(option):
                return self.__option[section][option]
        return ""
        
    def getMirror(self):
        return self.__mirror

    def __getMirrorOption(self, option):
        if self.__mirror.has_key(option):
            return self.__mirror[option]
        return ""

    def getServer(self):
        return self.__getMirrorOption("server");

    def getArchs(self):
        return self.__getMirrorOption("archs");

    def getFiles(self):
        return self.__getMirrorOption("files");

    def getLocalDirectory(self):
        return self.__getMirrorOption("local_directory");

    def getDists(self):
        return self.__dists

    def getSuggests(self):
        if self.__getMirrorOption('get_suggests') == 'true':
            return True
        else:
            return False
        
    def getRecomends(self):
        if self.__getMirrorOption('get_recomends') == 'true':
            return True
        else:
            return False

    def getSections(self, dist_name):
        if self.__dists.has_key(dist_name):
            return self.__dists[dist_name]
        return ""

    def getFilters(self, dist_name, section_name):
        if self.__dists.has_key(dist_name):
            if self.__dists[dist_name].has_key(section_name):
                return self.__dists[dist_name][section_name]
        return ""
1	otavio	2	import ConfigParser
2
3			# Process the configuration options
4			class Config:
5			# Initialize these vars
6			__mirror = {}
7			__dists = {}
8			__suggests = ""
9			__recomends = ""
10
11			def __init__(self, filename):
12			cnf = ConfigParser.ConfigParser()
13			cnf.read(filename)
14			for section in cnf.sections():
15			for option in cnf.options(section):
16			if section == "mirror":
17			if option == "archs" or option == "files":
18			self.__mirror[option] = cnf.get(section, option).split(" ")
19			else:
20			self.__mirror[option] = cnf.get(section, option)
21			else:
22			if not self.__dists.has_key(section):
23			self.__dists[section] = {}
24
25			if option == "filter" or option == "packages":
26			filters = cnf.get(section, option).split(" ")
27
28			for rule in filters:
29			rule = rule.split(":")
30
31			if not self.__dists[section].has_key(rule[0]):
32			self.__dists[section][rule[0]] = []
33
34			if option == "filter":
35			self.__dists[section][rule[0]].append({"section": rule[1], "priority": rule[2], "package": "*"})
36			else:
37			self.__dists[section][rule[0]].append({"section": "", "priority": "", "package": rule[1]})
38
39			def getValue(self, section, option):
40			if self.__options.has_key(section):
41			if self.__option[section].has_key(option):
42			return self.__option[section][option]
43			return ""
44
45			def getMirror(self):
46			return self.__mirror
47
48			def __getMirrorOption(self, option):
49			if self.__mirror.has_key(option):
50			return self.__mirror[option]
51			return ""
52
53			def getServer(self):
54			return self.__getMirrorOption("server");
55
56			def getArchs(self):
57			return self.__getMirrorOption("archs");
58
59			def getFiles(self):
60			return self.__getMirrorOption("files");
61
62			def getLocalDirectory(self):
63			return self.__getMirrorOption("local_directory");
64
65			def getDists(self):
66			return self.__dists
67
68			def getSuggests(self):
69			if self.__getMirrorOption('get_suggests') == 'true':
70			return True
71			else:
72			return False
73
74			def getRecomends(self):
75			if self.__getMirrorOption('get_recomends') == 'true':
76			return True
77			else:
78			return False
79
80			def getSections(self, dist_name):
81			if self.__dists.has_key(dist_name):
82			return self.__dists[dist_name]
83			return ""
84
85			def getFilters(self, dist_name, section_name):
86			if self.__dists.has_key(dist_name):
87			if self.__dists[dist_name].has_key(section_name):
88			return self.__dists[dist_name][section_name]
89			return ""