import web import gen from gen.db import GrampsDbBase from web.libdjango import DjangoInterface # from ReportBase._CommandLineReport import run_report # import djangodb # db = djangodb.DjangoDb() # run_report(db, "ancestor_report", off="txt", of="ar.txt", pid="I37") class Cursor(object): def __init__(self, model, func): self.model = model self.func = func def __enter__(self): return self def __iter__(self): return self.__next__() def __next__(self): for item in self.model.all(): yield (item.handle, self.func(item)) def __exit__(self, *args, **kwargs): pass class DjangoDb(GrampsDbBase): """ A Gramps Database Backend. This replicates the grampsdb functions. """ def __init__(self): super(DjangoDb, self).__init__() self.dji = DjangoInterface() self.readonly = False self.db_is_open = True def get_researcher(self): obj = gen.lib.Name() return obj def get_event_from_handle(self, handle): obj = gen.lib.Event() obj.unserialize(self.dji.get_event(self.dji.Event.get(handle=handle))) return obj def get_family_from_handle(self, handle): obj = gen.lib.Family() obj.unserialize(self.dji.get_family(self.dji.Family.get(handle=handle))) return obj def get_person_from_handle(self, handle): obj = gen.lib.Person() data = self.dji.get_person(self.dji.Person.get(handle=handle)) obj.unserialize(data) return obj def get_place_from_handle(self, handle): obj = gen.lib.Place() obj.unserialize(self.dji.get_place(self.dji.Place.get(handle=handle))) return obj def get_source_from_handle(self, handle): obj = gen.lib.Source() obj.unserialize(self.dji.get_source(self.dji.Source.get(handle=handle))) return obj def get_note_from_handle(self, handle): obj = gen.lib.Note() obj.unserialize(self.dji.get_note(self.dji.Note.get(handle=handle))) return obj def get_person_handles(self): return [person.handle for person in self.dji.Person.all()] def get_default_person(self): return None def iter_person_handles(self): return (person.handle for person in self.dji.Person.all()) def iter_families(self): return (self.get_family_from_handle(family.handle) for family in self.dji.Family.all()) def get_person_from_gramps_id(self, gramps_id): obj = gen.lib.Person() match_list = self.dji.Person.filter(gramps_id=gramps_id) if match_list.count() > 0: data = self.dji.get_person( match_list[0] ) obj.unserialize(data) return obj else: return None def get_number_of_people(self): return self.dji.Person.count() def get_number_of_families(self): return self.dji.Family.count() def get_number_of_notes(self): return self.dji.Note.count() def get_number_of_sources(self): return self.dji.Source.count() def get_number_of_media_objects(self): return self.dji.Media.count() def get_number_of_repositories(self): return self.dji.Repository.count() def get_place_cursor(self): return Cursor(self.dji.Place, self.dji.get_place) def get_person_cursor(self): return Cursor(self.dji.Person, self.dji.get_person) def get_family_cursor(self): return Cursor(self.dji.Family, self.dji.get_family) def get_events_cursor(self): return Cursor(self.dji.Event, self.dji.get_event) def get_source_cursor(self): return Cursor(self.dji.Source, self.dji.get_source) def has_person_handle(self, handle): return self.dji.Person.filter(handle=handle).count() == 1 def has_family_handle(self, handle): return self.dji.Family.filter(handle=handle).count() == 1 def has_source_handle(self, handle): return self.dji.Source.filter(handle=handle).count() == 1 def has_repository_handle(self, handle): return self.dji.Repository.filter(handle=handle).count() == 1 def has_note_handle(self, handle): return self.dji.Note.filter(handle=handle).count() == 1 def has_place_handle(self, handle): return self.dji.Place.filter(handle=handle).count() == 1 def get_raw_person_data(self, handle): return self.dji.get_person(self.dji.Person.get(handle=handle)) def get_raw_family_data(self, handle): return self.dji.get_family(self.dji.Family.get(handle=handle)) def get_raw_source_data(self, handle): return self.dji.get_source(self.dji.Source.get(handle=handle)) def get_raw_repository_data(self, handle): return self.dji.get_repository(self.dji.Repository.get(handle=handle)) def get_raw_note_data(self, handle): return self.dji.get_note(self.dji.Note.get(handle=handle)) def get_raw_place_data(self, handle): return self.dji.get_place(self.dji.Place.get(handle=handle)) def add_person(self, person, trans, set_gid=True): pass # if self.step == 0: # if not person.handle: # person.handle = Utils.create_id() # if not person.gramps_id: # person.gramps_id = self.find_next_person_gramps_id() # self.lookup[person.gramps_id] = person.handle # if self.dji.Person.filter(handle=person.handle).count() == 0: # print "add_person:", person.handle # self.dji.add_person(person.serialize()) # else: # print "update_person:", person.handle # person.handle = self.lookup[person.gramps_id] # self.dji.add_person_detail(person.serialize()) def add_family(self, family, trans, set_gid=True): pass # if self.step == 0: # if not family.handle: # family.handle = Utils.create_id() # if not family.gramps_id: # family.gramps_id = self.find_next_family_gramps_id() # self.lookup[family.gramps_id] = family.handle # if self.dji.Family.filter(handle=family.handle).count() == 0: # print "add_family:", family.handle # self.dji.add_family(family.serialize()) # else: # family.handle = self.lookup[family.gramps_id] # self.dji.add_family_detail(family.serialize()) def add_source(self, source, trans, set_gid=True): pass #print "add_source:", source.handle #if not source.handle: # source.handle = Utils.create_id() # self.dji.add_source(source.serialize()) #self.dji.add_source_detail(source.serialize()) def add_repository(self, repository, trans, set_gid=True): pass #print "add_repository:", repository.handle #if not repository.handle: # repository.handle = Utils.create_id() # self.dji.add_repository(repository.serialize()) #self.dji.add_repository_detail(repository.serialize()) def add_note(self, note, trans, set_gid=True): pass #print "add_note:", note.handle #if not note.handle: # note.handle = Utils.create_id() # self.dji.add_note(note.serialize()) #self.dji.add_note_detail(note.serialize()) def add_place(self, place, trans, set_gid=True): #print "add_place:", place.handle pass def add_event(self, event, trans, set_gid=True): pass #print "add_event:", event.handle #if not event.handle: # event.handle = Utils.create_id() # self.dji.add_event(event.serialize()) #self.dji.add_event_detail(event.serialize()) def commit_person(self, person, trans, change_time=None): pass #print "commit_person:", person.handle #self.add_person(person, trans) def commit_family(self, family, trans, change_time=None): pass #print "commit_family:", family.handle #self.add_family(family, trans) def commit_source(self, source, trans, change_time=None): pass #print "commit_source:", source.handle #self.add_source(source, change_time) def commit_repository(self, repository, trans, change_time=None): pass #print "commit_repository:", repository.handle #self.add_repository(repository, change_time) def commit_note(self, note, trans, change_time=None): pass #print "commit_note:", note.handle #self.add_note(note, change_time) def commit_place(self, place, trans, change_time=None): pass #print "commit_place:", place.handle #if self.dji.Place.filter(handle=place.handle).count() == 0: # self.dji.add_place(place.serialize()) #self.dji.add_place_detail(place.serialize()) def commit_event(self, event, change_time=None): pass #print "commit_event:", event.handle #self.add_event(event, change_time) # def find_family_from_handle(self, handle, trans): # obj = gen.lib.Family() # results = self.dji.Family.filter(handle=handle) # if results.count() == 0: # obj.handle = handle # new = True # else: # data = self.dji.get_family(results[0]) # obj.unserialize(data) # new = False # return obj, new