Nearing completition; most data is pulled out of the SQL db, now I need to get the relations fixed

svn: r12918
This commit is contained in:
Doug Blank 2009-08-08 04:18:34 +00:00
parent f3a7e259e5
commit 24750e0272
2 changed files with 307 additions and 99 deletions

View File

@ -67,10 +67,12 @@ def makeDB(db):
db.query("""drop table event_ref;""")
db.query("""drop table source_ref;""")
db.query("""drop table child_ref;""")
db.query("""drop table person_ref;""")
db.query("""drop table lds;""")
db.query("""drop table media_ref;""")
db.query("""drop table address;""")
db.query("""drop table attribute;""")
db.query("""drop table url;""")
# Completed
db.query("""CREATE TABLE note (
handle CHARACTER(25),
@ -166,6 +168,7 @@ def makeDB(db):
private BOOLEAN);""")
db.query("""CREATE TABLE source (
from_type CHARACTER(25),
handle CHARACTER(25),
gid CHARACTER(25),
title TEXT,
@ -224,7 +227,7 @@ def makeDB(db):
from_type CHARACTER(25),
from_handle CHARACTER(25),
handle CHARACTER(25),
desc TEXT,
description TEXT,
private BOOLEAN);""")
db.query("""CREATE TABLE source_ref (
@ -239,8 +242,10 @@ def makeDB(db):
from_type CHARACTER(25),
from_handle CHARACTER(25),
ref CHARACTER(25),
frel INTEGER,
mrel INTEGER,
frel0 INTEGER,
frel1 CHARACTER(25),
mrel0 INTEGER,
mrel1 CHARACTER(25),
private BOOLEAN);""")
db.query("""CREATE TABLE lds (
@ -321,6 +326,9 @@ class Database(object):
self.cursor.close()
self.db.close()
def export_place(db, from_type, from_handle, place_handle):
export_link(db, from_type, from_handle, "place", place_handle)
def export_url_list(db, from_type, handle, urls):
for url in urls:
# (False, u'http://www.gramps-project.org/', u'loleach', (0, u'kaabgo'))
@ -346,8 +354,8 @@ def export_person_ref_list(db, from_type, from_handle, person_ref_list):
db.query("""INSERT INTO person_ref (
from_type,
from_handle,
handle
desc,
handle,
description,
private) VALUES (?, ?, ?, ?, ?);""",
from_type,
from_handle,
@ -389,6 +397,7 @@ def export_lds(db, from_type, from_handle, handle, type, place, famc, temple, st
db.query("""INSERT into lds (from_type, from_handle, handle, type, place, famc, temple, status, private)
VALUES (?,?,?,?,?,?,?,?,?);""",
from_type, from_handle, handle, type, place, famc, temple, status, private)
# FIXME: remove place from here?
def export_media_ref(db, handle, ref, role, private):
db.query("""INSERT into media_ref (
@ -559,18 +568,22 @@ def export_attribute_list(db, from_type, from_handle, attribute_list):
export_list(db, "attribute", attr_handle, "note", note_list)
export_source_list(db, "atribute", attr_handle, source_list)
def export_child_ref_list(db, from_type, from_handle, to_type, ref_list):
for child_ref in ref_list:
# family -> child_ref
# (False, [], [], u'b305e96e39652d8f08c', (1, u''), (1, u''))
(private, source_list, note_list, ref, frel, mrel) = child_ref
db.query("""INSERT INTO child_ref (from_type, from_handle,
ref, frel0, frel1, mrel0, mrel1, private)
VALUES (?, ?, ?, ?, ?, ?, ?, ?);""",
from_type, from_handle, ref, frel[0], frel[1],
mrel[0], mrel[1], private)
export_source_list(db, from_type, ref, source_list)
export_list(db, from_type, ref, "note", note_list)
def export_list(db, from_type, from_handle, to_type, handle_list):
for to_handle in handle_list:
if type(to_handle) in [str, unicode]:
export_link(db, from_type, from_handle, to_type, to_handle)
else:# family -> child_ref
# (False, [], [], u'b305e96e39652d8f08c', (1, u''), (1, u''))
(private, source_list, note_list, ref, frel, mrel) = to_handle
db.query("""INSERT INTO child_ref (from_type, from_handle, ref, frel, mrel, private)
VALUES (?, ?, ?, ?, ?, ?);""",
from_type, from_handle, ref, frel[0], mrel[0], private)
export_source_list(db, "child_ref", ref, source_list)
export_list(db, "child_ref", ref, "note", note_list)
def export_link(db, from_type, from_handle, to_type, to_handle):
db.query("""insert into link (
@ -663,7 +676,7 @@ def exportData(database, filename, option_box=None, callback=None):
private)
if place:
export_link(db, "event", handle, "place", place)
export_place(db, "event", handle, place)
export_list(db, "event", handle, "note", note_list)
export_attribute_list(db, "event", handle, attribute_list)
export_media_list(db, "event", handle, media_list)
@ -754,6 +767,7 @@ def exportData(database, filename, option_box=None, callback=None):
(lsource_list, lnote_list, date, type, place,
famc, temple, status, lprivate) = ldsord
lds_handle = create_id()
# FIXME: place?
export_lds(db, "person", handle, lds_handle, type, place, famc, temple, status, lprivate)
if date:
export_date(db, "lds", lds_handle, date)
@ -794,7 +808,7 @@ def exportData(database, filename, option_box=None, callback=None):
the_type[0], the_type[1], change, marker[0], marker[1],
private)
export_list(db, "family", handle, "child_ref", child_ref_list)
export_child_ref_list(db, "family", handle, "child_ref", child_ref_list)
export_list(db, "family", handle, "note", pnote_list)
export_attribute_list(db, "family", handle, attribute_list)
export_source_list(db, "family", handle, source_list)
@ -814,6 +828,7 @@ def exportData(database, filename, option_box=None, callback=None):
(lsource_list, lnote_list, date, type, place,
famc, temple, status, lprivate) = ldsord
lds_handle = create_id()
# FIXME: place?
export_lds(db, "family", handle, lds_handle, type, place, famc, temple, status, lprivate)
if date:
export_date(db, "lds", lds_handle, date)
@ -864,14 +879,14 @@ def exportData(database, filename, option_box=None, callback=None):
# Place
# ---------------------------------
for place_handle in database.place_map.keys():
repository = database.place_map[place_handle]
place = database.place_map[place_handle]
(handle, gid, title, long, lat,
main_loc, alt_location_list,
urls,
media_list,
source_list,
note_list,
change, marker, private) = repository
change, marker, private) = place
db.query("""INSERT INTO place (
handle,
@ -891,12 +906,18 @@ def exportData(database, filename, option_box=None, callback=None):
export_source_list(db, "place", handle, source_list)
export_list(db, "place", handle, "note", note_list)
# FIX: losing link to places?
for location in alt_location_list:
((street, city, county, state, country, postal, phone), parish) = location
addr_handle = create_id()
export_address(db, "place", handle, addr_handle, street, city, county, state,
country, postal, phone, private, parish)
# main_loc
(street, city, county, state, country, postal, phone,
private, parish) = main_loc
export_address(db, "place", handle, create_id(), street, city,
county, state, country, postal, phone, private,
parish)
count += 1
callback(100 * count/total)

View File

@ -65,7 +65,6 @@ def lookup(handle, event_ref_list):
count += 1
return -1
#-------------------------------------------------------------------------
#
# SQL Reader
@ -121,12 +120,111 @@ class SQLReader(object):
return None
return sql
def get_event_refs(self, sql, from_type, from_handle):
# -----------------------------------------------
# Get methods to retrieve data from the tables
# -----------------------------------------------
def get_address_list(self, sql, from_type, from_handle):
# FIXME
return []
def get_attribute_list(self, sql, from_type, from_handle):
rows = sql.query("select * from attribute where from_type = ? and from_handle = ?;",
from_type, from_handle)
retval = []
for row in rows:
(handle,
from_type,
from_handle,
the_type0,
the_type1,
value,
private) = row
source_list = self.get_source_list(sql, "attribute", handle)
note_list = self.get_note_list(sql, "attribute", handle)
retval.append((private, source_list, note_list,
(the_type0, the_type1), value))
return retval
def get_child_ref_list(self, sql, from_type, from_handle):
rows = sql.query("select * from child_ref where from_type = ? and from_handle = ?;",
from_type, from_handle)
retval = []
for row in rows:
(rtype, rhandle, ref, frel0, frel1, mrel0, mrel1, private) = row
source_list = self.get_source_list(sql, "child_ref", rhandle)
note_list = self.get_note_list(sql, "child_ref", rhandle)
retval.append((private, source_list, note_list, ref,
(frel0, frel1), (mrel0, mrel1)))
return retval
def get_datamap(self, sql, from_type, from_handle):
# FIXME:
return {}
def get_event_ref_list(self, sql, from_type, from_handle):
results = sql.query("select * from event_ref where from_type = ? and from_handle = ?;",
from_type,
from_handle)
return [self.pack_event_ref(sql, result) for result in results]
def get_family_list(self, sql, from_type, from_handle):
return self.get_list(sql, from_type, from_handle, "family")
def get_parent_family_list(self, sql, from_type, from_handle):
return self.get_list(sql, from_type, from_handle, "parent_family")
def get_person_ref_list(self, sql, from_type, from_handle):
rows = sql.query("select * from person_ref where from_type = ? and from_handle = ?;",
from_type, from_handle)
retval = []
for row in rows:
(from_type,
from_handle,
handle,
description,
private) = row
source_list = self.get_source_list(sql, "person_ref", handle)
note_list = self.get_note_list(sql, "person_ref", handle)
retval.append((private,
source_list,
note_list,
handle,
description))
return retval
def get_location_list(self, sql, from_type, from_handle):
# FIXME
return []
def get_lds_list(self, sql, from_type, from_handle):
# FIXME
return []
def get_media_list(self, sql, from_type, from_handle):
# FIXME
return []
def get_note_list(self, sql, from_type, from_handle):
return self.get_list(sql, from_type, from_handle, "note")
def get_repository_ref_list(self, sql, from_type, from_handle):
# FIXME
return []
def get_source_list(self, sql, from_type, from_handle):
results = sql.query("""select * from source where from_type = ? and handle = ?;""",
from_type, from_handle)
return [self.pack_source(sql, result) for result in results]
def get_url_list(self, sql, from_type, from_handle):
# FIXME
return []
# ---------------------------------
# Helpers
# ---------------------------------
def pack_event_ref(self, sql, data):
(from_type,
from_handle,
@ -134,27 +232,10 @@ class SQLReader(object):
role0,
role1,
private) = data
note_list = self.get_links(sql, "event_ref", from_handle, "note")
note_list = self.get_note_list(sql, "event_ref", from_handle)
attribute_list = self.get_attribute_list(sql, "event_ref", from_handle)
return (private, note_list, attribute_list, ref, (role0, role1))
def get_url_list(self, sql, from_type, from_handle):
# FIXME
return []
def get_attribute_list(self, sql, from_type, from_handle):
# FIXME
return []
def get_media_list(self, sql, from_type, from_handle):
# FIXME
return []
def get_sources(self, sql, handle):
results = sql.query("""select * from source where handle = ?;""",
handle)
return [self.pack_source(sql, result) for result in results]
def pack_source(self, sql, data):
(handle,
gid,
@ -167,10 +248,9 @@ class SQLReader(object):
marker1,
private) = data
# FIXME
note_list = self.get_links(sql, "source", handle, "note")
note_list = self.get_note_list(sql, "source", handle)
media_list = self.get_media_list(sql, "source", handle)
reporef_list = []
reporef_list = self.get_repository_ref_list(sql, "source", handle)
datamap = None
return (handle, gid, title,
@ -182,30 +262,26 @@ class SQLReader(object):
reporef_list,
(marker0, marker1), private)
def get_links(self, sql, from_type, from_handle, to_type):
def get_list(self, sql, from_type, from_handle, to_type):
results = sql.query("""select to_handle from link where from_type = ? and from_handle = ? and to_type = ?;""",
from_type, from_handle, to_type)
return [str(result) for result in results]
def get_names(self, sql, handle, primary):
result = sql.query("""select * from name where from_handle = ? and primary_name = ?;""",
names = sql.query("""select * from name where from_handle = ? and primary_name = ?;""",
handle, primary)
return result
def package_name(self, sql, handle, primary):
names = self.get_names(sql, handle, primary)
result = [self.pack(sql, name) for name in names]
result = [self.pack_name(sql, name) for name in names]
if primary:
if len(result) == 1:
return result[0]
elif len(result) == 0:
return None
else:
raise Exception("too many primary names")
print Exception("too many primary names")
else:
return result
def pack(self, sql, data):
def pack_name(self, sql, data):
# unpack name from SQL table:
(from_handle,
handle,
@ -223,15 +299,22 @@ class SQLReader(object):
sort_as,
display_as,
call) = data
# FIXME
source_list = self.get_links(sql, "name", from_handle, "source")
note_list = self.get_links(sql, "name", from_handle, "note")
source_list = self.get_source_list(sql, "name", from_handle)
note_list = self.get_note_list(sql, "name", from_handle)
date = self.get_date(sql, "name", from_handle)
return (private, source_list, note_list, date,
first_name, surname, suffix, title,
(name_type0, name_type1), prefix, patronymic,
group_as, sort_as, display_as, call)
def get_place(self, sql, from_type, handle):
row = self.get_list(sql, from_type, handle, "place")
if len(row) == 1:
return row[0]
else:
print AttributeError("invalid place '%s' '%s'" % (from_type, handle))
def get_date(self, sql, from_type, from_handle):
rows = sql.query("select * from date where from_type = ? and from_handle = ?;",
from_type, from_handle)
@ -257,7 +340,7 @@ class SQLReader(object):
elif len(rows) == 0:
return None
else:
raise Exception("ERROR, wrong number of dates: %s" % rows)
print Exception("ERROR, wrong number of dates: %s" % rows)
def process(self):
sql = self.openSQL()
@ -276,7 +359,6 @@ class SQLReader(object):
# ---------------------------------
# Process note
# ---------------------------------
# FIXME: 11 note objects were referenced but not found
notes = sql.query("""select * from note;""")
for note in notes:
(handle,
@ -321,32 +403,19 @@ class SQLReader(object):
marker1, # 18
private, # 19
) = person
# primary_name, # 3
primary_name = self.package_name(sql, handle, True)
# alternate_names, # 4
alternate_names = self.package_name(sql, handle, False) # list
# event_ref_list, # 7
event_ref_list = self.get_event_refs(sql, "person", handle)
# family_list, # 8
family_list = []
# parent_family_list, # 9
parent_family_list = []
# media_list, # 10
media_list = []
# address_list, # 11
address_list = []
# attribute_list, # 12
attribute_list = []
# urls, # 13
primary_name = self.get_names(sql, handle, True) # one
alternate_names = self.get_names(sql, handle, False) # list
event_ref_list = self.get_event_ref_list(sql, "person", handle)
family_list = self.get_family_list(sql, "person", handle)
parent_family_list = self.get_parent_family_list(sql, "person", handle)
media_list = self.get_media_list(sql, "person", handle)
address_list = self.get_address_list(sql, "person", handle)
attribute_list = self.get_attribute_list(sql, "person", handle)
urls = self.get_url_list(sql, "person", handle)
# lds_ord_list, # 14
lds_ord_list = []
# psource_list, # 15
psource_list = self.get_links(sql, "person", handle, "source")
# pnote_list, # 16
pnote_list = []
# person_ref_list, # 20
person_ref_list = []
lds_ord_list = self.get_lds_list(sql, "person", handle)
psource_list = self.get_source_list(sql, "person", handle)
pnote_list = self.get_note_list(sql, "person", handle)
person_ref_list = self.get_person_ref_list(sql, "person", handle)
death_ref_index = lookup(death_ref_handle, event_ref_list)
birth_ref_index = lookup(birth_ref_handle, event_ref_list)
self.db.person_map[str(handle)] = (str(handle), # 0
@ -388,14 +457,13 @@ class SQLReader(object):
marker1,
private) = event
# FIXME:
source_list = self.get_links(sql, "event", handle, "source")
note_list = self.get_links(sql, "event", handle, "note")
media_list = self.get_links(sql, "event", handle, "media")
attribute_list = self.get_links(sql, "event", handle, "attribute")
source_list = self.get_source_list(sql, "event", handle)
note_list = self.get_note_list(sql, "event", handle)
media_list = self.get_media_list(sql, "event", handle)
attribute_list = self.get_attribute_list(sql, "event", handle)
date = self.get_date(sql, "event", handle)
place = None # self.get_place()
place = self.get_place(sql, "event", handle)
data = (str(handle), gid, (the_type0, the_type1), date, description, place,
source_list, note_list, media_list, attribute_list,
@ -408,36 +476,155 @@ class SQLReader(object):
# ---------------------------------
# Process family
# ---------------------------------
people = sql.query("""select * from family;""")
for person in people:
families = sql.query("""select * from family;""")
for family in families:
(handle,
gid,
father_handle,
mother_handle,
the_type0,
the_type1,
change,
marker0,
marker1,
private) = family
child_ref_list = self.get_child_ref_list(sql, "family", handle)
event_ref_list = self.get_event_ref_list(sql, "family", handle)
media_list = self.get_media_list(sql, "family", handle)
attribute_list = self.get_attribute_list(sql, "family", handle)
lds_seal_list = self.get_lds_list(sql, "family", handle)
source_list = self.get_source_list(sql, "family", handle)
note_list = self.get_note_list(sql, "family", handle)
self.db.family_map[str(handle)] = (str(handle), gid,
father_handle, mother_handle,
child_ref_list, (the_type0, the_type1),
event_ref_list, media_list,
attribute_list, lds_seal_list,
source_list, note_list,
change, (marker0, marker1), private)
count += 1
self.callback(100 * count/total)
# ---------------------------------
# Process repository
# ---------------------------------
people = sql.query("""select * from repository;""")
for person in people:
repositories = sql.query("""select * from repository;""")
for repo in repositories:
(handle,
gid,
the_type0,
the_type1,
name,
change,
marker0,
marker1,
private) = repo
note_list = self.get_note_list(sql, "repository", handle)
address_list = self.get_address_list(sql, "repository", handle)
urls = self.get_url_list(sql, "repository", handle)
self.db.repository_map[str(handle)] = (str(handle), gid,
(the_type0, the_type1),
name, note_list,
address_list, urls, change,
(marker0, marker1), private)
count += 1
self.callback(100 * count/total)
# ---------------------------------
# Process place
# ---------------------------------
people = sql.query("""select * from place;""")
for person in people:
places = sql.query("""select * from place;""")
for place in places:
count += 1
(handle,
gid,
title,
long,
lat,
change,
marker0,
marker1,
private) = place
main_loc = None # FIXME
alt_location_list = self.get_location_list(sql, "place", handle)
urls = self.get_url_list(sql, "place", handle)
media_list = self.get_media_list(sql, "place", handle)
source_list = self.get_source_list(sql, "place", handle)
note_list = self.get_note_list(sql, "place", handle)
self.db.place_map[str(handle)] = (str(handle), gid, title, long, lat,
main_loc, alt_location_list,
urls,
media_list,
source_list,
note_list,
change, (marker0, marker1),
private)
self.callback(100 * count/total)
# ---------------------------------
# Process source
# ---------------------------------
people = sql.query("""select * from source;""")
for person in people:
sources = sql.query("""select * from source;""")
for source in sources:
(from_type,
handle,
gid,
title,
author,
pubinfo,
abbrev,
change,
marker0,
marker1,
private) = source
note_list = self.get_note_list(sql, "source", handle)
media_list = self.get_media_list(sql, "source", handle)
datamap = self.get_datamap(sql, "source", handle)
reporef_list = self.get_repository_ref_list(sql, "source", handle)
self.db.source_map[str(handle)] = (str(handle), gid, title,
author, pubinfo,
note_list,
media_list,
abbrev,
change, datamap,
reporef_list,
(marker0, marker1), private)
count += 1
self.callback(100 * count/total)
# ---------------------------------
# Process media
# ---------------------------------
people = sql.query("""select * from media;""")
for person in people:
media = sql.query("""select * from media;""")
for med in media:
(handle,
gid,
path,
mime,
desc,
change,
marker0,
marker1,
private) = med
attribute_list = self.get_attribute_list(sql, "media", handle)
source_list = self.get_source_list(sql, "media", handle)
note_list = self.get_note_list(sql, "media", handle)
date = self.get_date(sql, "media", handle)
self.db.media_map[str(handle)] = (str(handle), gid, path, mime, desc,
attribute_list,
source_list,
note_list,
change,
date,
(marker0, marker1),
private)
count += 1
self.callback(100 * count/total)