Revision 2641
Added by Aaron Marcuse-Kubitza about 12 years ago
| sql_gen.py | ||
|---|---|---|
| 178 | 178 |
class ColDict(dict): |
| 179 | 179 |
'''A dict that automatically makes inserted entries Col objects''' |
| 180 | 180 |
|
| 181 |
'''For params, see dict()''' |
|
| 182 |
def __init__(self, keys_table, *args, **kw_args): |
|
| 183 |
dict.__init__(self, *args, **kw_args) |
|
| 184 |
|
|
| 185 |
self.table = keys_table |
|
| 186 |
|
|
| 187 |
def __getitem__(self, key): |
|
| 188 |
return dict.__getitem__(self, self._key(key)) |
|
| 189 |
|
|
| 181 | 190 |
def __setitem__(self, key, value): |
| 182 |
return dict.__setitem__(self, key, as_Col(value, name=key))
|
|
| 191 |
return dict.__setitem__(self, self._key(key), as_Col(value, name=key))
|
|
| 183 | 192 |
|
| 184 | 193 |
def update(self, dict_): |
| 185 | 194 |
for key, value in dict_.iteritems(): self[key] = value |
| 195 |
|
|
| 196 |
def _key(self, key): return as_Col(key, self.table) |
|
| 186 | 197 |
|
| 187 | 198 |
##### Functions |
| 188 | 199 |
|
Also available in: Unified diff
sql_gen.py: ColDict: Take a keys_table param for operations that require the output table. Use this to convert all string keys to sql_gen.Col objects.