From 33e58d3e2b27670243c0cb06679fd078cd2e5e6c Mon Sep 17 00:00:00 2001 From: Alex Roitman Date: Mon, 19 Dec 2005 23:18:03 +0000 Subject: [PATCH] * src/RelLib.py: Add privacy flag to all objects via making PrimaryObject inherit from PrivacyBase; Add AddressBase and UrlBase classes to manage Address and Url lists in other objects; Add tuple type to the Url class; make Repository inherit from AddressBase and UrlBase to be able to store lists of both. * src/GrampsBSDDB.py (gramps_upgrade_9): Provide upgrade for the changes in RelLib. svn: r5585 --- ChangeLog | 8 + src/GrampsBSDDB.py | 45 ++-- src/RelLib.py | 535 +++++++++++++++++++++++---------------------- 3 files changed, 315 insertions(+), 273 deletions(-) diff --git a/ChangeLog b/ChangeLog index f9b37fdef..d23cefa17 100644 --- a/ChangeLog +++ b/ChangeLog @@ -3,6 +3,14 @@ Attribute objects to use tuple types. * src/RelLib.py (Name.set_type): Typo in the wrapper code. + * src/RelLib.py: Add privacy flag to all objects via making + PrimaryObject inherit from PrivacyBase; Add AddressBase and + UrlBase classes to manage Address and Url lists in other objects; + Add tuple type to the Url class; make Repository inherit from + AddressBase and UrlBase to be able to store lists of both. + * src/GrampsBSDDB.py (gramps_upgrade_9): Provide upgrade for the + changes in RelLib. + 2005-12-19 Richard Taylor * src/RelLib.py: added get_backlink_handles method to PrimaryObject * test/GrampsDbBase_Test.py: factored out common db test methods diff --git a/src/GrampsBSDDB.py b/src/GrampsBSDDB.py index ee0f69f6c..681e02ef2 100644 --- a/src/GrampsBSDDB.py +++ b/src/GrampsBSDDB.py @@ -1086,6 +1086,10 @@ class GrampsBSDDB(GrampsDbBase): # Cover attributes contained in MediaRefs for media_ref in person.media_list: convert_mediaref_9(media_ref) + + # In all Urls, add type attribute + for url in person.urls: + convert_url_9(url) self.commit_person(person,trans) #data = cursor.next() @@ -1123,7 +1127,7 @@ class GrampsBSDDB(GrampsDbBase): for attribute in family.attribute_list: convert_attribute_9(attribute) # Cover attributes contained in MediaRefs - for media_ref in person.media_list: + for media_ref in family.media_list: convert_mediaref_9(media_ref) self.commit_family(family,trans) @@ -1224,9 +1228,13 @@ class GrampsBSDDB(GrampsDbBase): place.source_list, place.note, place.change) = info # Cover attributes contained in MediaRefs - for media_ref in person.media_list: + for media_ref in place.media_list: convert_mediaref_9(media_ref) + # In all Urls, add type attribute + for url in place.urls: + convert_url_9(url) + self.commit_place(place,trans) # data = cursor.next() # cursor.close() @@ -1246,7 +1254,7 @@ class GrampsBSDDB(GrampsDbBase): media_object.date) = info # In all Attributes, convert type from string to a tuple - for attribute in family.attribute_list: + for attribute in media_object.attribute_list: convert_attribute_9(attribute) self.commit_media_object(media_object,trans) @@ -1256,19 +1264,20 @@ class GrampsBSDDB(GrampsDbBase): self.transaction_commit(trans,"Upgrade to DB version 9") print "Done upgrading to DB version 9" +_attribute_conversion_9 = { + "Caste" : (Attribute.CASTE,""), + "Description" : (Attribute.DESCRIPTION,""), + "Identification Number" : (Attribute.ID,""), + "National Origin" : (Attribute.NATIONAL,""), + "Number of Children" : (Attribute.NUM_CHILD,""), + "Social Security Number" : (Attribute.SSN,""), + } + def convert_attribute_9(attribute): - attribute_conversion = { - "Caste" : (Attribute.CASTE,""), - "Description" : (Attribute.DESCRIPTION,""), - "Identification Number" : (Attribute.ID,""), - "National Origin" : (Attribute.NATIONAL,""), - "Number of Children" : (Attribute.NUM_CHILD,""), - "Social Security Number" : (Attribute.SSN,""), - } old_type = attribute.type if old_type: - if attribute_conversion.has_key(old_type): - new_type = attribute_conversion[old_type] + if _attribute_conversion_9.has_key(old_type): + new_type = _attribute_conversion_9[old_type] else: new_type = (Attribute.CUSTOM,old_type) else: @@ -1279,6 +1288,16 @@ def convert_mediaref_9(media_ref): for attribute in media_ref.attribute_list: convert_attribute_9(attribute) +def convert_url_9(url): + path = url.path.strip() + if path.find('mailto:') == 0 or url.path.find('@') != -1: + url.type = (Url.EMAIL,'') + elif path.find('http://') == 0: + url.type = (Url.WEB_HOME,'') + elif path.find('ftp://') == 0: + url.type = (Url.WEB_FTP,'') + else: + url.type = (Url.CUSTOM,'') if __name__ == "__main__": diff --git a/src/RelLib.py b/src/RelLib.py index 2ab2da4bb..ce4d69c4a 100644 --- a/src/RelLib.py +++ b/src/RelLib.py @@ -188,7 +188,45 @@ class BaseObject: return ret -class PrimaryObject(BaseObject): +class PrivacyBase: + """ + Base class for privacy-aware objects. + """ + + def __init__(self,source=None): + """ + Initialize a PrivacyBase. If the source is not None, then object + is initialized from values of the source object. + + @param source: Object used to initialize the new object + @type source: PrivacyBase + """ + + if source: + self.private = source.private + else: + self.private = False + + def set_privacy(self,val): + """ + Sets or clears the privacy flag of the data + + @param val: value to assign to the privacy flag. True indicates that the + record is private, False indicates that it is public. + @type val: bool + """ + self.private = val + + def get_privacy(self): + """ + Returns the privacy level of the data. + + @returns: True indicates that the record is private + @rtype: bool + """ + return self.private + +class PrimaryObject(BaseObject,PrivacyBase): """ The PrimaryObject is the base class for all primary objects in the database. Primary objects are the core objects in the database. @@ -212,6 +250,7 @@ class PrimaryObject(BaseObject): @type source: PrimaryObject """ BaseObject.__init__(self) + PrivacyBase.__init__(self,source) if source: self.gramps_id = source.gramps_id self.handle = source.handle @@ -735,44 +774,6 @@ class DateBase: """ self.date = date -class PrivacyBase: - """ - Base class for privacy-aware objects. - """ - - def __init__(self,source=None): - """ - Initialize a PrivacyBase. If the source is not None, then object - is initialized from values of the source object. - - @param source: Object used to initialize the new object - @type source: PrivacyBase - """ - - if source: - self.private = source.private - else: - self.private = False - - def set_privacy(self,val): - """ - Sets or clears the privacy flag of the data - - @param val: value to assign to the privacy flag. True indicates that the - record is private, False indicates that it is public. - @type val: bool - """ - self.private = val - - def get_privacy(self): - """ - Returns the privacy level of the data. - - @returns: True indicates that the record is private - @rtype: bool - """ - return self.private - class AttributeBase: """ Base class for attribute-aware objects. @@ -840,6 +841,137 @@ class AttributeBase: """ self.attribute_list = attribute_list +class AddressBase: + """ + Base class for address-aware objects. + """ + + def __init__(self,source=None): + """ + Initialize a AddressBase. If the source is not None, then object + is initialized from values of the source object. + + @param source: Object used to initialize the new object + @type source: AddressBase + """ + + if source: + self.address_list = [ Address(address) \ + for address in source.address_list ] + else: + self.address_list = [] + + def add_address(self,address): + """ + Adds the L{Address} instance to the object's list of addresses + + @param address: L{Address} instance to add to the object's address list + @type address: list + """ + self.address_list.append(address) + + def remove_address(self,address): + """ + Removes the specified L{Address} instance from the address list + If the instance does not exist in the list, the operation has + no effect. + + @param address: L{Address} instance to remove from the list + @type address: L{Address} + + @return: True if the address was removed, False if it was not in the list. + @rtype: bool + """ + if address in self.address_list: + self.address_list.remove(address) + return True + else: + return False + + def get_address_list(self): + """ + Returns the list of L{Address} instances associated with the object + + @return: Returns the list of L{Address} instances + @rtype: list + """ + return self.address_list + + def set_address_list(self,address_list): + """ + Assigns the passed list to the object's list of L{Address} instances. + @param address_list: List of L{Address} instances to be associated + with the object + @type address_list: list + """ + self.address_list = address_list + +class UrlBase: + """ + Base class for url-aware objects. + """ + + def __init__(self,source=None): + """ + Initialize an UrlBase. If the source is not None, then object + is initialized from values of the source object. + + @param source: Object used to initialize the new object + @type source: UrlBase + """ + + if source: + self.urls = [ Url(url) for url in source.urls ] + else: + self.urls = [] + + def get_url_list(self): + """ + Returns the list of L{Url} instances associated with the object. + + @returns: List of L{Url} instances + @rtype: list + """ + return self.urls + + def set_url_list(self,url_list): + """ + Sets the list of L{Url} instances to passed the list. + + @param url_list: List of L{Url} instances + @type url_list: list + """ + self.urls = url_list + + def add_url(self,url): + """ + Adds a L{Url} instance to the object's list of L{Url} instances + + @param url: L{Url} instance to be added to the Person's list of + related web sites. + @type url: L{Url} + """ + self.urls.append(url) + + + def remove_url(self,url): + """ + Removes the specified L{Url} instance from the url list + If the instance does not exist in the list, the operation has + no effect. + + @param attribute: L{Url} instance to remove from the list + @type attribute: L{Url} + + @return: True if the url was removed, False if it was not in the list. + @rtype: bool + """ + if url in self.urls: + self.urls.remove(url) + return True + else: + return False + class PlaceBase: """ Base class for place-aware objects. @@ -896,7 +1028,8 @@ class PrivateSourceNote(SourceNote,PrivacyBase): # Actual GRAMPS objects # #------------------------------------------------------------------------- -class Person(PrimaryObject,PrivateSourceNote,MediaBase,AttributeBase): +class Person(PrimaryObject,SourceNote, + MediaBase,AttributeBase,AddressBase,UrlBase): """ Introduction ============ @@ -943,9 +1076,11 @@ class Person(PrimaryObject,PrivateSourceNote,MediaBase,AttributeBase): self.unserialize(data) else: PrimaryObject.__init__(self) - PrivateSourceNote.__init__(self) + SourceNote.__init__(self) MediaBase.__init__(self) AttributeBase.__init__(self) + AddressBase.__init__(self) + UrlBase.__init__(self) self.primary_name = Name() self.event_ref_list = [] self.family_list = [] @@ -955,8 +1090,6 @@ class Person(PrimaryObject,PrivateSourceNote,MediaBase,AttributeBase): self.gender = Person.UNKNOWN self.death_ref = None self.birth_ref = None - self.address_list = [] - self.urls = [] self.lds_bapt = None self.lds_endow = None self.lds_seal = None @@ -1189,34 +1322,6 @@ class Person(PrimaryObject,PrivateSourceNote,MediaBase,AttributeBase): """ self.alternate_names.append(name) - def get_url_list(self): - """ - Returns the list of L{Url} instances associated with the Person - - @returns: List of L{Url} instances - @rtype: list - """ - return self.urls - - def set_url_list(self,url_list): - """ - Sets the list of L{Url} instances to passed the list. - - @param url_list: List of L{Url} instances - @type url_list: list - """ - self.urls = url_list - - def add_url(self,url): - """ - Adds a L{Url} instance to the Person's list of L{Url} instances - - @param url: L{Url} instance to be added to the Person's list of - related web sites. - @type url: L{Url} - """ - self.urls.append(url) - def set_nick_name(self,name): """ Sets the nickname field for the Person @@ -1498,53 +1603,6 @@ class Person(PrimaryObject,PrivateSourceNote,MediaBase,AttributeBase): else: return False - def add_address(self,address): - """ - Adds the L{Address} instance to the Person's list of addresses - - @param address: L{Address} instance to add to the Person's address - list - @type address: list - """ - self.address_list.append(address) - - def remove_address(self,address): - """ - Removes the specified L{Address} instance from the address list - If the instance does not exist in the list, the operation has - no effect. - - @param address: L{Address} instance to remove from the list - @type address: L{Address} - - @return: True if the address was removed, False if it was not - in the list. - @rtype: bool - """ - if address in self.address_list: - self.address_list.remove(address) - return True - else: - return False - - def get_address_list(self): - """ - Returns the list of L{Address} instances associated with the - Person - @return: Returns the list of L{Address} instances - @rtype: list - """ - return self.address_list - - def set_address_list(self,address_list): - """ - Assigns the passed list to the Person's list of L{Address} instances. - @param address_list: List of L{Address} instances to ba associated - with the Person - @type address_list: list - """ - self.address_list = address_list - def get_parent_family_handle_list(self): """ Returns the list of L{Family} handles in which the person is a @@ -1822,7 +1880,7 @@ class Family(PrimaryObject,SourceNote,MediaBase,AttributeBase): self.child_list, self.type, self.event_ref_list, self.media_list, self.attribute_list, self.lds_seal, self.complete, self.source_list, self.note, - self.change, self.marker) + self.change, self.marker, self.private) def unserialize(self, data): """ @@ -1832,7 +1890,8 @@ class Family(PrimaryObject,SourceNote,MediaBase,AttributeBase): (self.handle, self.gramps_id, self.father_handle, self.mother_handle, self.child_list, self.type, self.event_ref_list, self.media_list, self.attribute_list, self.lds_seal, - self.complete, self.source_list, self.note, self.change, self.marker) = data + self.complete, self.source_list, self.note, self.change, + self.marker, self.private) = data def _has_handle_reference(self,classname,handle): if classname == 'Event': @@ -2181,7 +2240,7 @@ class Witness(BaseObject,PrivacyBase): def __init__(self): pass -class Event(PrimaryObject,PrivateSourceNote,MediaBase,DateBase,PlaceBase): +class Event(PrimaryObject,SourceNote,MediaBase,DateBase,PlaceBase): """ Introduction ============ @@ -2246,7 +2305,7 @@ class Event(PrimaryObject,PrivateSourceNote,MediaBase,DateBase,PlaceBase): """ PrimaryObject.__init__(self,source) - PrivateSourceNote.__init__(self,source) + SourceNote.__init__(self,source) MediaBase.__init__(self,source) DateBase.__init__(self,source) PlaceBase.__init__(self,source) @@ -2281,7 +2340,7 @@ class Event(PrimaryObject,PrivateSourceNote,MediaBase,DateBase,PlaceBase): return (self.handle, self.gramps_id, self.type, self.date, self.description, self.place, self.cause, self.private, self.source_list, self.note, self.media_list, self.change, - self.marker) + self.marker, self.private) def unserialize(self,data): """ @@ -2295,7 +2354,7 @@ class Event(PrimaryObject,PrivateSourceNote,MediaBase,DateBase,PlaceBase): (self.handle, self.gramps_id, self.type, self.date, self.description, self.place, self.cause, self.private, self.source_list, self.note, self.media_list, self.change, - self.marker) = data + self.marker, self.private) = data def _has_handle_reference(self,classname,handle): if classname == 'Place': @@ -2508,7 +2567,7 @@ class Event(PrimaryObject,PrivateSourceNote,MediaBase,DateBase,PlaceBase): def set_ext_witness_list(self,witness_list): self.ext_witness_list = witness_list -class Place(PrimaryObject,SourceNote,MediaBase): +class Place(PrimaryObject,SourceNote,MediaBase,UrlBase): """ Contains information related to a place, including multiple address information (since place names can change with time), longitude, latitude, @@ -2525,6 +2584,7 @@ class Place(PrimaryObject,SourceNote,MediaBase): PrimaryObject.__init__(self,source) SourceNote.__init__(self,source) MediaBase.__init__(self,source) + UrlBase.__init__(self,source) if source: self.long = source.long self.lat = source.lat @@ -2533,16 +2593,12 @@ class Place(PrimaryObject,SourceNote,MediaBase): self.alt_loc = [] for loc in source.alt_loc: self.alt_loc = Location(loc) - self.urls = [] - for u in source.urls: - self.urls.append(Url(u)) else: self.long = "" self.lat = "" self.title = "" self.main_loc = None self.alt_loc = [] - self.urls = [] def serialize(self): """ @@ -2562,7 +2618,8 @@ class Place(PrimaryObject,SourceNote,MediaBase): """ return (self.handle, self.gramps_id, self.title, self.long, self.lat, self.main_loc, self.alt_loc, self.urls, self.media_list, - self.source_list, self.note, self.change, self.marker) + self.source_list, self.note, self.change, self.marker, + self.private) def unserialize(self,data): """ @@ -2575,7 +2632,8 @@ class Place(PrimaryObject,SourceNote,MediaBase): """ (self.handle, self.gramps_id, self.title, self.long, self.lat, self.main_loc, self.alt_loc, self.urls, self.media_list, - self.source_list, self.note, self.change, self.marker) = data + self.source_list, self.note, self.change, self.marker, + self.private) = data def get_text_data_list(self): """ @@ -2618,34 +2676,6 @@ class Place(PrimaryObject,SourceNote,MediaBase): """ return self.media_list + self.source_list - def get_url_list(self): - """ - Returns the list of L{Url} instances associated with the Place - - @returns: List of L{Url} instances - @rtype: list - """ - return self.urls - - def set_url_list(self,url_list): - """ - Sets the list of L{Url} instances to passed the list. - - @param url_list: List of L{Url} instances - @type url_list: list - """ - self.urls = url_list - - def add_url(self,url): - """ - Adds a L{Url} instance to the Place's list of L{Url} instances - - @param url: L{Url} instance to be added to the Place's list of - related web sites. - @type url: L{Url} - """ - self.urls.append(url) - def set_title(self,title): """ Sets the descriptive title of the Place object @@ -2822,7 +2852,7 @@ class MediaObject(PrimaryObject,SourceNote,DateBase,AttributeBase): """ return (self.handle, self.gramps_id, self.path, self.mime, self.desc, self.attribute_list, self.source_list, self.note, - self.change, self.date, self.marker) + self.change, self.date, self.marker, self.private) def unserialize(self,data): """ @@ -2834,7 +2864,7 @@ class MediaObject(PrimaryObject,SourceNote,DateBase,AttributeBase): """ (self.handle, self.gramps_id, self.path, self.mime, self.desc, self.attribute_list, self.source_list, self.note, self.change, - self.date, self.marker) = data + self.date, self.marker, self.private) = data def get_text_data_list(self): """ @@ -2931,7 +2961,8 @@ class Source(PrimaryObject,MediaBase,NoteBase): return (self.handle, self.gramps_id, unicode(self.title), unicode(self.author), unicode(self.pubinfo), self.note, self.media_list, unicode(self.abbrev), - self.change,self.datamap,self.reporef_list, self.marker) + self.change,self.datamap,self.reporef_list, self.marker, + self.private) def unserialize(self,data): """ @@ -2941,7 +2972,7 @@ class Source(PrimaryObject,MediaBase,NoteBase): (self.handle, self.gramps_id, self.title, self.author, self.pubinfo, self.note, self.media_list, self.abbrev, self.change, self.datamap, self.reporef_list, - self.marker) = data + self.marker, self.private) = data def get_text_data_list(self): """ @@ -4082,6 +4113,13 @@ class Url(BaseObject,PrivacyBase): """Contains information related to internet Uniform Resource Locators, allowing gramps to store information about internet resources""" + UNKNOWN = -1 + CUSTOM = 0 + EMAIL = 1 + WEB_HOME = 2 + WEB_SEARCH = 3 + WEB_FTP = 4 + def __init__(self,source=None): """creates a new URL instance, copying from the source if present""" BaseObject.__init__(self) @@ -4089,9 +4127,11 @@ class Url(BaseObject,PrivacyBase): if source: self.path = source.path self.desc = source.desc + self.type = source.type else: self.path = "" self.desc = "" + self.type = (Url.CUSTOM,"") def get_text_data_list(self): """ @@ -4118,6 +4158,28 @@ class Url(BaseObject,PrivacyBase): """returns the description of the URL""" return self.desc + def set_type(self,the_type): + """ + @param type: descriptive type of the Url + @type type: str + """ + if not type(the_type) == tuple: + warn( "set_type now takes a tuple", DeprecationWarning, 2) + # Wrapper for old API + # remove when transitition done. + if the_type in range(-1,5): + the_type = (the_type,'') + else: + the_type = (Url.CUSTOM,the_type) + self.type = the_type + + def get_type(self): + """ + @returns: the descriptive type of the Url + @rtype: str + """ + return self.type + def are_equal(self,other): """returns 1 if the specified URL is the same as the instance""" if other == None: @@ -4126,6 +4188,8 @@ class Url(BaseObject,PrivacyBase): return 0 if self.desc != other.desc: return 0 + if self.type != other.type: + return 0 return 1 class SourceRef(BaseObject,DateBase,PrivacyBase,NoteBase): @@ -4495,7 +4559,7 @@ class RepoRef(BaseObject,NoteBase): def set_media_type(self,media_type): self.media_type = media_type -class Repository(PrimaryObject,NoteBase): +class Repository(PrimaryObject,NoteBase,AddressBase,UrlBase): """A location where collections of Sources are found""" UNKNOWN = -1 @@ -4514,29 +4578,23 @@ class Repository(PrimaryObject,NoteBase): """creates a new Repository instance""" PrimaryObject.__init__(self) NoteBase.__init__(self) + AddressBase.__init__(self) + UrlBase.__init__(self) self.type = (Repository.LIBRARY,"") self.name = "" - self.address = Location() #FIXME: This needs to become address - self.email = "" - self.search_url = "" - self.home_url = "" - self.note = Note() def serialize(self): - return (self.handle, self.gramps_id, self.type, - unicode(self.name), self.address, - unicode(self.email), - unicode(self.search_url), unicode(self.home_url), - self.note) + return (self.handle, self.gramps_id, self.type, unicode(self.name), + self.note, self.address_list, self.urls,self.marker, + self.private) def unserialize(self,data): """ Converts the data held in a tuple created by the serialize method back into the data in an Repository structure. """ - (self.handle, self.gramps_id, self.type, self.name, - self.address, self.email, - self.search_url, self.home_url, self.note) = data + (self.handle, self.gramps_id, self.type, self.name, self.note, + self.address_list, self.urls ,self.marker, self.private) = data def get_text_data_list(self): """ @@ -4545,7 +4603,7 @@ class Repository(PrimaryObject,NoteBase): @return: Returns the list of all textual attributes of the object. @rtype: list """ - return [self.name,self.type[1],self.email,self.search_url,self.home_url] + return [self.name,self.type[1]] def get_text_data_child_list(self): """ @@ -4554,7 +4612,7 @@ class Repository(PrimaryObject,NoteBase): @return: Returns the list of child objects that may carry textual data. @rtype: list """ - check_list = [self.address] + check_list = self.address_list + self.urls if self.note: check_list.append(self.note) return check_list @@ -4593,12 +4651,20 @@ class Repository(PrimaryObject,NoteBase): """ pass - def set_type(self,type): + def set_type(self,the_type): """ @param type: descriptive type of the Repository @type type: str """ - self.type = type + if not type(the_type) == tuple: + warn( "set_type now takes a tuple", DeprecationWarning, 2) + # Wrapper for old API + # remove when transitition done. + if the_type in range(-1,10): + the_type = (the_type,'') + else: + the_type = (Repository.CUSTOM,the_type) + self.type = the_type def get_type(self): """ @@ -4621,62 +4687,6 @@ class Repository(PrimaryObject,NoteBase): """ return self.name - def set_address(self,address): - """ - @param address: L{Location} instance to set as Repository's address - @type address: L{Location} - """ - self.address = address - - def get_address(self): - """ - @returns: L{Location} instance representing Repository's address - @rtype: L{Location} - """ - return self.address - - def set_email(self,email): - """ - @param email: descriptive email of the Repository - @type email: str - """ - self.email = email - - def get_email(self): - """ - @returns: the descriptive email of the Repository - @rtype: str - """ - return self.email - - def set_search_url(self,search_url): - """ - @param search_url: descriptive search_url of the Repository - @type search_url: str - """ - self.search_url = search_url - - def get_search_url(self): - """ - @returns: the descriptive search_url of the Repository - @rtype: str - """ - return self.search_url - - def set_home_url(self,home_url): - """ - @param home_url: descriptive home_url of the Repository - @type home_url: str - """ - self.home_url = home_url - - def get_home_url(self): - """ - @returns: the descriptive home_url of the Repository - @rtype: str - """ - return self.home_url - #------------------------------------------------------------------------- # @@ -4693,33 +4703,39 @@ if __name__ == "__main__": def test_simple(self): rep1 = Repository() - rep1.set_type("type") + rep1.set_type((Repository.LIBRARY,'')) rep1.set_name("name") - addr1 = Location() - rep1.set_address(addr1) - rep1.set_email("email") - rep1.set_search_url("search url") - rep1.set_home_url("home url") + addr1 = Address() + rep1.add_address(addr1) + email = Url() + email.set_type((Url.EMAIL,'')) + email.set_path('mailto:nobody@domain.com') + rep1.add_url(email) + home_url = Url() + home_url.set_type((Url.WEB_HOME,'')) + rep1.add_url(home_url) rep1.set_note("a note") - assert rep1.get_type() == "type" + assert rep1.get_type() == (Repository.LIBRARY,'') assert rep1.get_name() == "name" - assert rep1.get_address() == addr1 - assert rep1.get_email() == "email" - assert rep1.get_search_url() == "search url" - assert rep1.get_home_url() == "home url" + assert rep1.get_address_list() == [addr1] + assert rep1.get_url_list() == [email,home_url] assert rep1.get_note() == "a note" def test_serialize(self): rep1 = Repository() - addr1 = Location() - rep1.set_type("type") + rep1.set_type((Repository.LIBRARY,'')) rep1.set_name("name") - rep1.set_address(addr1) - rep1.set_email("email") - rep1.set_search_url("search url") - rep1.set_home_url("home url") + addr1 = Address() + rep1.add_address(addr1) + email = Url() + email.set_type((Url.EMAIL,'')) + email.set_path('mailto:nobody@domain.com') + rep1.add_url(email) + home_url = Url() + home_url.set_type((Url.WEB_HOME,'')) + rep1.add_url(home_url) rep1.set_note("a note") rep2 = Repository() @@ -4728,18 +4744,17 @@ if __name__ == "__main__": assert rep1.get_gramps_id() == rep2.get_gramps_id() assert rep1.get_type() == rep2.get_type() assert rep1.get_name() == rep2.get_name() - assert rep1.get_address() == rep2.get_address() - assert rep1.get_email() == rep2.get_email() - assert rep1.get_search_url() == rep2.get_search_url() - assert rep1.get_home_url() == rep2.get_home_url() + for idx in range(len(rep1.get_url_list())): + assert rep1.get_url_list()[idx] == rep2.get_url_list()[idx] + #assert rep1.get_address_list() == rep2.get_address_list() assert rep1.get_note() == rep2.get_note() def test_methods(self): rep1 = Repository() - addr1 = Location() + addr1 = Address() rep1.set_note("a note") - rep1.set_address(addr1) + rep1.add_address(addr1) assert type(rep1.get_text_data_list()) == type([]) assert rep1.get_text_data_child_list() == [addr1,rep1.note]