def install_migration_projects(): """Migrate the projects data from the old CouchDB storage""" from models.account import Account from models.project import Project, Component, Label, Role for row in __couchdb().view('_design/projects/_view/list'): value = row['value'] value['_id'] = value.get('_id', '').replace(':', '/') # resolve the labels INT or EXT value['labels'] = [] if value.get('_id', '').startswith('INT'): value['labels'].append(Label.LABEL_INTERNAL) elif value.get('_id', '').startswith('EXT'): value['labels'].append(Label.LABEL_EXTERNAL) elif value.get('_id', '').startswith('MGM') or value.get('_id', '').startswith('OUT'): value['labels'].append(Label.LABEL_VACATION) elif value.get('_id', '') == 'NOKIA/OFFICIAL': value['labels'].append(Label.LABEL_VACATION) value['_id'] = value.get('_id', '').replace('EXT/', '').replace('INT/', '') project = Project.query.filter_by(alias=value.get('_id', '')).first() if not project: project = Project() project.alias = value.get('_id', '') project.title = value.get('title') project.info = value.get('description') if value.get('deleted', False): project.status = Project.STATUS_ACTIVE | Project.STATUS_DELETED project.save() print '[MIGRATION:PROJECT]', project.__str__() items = [(None, value.get('title'))] for k,v in value.get('partitions', {}).items(): treeRecursion(k, v, None, items) for component_alias, component_title in items: if not component_alias: continue component = Component.query.filter_by(project=project, alias=component_alias).first() if not component: component = Component() component.alias = component_alias component.title = component_title component.project = project component.save() print '[MIGRATION:COMPONENT]', project.__str__(), component.__str__() for labelItem in value.get('labels', []): label = Label.query.filter_by(title=labelItem, project=project).first() if not label: label = Label() label.title = labelItem label.project = project label.save()