Mercurial > hg > pyloader
annotate pyloader/factory.py @ 22:b16d6a204ac1
stub a command line entry point
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Mon, 06 Jun 2011 07:41:23 -0700 |
parents | 4f7c05630f36 |
children | 9b2ca32e7a36 |
rev | line source |
---|---|
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
1 """ |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
2 abstract factories |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
3 """ |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
4 |
20
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
5 import cast |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
6 import loader |
15
0bea5297c156
introduce a more betterer test
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
7 import os |
22
b16d6a204ac1
stub a command line entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
8 import sys |
19
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
9 from ConfigParser import InterpolationMissingOptionError |
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
10 from ConfigParser import InterpolationSyntaxError |
16
edecb6fbd5a7
further stubbing of ini factory
Jeff Hammel <jhammel@mozilla.com>
parents:
15
diff
changeset
|
11 from ConfigParser import SafeConfigParser as ConfigParser |
15
0bea5297c156
introduce a more betterer test
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
12 |
0bea5297c156
introduce a more betterer test
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
13 __all__ = ['CircularReferenceError', 'PyFactory', 'IniFactory'] |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
14 |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
15 class CircularReferenceError(Exception): |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
16 """factory has detected a circular reference""" |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
17 |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
18 class PyFactory(object): |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
19 |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
20 delimeters = ('%(', ')s') |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
21 |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
22 def __init__(self, config=None, main=''): |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
23 self.main = main # main section |
10
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
24 self.configure(config or {}) |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
25 |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
26 def configure(self, config): |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
27 """load a new configuration""" |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
28 # TODO: this should really be a configuration update. If you keep |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
29 # track of all "apps" and their parents (i.e. as a ADG) |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
30 # you should be able to update only relevent apps |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
31 self.config = config |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
32 self.seen = set() # already seen apps to note cyclic dependencies |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
33 self.parsed = {} # instantiated apps |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
34 |
10
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
35 def load(self, name=None): |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
36 """load an object""" |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
37 |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
38 name = name or self.main # load main section by default |
10
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
39 assert name in self.config, "'%s' not found in configuration" |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
40 if name in self.parsed: |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
41 return self.parsed[name] |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
42 if name in self.seen: |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
43 raise CircularReferenceError('Circular reference! : %s' % name) |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
44 self.seen.add(name) |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
45 |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
46 # get section |
11 | 47 section = self.config[name] |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
48 assert 'path' in section |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
49 |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
50 # load object |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
51 obj = loader.load(section['path']) |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
52 |
10
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
53 # get the object's arguments (if any) |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
54 args = section.get('args', None) |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
55 kwargs = section.get('kwargs', None) |
10
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
56 |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
57 # if args and kwargs aren't there, you're done! |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
58 if args is None and kwargs is None: |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
59 self.parsed[name] = obj |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
60 return obj |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
61 |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
62 # interpolate arguments |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
63 if args: |
11 | 64 args = [self.interpolate(arg) for arg in args] |
65 else: | |
66 args = [] | |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
67 if kwargs: |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
68 kwargs = dict([(key, self.interpolate(value)) |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
69 for key, value in kwargs.items()]) |
11 | 70 else: |
71 kwargs = {} | |
10
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
72 |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
73 # invoke |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
74 self.parsed[name] = obj(*args, **kwargs) |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
75 return self.parsed[name] |
9
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
76 |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
77 def interpolate(self, value): |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
78 |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
79 # only interpolate strings |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
80 if not isinstance(value, basestring): |
ff634cc2e62b
unfinished sketch of an abstract factory
Jeff Hammel <jhammel@mozilla.com>
parents:
diff
changeset
|
81 return value |
10
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
82 |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
83 if value.startswith(self.delimeters[0]) and value.endswith(self.delimeters[1]): |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
84 value = value[len(self.delimeters[0]):-len(self.delimeters[1])] |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
85 if value in self.config: |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
86 return self.load(value) |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
87 return value |
ba2355d57998
finish baseline abstract factory implementation
Jeff Hammel <jhammel@mozilla.com>
parents:
9
diff
changeset
|
88 |
15
0bea5297c156
introduce a more betterer test
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
89 class IniFactory(PyFactory): |
0bea5297c156
introduce a more betterer test
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
90 |
0bea5297c156
introduce a more betterer test
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
91 def __init__(self, inifile, main=''): |
0bea5297c156
introduce a more betterer test
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
92 assert os.path.exists(inifile), "File not found: %s" % inifile |
21
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
93 self.inifile = inifile |
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
94 config = self.read(inifile) |
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
95 PyFactory.__init__(self, config, main) |
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
96 |
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
97 @classmethod |
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
98 def read(cls, inifile): |
15
0bea5297c156
introduce a more betterer test
Jeff Hammel <jhammel@mozilla.com>
parents:
14
diff
changeset
|
99 """reads configuration from an .ini file""" |
16
edecb6fbd5a7
further stubbing of ini factory
Jeff Hammel <jhammel@mozilla.com>
parents:
15
diff
changeset
|
100 |
21
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
101 here = os.path.dirname(os.path.abspath(inifile)) |
16
edecb6fbd5a7
further stubbing of ini factory
Jeff Hammel <jhammel@mozilla.com>
parents:
15
diff
changeset
|
102 |
edecb6fbd5a7
further stubbing of ini factory
Jeff Hammel <jhammel@mozilla.com>
parents:
15
diff
changeset
|
103 # read configuration |
edecb6fbd5a7
further stubbing of ini factory
Jeff Hammel <jhammel@mozilla.com>
parents:
15
diff
changeset
|
104 defaults={'here': here, |
21
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
105 'this': os.path.abspath(inifile)} |
16
edecb6fbd5a7
further stubbing of ini factory
Jeff Hammel <jhammel@mozilla.com>
parents:
15
diff
changeset
|
106 parser = ConfigParser(defaults=defaults) |
edecb6fbd5a7
further stubbing of ini factory
Jeff Hammel <jhammel@mozilla.com>
parents:
15
diff
changeset
|
107 parser.optionxform = str # use whole case |
21
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
108 parser.read(inifile) |
18 | 109 |
110 # parse configuration | |
111 config = {} | |
112 for section in parser.sections(): | |
113 | |
114 # sanity check | |
115 assert ':' in section, "No : in section: %s" % section | |
20
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
116 |
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
117 # make a dict for the section |
18 | 118 name, path = section.split(':', 1) |
119 sect = config[name] = dict(path=path) | |
20
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
120 |
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
121 # read the options |
18 | 122 for option in parser.options(section): |
19
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
123 |
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
124 if option in parser.defaults(): |
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
125 # don't include the defaults |
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
126 continue |
20
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
127 |
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
128 # try to interpolate the option |
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
129 # otherwise, use the raw value |
19
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
130 try: |
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
131 value = parser.get(section, option) |
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
132 except (InterpolationMissingOptionError, InterpolationSyntaxError): |
8987867698ee
a bit more stubbing of .ini parsing
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
133 value = parser.get(section, option, raw=True) |
20
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
134 |
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
135 if option = '.': # positional arguments |
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
136 sect['args'] = cast.str2list(value) |
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
137 else: |
057ccfe310b2
finish basic form of .ini factory - that was easy
Jeff Hammel <jhammel@mozilla.com>
parents:
19
diff
changeset
|
138 sect.setdefault('kwargs', {})[option] = value |
21
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
139 |
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
140 return config |
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
141 |
22
b16d6a204ac1
stub a command line entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
142 def main(args=sys.argv[1:]): |
b16d6a204ac1
stub a command line entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
143 """command line entry point""" |
b16d6a204ac1
stub a command line entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
144 |
21
4f7c05630f36
tie up some factory loose ends and sweeten the API with syntactic sugar
Jeff Hammel <jhammel@mozilla.com>
parents:
20
diff
changeset
|
145 if __name__ == '__main__': |
22
b16d6a204ac1
stub a command line entry point
Jeff Hammel <jhammel@mozilla.com>
parents:
21
diff
changeset
|
146 main() |