def update_tags(old_obj, translate_dict):
new_tags = []
for (id, (_, tag, is_new, is_dirty)) in tag_annotations.iteritems():
new_tag = DBTag(id=id, name=tag)
new_tag.is_new = is_new
new_tag.is_dirty = is_dirty
new_tags.append(new_tag)
return new_tags
def update_tags(old_obj, translate_dict):
new_tags = []
for (id, (_, tag, is_new, is_dirty)) in tag_annotations.iteritems():
new_tag = DBTag(id=id, name=tag)
new_tag.is_new = is_new
new_tag.is_dirty = is_dirty
new_tags.append(new_tag)
return new_tags
def translateVistrail(_vistrail):
tag_annotations = {}
notes_annotations = {}
thumb_annotations = {}
upgrade_annotations = {}
prune_annotations = {}
del_tag_annotations = {}
del_notes_annotations = {}
del_thumb_annotations = {}
del_upgrade_annotations = {}
del_prune_annotations = {}
key_lists = {'__tag__': tag_annotations,
'__notes__': notes_annotations,
'__thumb__': thumb_annotations,
'__upgrade__': upgrade_annotations,
'__prune__': prune_annotations}
del_key_lists = {'__tag__': del_tag_annotations,
'__notes__': del_notes_annotations,
'__thumb__': del_thumb_annotations,
'__upgrade__': del_upgrade_annotations,
'__prune__': del_prune_annotations}
_vistrail.update_id_scope()
id_scope = _vistrail.idScope
def update_tags(old_obj, translate_dict):
new_tags = []
for (id, (_, tag, is_new, is_dirty)) in tag_annotations.iteritems():
new_tag = DBTag(id=id, name=tag)
new_tag.is_new = is_new
new_tag.is_dirty = is_dirty
new_tags.append(new_tag)
return new_tags
def update_prune(old_obj, translate_dict):
if old_obj.db_id in prune_annotations:
(_, prune_val, _, _) = prune_annotations[old_obj.db_id]
if prune_val == str(True):
return 1
elif prune_val == str(False):
return 0
return None
def update_annotations(old_obj, translate_dict):
new_annotations = [DBAnnotation.update_version(a, translate_dict)
for a in old_obj.db_annotations]
if old_obj.db_id in notes_annotations:
(id, notes, is_new, is_dirty) = notes_annotations[old_obj.db_id]
ann = DBAnnotation(id=id,
key='__notes__',
value=notes)
ann.is_new = is_new
ann.is_dirty = is_dirty
new_annotations.append(ann)
if old_obj.db_id in upgrade_annotations:
(id, upgrade, is_new, is_dirty) = \
upgrade_annotations[old_obj.db_id]
ann = DBAnnotation(id=id,
key='__upgrade__',
value=upgrade)
ann.is_new = is_new
ann.is_dirty = is_dirty
new_annotations.append(ann)
if old_obj.db_id in thumb_annotations:
(id, thumb, is_new, is_dirty) = thumb_annotations[old_obj.db_id]
ann = DBAnnotation(id=id,
key='__thumb__',
value=thumb)
ann.is_new = is_new
ann.is_dirty = is_dirty
new_annotations.append(ann)
return new_annotations
def update_actions(old_obj, translate_dict):
new_actions = []
for action in old_obj.db_actions:
if action.db_id in del_notes_annotations:
(id, notes, is_new, is_dirty) = \
del_notes_annotations[action.db_id]
ann = DBAnnotation(id=id,
key='__notes__',
value=notes)
ann.is_new = is_new
ann.is_dirty = is_dirty
action.db_deleted_annotations.append(ann)
if action.db_id in del_upgrade_annotations:
(id, upgrade, is_new, is_dirty) = \
del_upgrade_annotations[action.db_id]
ann = DBAnnotation(id=id,
key='__upgrade__',
value=upgrade)
ann.is_new = is_new
ann.is_dirty = is_dirty
action.db_deleted_annotations.append(ann)
if action.db_id in del_thumb_annotations:
(id, thumb, is_new, is_dirty) = \
del_thumb_annotations[action.db_id]
ann = DBAnnotation(id=id,
key='__thumb__',
value=thumb)
ann.is_new = is_new
ann.is_dirty = is_dirty
action.db_deleted_annotations.append(ann)
new_actions.append(DBAction.update_version(action, translate_dict))
return new_actions
def update_workflow(old_obj, translate_dict):
return DBWorkflow.update_version(old_obj.db_workflow, translate_dict)
for a in _vistrail.db_actionAnnotations:
if a.db_key in key_lists:
key_lists[a.db_key][a.db_action_id] = \
(a.db_id, a.db_value, a.is_new, a.is_dirty)
for a in _vistrail.db_deleted_actionAnnotations:
if a.db_key in del_key_lists:
del_key_lists[a.db_key][a.db_action_id] = \
(a.db_id, a.db_value, a.is_new, a.is_dirty)
translate_dict = {'DBGroup': {'workflow': update_workflow},
'DBVistrail': {'tags': update_tags,
'actions': update_actions},
'DBAction': {'annotations': update_annotations,
'prune': update_prune}}
vistrail = DBVistrail.update_version(_vistrail, translate_dict)
for (id, (_, tag, is_new, is_dirty)) in del_tag_annotations.iteritems():
new_tag = DBTag(id=id, name=tag)
new_tag.is_new = is_new
new_tag.is_dirty = is_dirty
vistrail.db_deleted_tags.append(new_tag)
vistrail.db_version = '1.0.1'
return vistrail
def translateVistrail(_vistrail):
tag_annotations = {}
notes_annotations = {}
thumb_annotations = {}
upgrade_annotations = {}
prune_annotations = {}
del_tag_annotations = {}
del_notes_annotations = {}
del_thumb_annotations = {}
del_upgrade_annotations = {}
del_prune_annotations = {}
key_lists = {'__tag__': tag_annotations,
'__notes__': notes_annotations,
'__thumb__': thumb_annotations,
'__upgrade__': upgrade_annotations,
'__prune__': prune_annotations}
del_key_lists = {'__tag__': del_tag_annotations,
'__notes__': del_notes_annotations,
'__thumb__': del_thumb_annotations,
'__upgrade__': del_upgrade_annotations,
'__prune__': del_prune_annotations}
_vistrail.update_id_scope()
id_scope = _vistrail.idScope
def update_tags(old_obj, translate_dict):
new_tags = []
for (id, (_, tag, is_new, is_dirty)) in tag_annotations.iteritems():
new_tag = DBTag(id=id, name=tag)
new_tag.is_new = is_new
new_tag.is_dirty = is_dirty
new_tags.append(new_tag)
return new_tags
def update_prune(old_obj, translate_dict):
if old_obj.db_id in prune_annotations:
(_, prune_val, _, _) = prune_annotations[old_obj.db_id]
if prune_val == str(True):
return 1
elif prune_val == str(False):
return 0
return None
def update_annotations(old_obj, translate_dict):
new_annotations = [DBAnnotation.update_version(a, translate_dict)
for a in old_obj.db_annotations]
if old_obj.db_id in notes_annotations:
(id, notes, is_new, is_dirty) = notes_annotations[old_obj.db_id]
ann = DBAnnotation(id=id,
key='__notes__',
value=notes)
ann.is_new = is_new
ann.is_dirty = is_dirty
new_annotations.append(ann)
if old_obj.db_id in upgrade_annotations:
(id, upgrade, is_new, is_dirty) = \
upgrade_annotations[old_obj.db_id]
ann = DBAnnotation(id=id,
key='__upgrade__',
value=upgrade)
ann.is_new = is_new
ann.is_dirty = is_dirty
new_annotations.append(ann)
if old_obj.db_id in thumb_annotations:
(id, thumb, is_new, is_dirty) = thumb_annotations[old_obj.db_id]
ann = DBAnnotation(id=id,
key='__thumb__',
value=thumb)
ann.is_new = is_new
ann.is_dirty = is_dirty
new_annotations.append(ann)
return new_annotations
def update_actions(old_obj, translate_dict):
new_actions = []
for action in old_obj.db_actions:
if action.db_id in del_notes_annotations:
(id, notes, is_new, is_dirty) = \
del_notes_annotations[action.db_id]
ann = DBAnnotation(id=id,
key='__notes__',
value=notes)
ann.is_new = is_new
ann.is_dirty = is_dirty
action.db_deleted_annotations.append(ann)
if action.db_id in del_upgrade_annotations:
(id, upgrade, is_new, is_dirty) = \
del_upgrade_annotations[action.db_id]
ann = DBAnnotation(id=id,
key='__upgrade__',
value=upgrade)
ann.is_new = is_new
ann.is_dirty = is_dirty
action.db_deleted_annotations.append(ann)
if action.db_id in del_thumb_annotations:
(id, thumb, is_new, is_dirty) = \
del_thumb_annotations[action.db_id]
ann = DBAnnotation(id=id,
key='__thumb__',
value=thumb)
ann.is_new = is_new
ann.is_dirty = is_dirty
action.db_deleted_annotations.append(ann)
new_actions.append(DBAction.update_version(action, translate_dict))
return new_actions
def update_workflow(old_obj, translate_dict):
return DBWorkflow.update_version(old_obj.db_workflow, translate_dict)
for a in _vistrail.db_actionAnnotations:
if a.db_key in key_lists:
key_lists[a.db_key][a.db_action_id] = \
(a.db_id, a.db_value, a.is_new, a.is_dirty)
for a in _vistrail.db_deleted_actionAnnotations:
if a.db_key in del_key_lists:
del_key_lists[a.db_key][a.db_action_id] = \
(a.db_id, a.db_value, a.is_new, a.is_dirty)
translate_dict = {'DBGroup': {'workflow': update_workflow},
'DBVistrail': {'tags': update_tags,
'actions': update_actions},
'DBAction': {'annotations': update_annotations,
'prune': update_prune}}
vistrail = DBVistrail.update_version(_vistrail, translate_dict)
for (id, (_, tag, is_new, is_dirty)) in del_tag_annotations.iteritems():
new_tag = DBTag(id=id, name=tag)
new_tag.is_new = is_new
new_tag.is_dirty = is_dirty
vistrail.db_deleted_tags.append(new_tag)
vistrail.db_version = '1.0.1'
return vistrail