56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from Citavi import ProjectManager
|
|
from Django import PersonIdentityManager
|
|
|
|
class PersonMapper():
|
|
|
|
def __init__(self):
|
|
self._citavi_project_manager = ProjectManager()
|
|
self._person_identity_manager = PersonIdentityManager()
|
|
|
|
def __del__(self):
|
|
del self._citavi_project_manager
|
|
del self._person_identity_manager
|
|
|
|
def get_unmapped_identities(self, project):
|
|
citavi_persons = self._citavi_project_manager.get_persons_from_project(project.id)
|
|
mapped_persons = self._person_identity_manager.get_mapped_identities_for_project(project)
|
|
|
|
print "Len citavi persons: " + str(len(citavi_persons))
|
|
print "Len mapped persons: " + str(len(mapped_persons))
|
|
|
|
# Prepare citavi_persons into a uuid->person dict, then eliminate mapped ones. This is ugly and a little slow.
|
|
citavi_uuid_dict = {}
|
|
for person in citavi_persons:
|
|
citavi_uuid_dict[person.ID] = person
|
|
|
|
if len(mapped_persons) == 0:
|
|
return citavi_uuid_dict
|
|
|
|
print len(citavi_uuid_dict)
|
|
|
|
for person in mapped_persons:
|
|
if person.citavi_uuid in citavi_uuid_dict:
|
|
# print "Match for: " + str(person.id) + " - uuid: " + str(person.citavi_uuid)
|
|
del citavi_uuid_dict[person.citavi_uuid]
|
|
|
|
print len(citavi_uuid_dict)
|
|
|
|
return citavi_uuid_dict
|
|
|
|
def create_new_identity(self, project, person):
|
|
self._person_identity_manager.create_identity(project, person.ID)
|
|
|
|
def get_person_by_uuid(self, project, uuid): #
|
|
return self._citavi_project_manager.get_person_by_uuid(project.id, uuid)
|
|
|
|
def test():
|
|
from frontend.models import Project
|
|
project = Project.objects.get(id=1)
|
|
p = PersonMapper()
|
|
unmapped = p.get_unmapped_identities(project)
|
|
print unmapped
|
|
for person in unmapped:
|
|
p.create_new_identity(project, person)
|