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()
