| import sys |
| import json |
| import logging |
| |
| from pylons import c |
| |
| from ming.orm import session, MappedClass |
| |
| from allura import model as M |
| |
| log = logging.getLogger(__name__) |
| |
| def main(): |
| test = sys.argv[-1] == 'test' |
| projects = M.Project.query.find().all() |
| log.info('Restoring labels on projects') |
| for p in projects: |
| restore_labels(p, test) |
| if not test: |
| session(p).flush() |
| log.info('Restoring labels on artifacts') |
| for p in projects: |
| if p.parent_id: continue |
| c.project = p |
| for name, cls in MappedClass._registry.iteritems(): |
| if not issubclass(cls, M.Artifact): continue |
| if session(cls) is None: continue |
| for a in cls.query.find(): |
| restore_labels(a, test) |
| if not test: |
| M.artifact_orm_session.flush() |
| M.artifact_orm_session.clear() |
| |
| def restore_labels(obj, test=True): |
| if not obj.labels: return |
| labels = obj.labels |
| while True: |
| if not labels or labels[0] != '[': return |
| lbllen = map(len, labels) |
| if max(lbllen) != 1: return |
| if min(lbllen) != 1: return |
| s = ''.join(labels) |
| s = s.replace("u'", "'") |
| s = s.replace('u"', '"') |
| jobj = '{"obj":' + s.replace("'", '"') + '}' |
| try: |
| new_labels = json.loads(jobj)['obj'] |
| except ValueError: |
| # some weird problem with json decoding, just erase the labels |
| new_labels = [] |
| if not isinstance(new_labels, list): return |
| for lbl in new_labels: |
| if not isinstance(lbl, basestring): return |
| log.info('%s: %s => %s', obj.__class__, labels, new_labels) |
| labels = new_labels |
| if not test: |
| log.info('...actually restoring labels') |
| obj.labels = new_labels |
| |
| if __name__ == '__main__': |
| main() |
