m5/util/code_formatter.py

40     def __init__(self, formatter, frame, *args, **kwargs):
41         self.frame = frame
42         self.formatter = formatter
43         self.dict = self.formatter._dict
44         self.args = args
45         self.kwargs = kwargs
46         self.locals = {}
48     def __setitem__(self, item, val):
49         self.locals[item] = val
51     def __getitem__(self, item):
52         if item in self.locals:
53             return self.locals[item]
55         if item in self.kwargs:
56             return self.kwargs[item]
59             return self.frame.f_code.co_filename
62             return self.frame.f_lineno
64         if self.formatter.locals and item in self.frame.f_locals:
65             return self.frame.f_locals[item]
67         if item in self.dict:
68             return self.dict[item]
70         if self.formatter.globals and item in self.frame.f_globals:
71             return self.frame.f_globals[item]
78             return self.args[item]
127     def __init__(self, *args, **kwargs):
128         self._data = []
129         self._dict = {}
130         self._indent_level = 0
131         self._indent_spaces = 4
132         self.globals = kwargs.pop('globals', type(self).globals)
133         self.locals = kwargs.pop('locals', type(self).locals)
134         self._fix_newlines = \
135                 kwargs.pop('fix_newlines', type(self).fix_newlines)
138             self.__call__(args)
140     def indent(self, count=1):
141         self._indent_level += self._indent_spaces * count
143     def dedent(self, count=1):
144         assert self._indent_level >= (self._indent_spaces * count)
145         self._indent_level -= self._indent_spaces * count
147     def fix(self, status):
148         previous = self._fix_newlines
149         self._fix_newlines = status
152     def nofix(self):
153         previous = self._fix_newlines
154         self._fix_newlines = False
158         self._data = []
160     def write(self, *args):
162         for data in self._data:
166     def __str__(self):
167         data = string.join(self._data, '')
168         self._data = [ data ]
171     def __getitem__(self, item):
172         return self._dict[item]
174     def __setitem__(self, item, value):
175         self._dict[item] = value
177     def __delitem__(self, item):
178         del self._dict[item]
180     def __contains__(self, item):
181         return item in self._dict
183     def __iadd__(self, data):
184         self.append(data)
186     def append(self, data):
188             self._data.extend(data._data)
190             self._append(str(data))
192     def _append(self, data):
193         if not self._fix_newlines:
194             self._data.append(data)
197         initial_newline = not self._data or self._data[-1] == '\n'
200                 if self._indent_level:
201                     self._data.append(' ' * self._indent_level)
202                 self._data.append(line)
205                 self._data.append('\n')
209     def __call__(self, *args, **kwargs):
211             self._data.append('\n')
219         l = lookup(self, frame, *args, **kwargs)
277         self._append(d)
286         def __init__(self, **kwargs):
287             self.update(kwargs)
288         def __getattr__(self, attr):
289             return self[attr]