Cyclone-Team/gramps
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

* src/plugins/TestcaseGenerator.py: Randomize more; Add LDS events

Browse Source 
svn: r5312
This commit is contained in:
Martin Hawlisch
2005-10-13 09:25:30 +00:00
parent fb6cc6e535
commit e9b39cf952
2 changed files with 163 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gramps2/ChangeLog
View File

@@ -1,3 +1,6 @@
2005-10-13 Martin Hawlisch <Martin.Hawlisch@gmx.de>
* src/plugins/TestcaseGenerator.py: Randomize more; Add LDS events
2005-10-12 Don Allingham <don@gramps-project.org> 2005-10-12 Don Allingham <don@gramps-project.org>
* src/plugins/NavWebPage.py: option for choosing the ancestor graph * src/plugins/NavWebPage.py: option for choosing the ancestor graph

198
gramps2/src/plugins/TestcaseGenerator.py
View File

@@ -71,7 +71,39 @@ class TestcaseGenerator(Tool.Tool):
self.persons_todo = [] self.persons_todo = []
self.parents_todo = [] self.parents_todo = []
self.person_dates = {} self.person_dates = {}
self.random_marrtype_list = (RelLib.Family.UNMARRIED, RelLib.Family.CIVIL_UNION,
RelLib.Family.UNKNOWN, RelLib.Family.OTHER)
self.random_childrel_list = (RelLib.Person.CHILD_REL_NONE, RelLib.Person.CHILD_REL_ADOPT,
RelLib.Person.CHILD_REL_STEP, RelLib.Person.CHILD_REL_SPONS,
RelLib.Person.CHILD_REL_FOST, RelLib.Person.CHILD_REL_UNKWN,
RelLib.Person.CHILD_REL_OTHER)
# If an active persons exists the generated tree is connected to that person
if person: if person:
# try to get birth and death year
try:
bh = person.get_birth_handle()
b = self.db.get_event_from_handle( bh)
do = b.get_date_object()
birth = do.get_year()
except AttributeError:
birth = None
try:
dh = person.get_death_handle()
b = self.db.get_event_from_handle( dh)
do = b.get_date_object()
death = do.get_year()
except AttributeError:
death = None
if not birth and not death:
birth = randint(1700,1900)
if birth and not death:
death = birth + randint(20,90)
if death and not birth:
birth = death - randint(20,90)
self.person_dates[person.get_handle()] = (birth,death)
self.persons_todo.append(person.get_handle()) self.persons_todo.append(person.get_handle())
self.parents_todo.append(person.get_handle()) self.parents_todo.append(person.get_handle())
@@ -704,6 +736,8 @@ class TestcaseGenerator(Tool.Tool):
self.commit_transaction() # COMMIT TRANSACTION STEP self.commit_transaction() # COMMIT TRANSACTION STEP
np = RelLib.Person() np = RelLib.Person()
np.set_privacy( randint(0,5) == 1)
np.set_complete_flag( randint(0,5) == 1)
self.add_defaults(np) self.add_defaults(np)
@@ -720,32 +754,11 @@ class TestcaseGenerator(Tool.Tool):
np.set_gender(gender) np.set_gender(gender)
# Name # Name
syllables1 = ["sa","li","na","ma","no","re","mi","cha","ki","du","ba","ku","el"]
syllables2 = ["as","il","an","am","on","er","im","ach","ik","ud","ab","ul","le"]
name = RelLib.Name() name = RelLib.Name()
firstname = "" name.set_privacy( randint(0,5) == 1)
maxnames = 5 (firstname,lastname) = self.rand_name(lastname, gender)
maxsyllables = 5 if alive_in_year:
if not self.options.handler.options_dict['long_names']: firstname = "%s (%d)" % (firstname, alive_in_year)
maxnames = 2
maxsyllables = 3
for i in range(0,randint(1,maxnames)):
for j in range(0,randint(2,maxsyllables)):
firstname = firstname + choice(syllables2)
if gender == RelLib.Person.FEMALE:
firstname = firstname + choice(("a","e","i","o","u"))
firstname = firstname + " "
firstname = firstname.title().strip()
if not lastname:
lastname = ""
for i in range(0,randint(2,maxsyllables)):
lastname = lastname + choice(syllables1)
n = randint(0,2)
if n == 0:
lastname = lastname.title()
elif n == 1:
lastname = lastname.upper()
name.set_first_name(firstname) name.set_first_name(firstname)
name.set_surname(lastname) name.set_surname(lastname)
name.add_source_reference(self.default_sourceref) name.add_source_reference(self.default_sourceref)
@@ -754,6 +767,7 @@ class TestcaseGenerator(Tool.Tool):
# generate some slightly different alternate name # generate some slightly different alternate name
alt_name = RelLib.Name(name) alt_name = RelLib.Name(name)
alt_name.set_privacy( randint(0,5) == 1)
firstname2 = firstname.replace("m", "n").replace("l", "i").replace("b", "d") firstname2 = firstname.replace("m", "n").replace("l", "i").replace("b", "d")
if firstname2 != firstname: if firstname2 != firstname:
alt_name.set_first_name( firstname2) alt_name.set_first_name( firstname2)
@@ -795,6 +809,17 @@ class TestcaseGenerator(Tool.Tool):
(bur_year, e) = self.rand_event( choice( ("Burial", "Cremation")), dy, dy+2) (bur_year, e) = self.rand_event( choice( ("Burial", "Cremation")), dy, dy+2)
self.db.add_event(e, self.trans) self.db.add_event(e, self.trans)
np.add_event_handle(e.get_handle()) np.add_event_handle(e.get_handle())
#LDS
if randint(0,1) == 1:
lds = self.rand_ldsord( const.lds_baptism)
np.set_lds_baptism( lds)
if randint(0,1) == 1:
lds = self.rand_ldsord( const.lds_baptism)
np.set_lds_endowment( lds)
if randint(0,1) == 1:
lds = self.rand_ldsord( const.lds_csealing)
np.set_lds_sealing( lds)
person_handle = self.db.add_person(np,self.trans) person_handle = self.db.add_person(np,self.trans)
@@ -833,7 +858,12 @@ class TestcaseGenerator(Tool.Tool):
self.add_defaults(fam) self.add_defaults(fam)
fam.set_father_handle(person1_h) fam.set_father_handle(person1_h)
fam.set_mother_handle(person2_h) fam.set_mother_handle(person2_h)
fam.set_relationship(RelLib.Family.MARRIED) if randint(0,2) == 1:
fam.set_relationship( choice( self.random_marrtype_list))
else:
fam.set_relationship( RelLib.Family.MARRIED)
lds = self.rand_ldsord( const.lds_ssealing)
fam.set_lds_sealing( lds)
fam_h = self.db.add_family(fam,self.trans) fam_h = self.db.add_family(fam,self.trans)
fam = self.db.commit_family(fam,self.trans) fam = self.db.commit_family(fam,self.trans)
person1 = self.db.get_person_from_handle(person1_h) person1 = self.db.get_person_from_handle(person1_h)
@@ -856,7 +886,13 @@ class TestcaseGenerator(Tool.Tool):
fam.add_child_handle(child_h) fam.add_child_handle(child_h)
self.db.commit_family(fam,self.trans) self.db.commit_family(fam,self.trans)
child = self.db.get_person_from_handle(child_h) child = self.db.get_person_from_handle(child_h)
child.add_parent_family_handle(fam_h,RelLib.Person.CHILD_REL_BIRTH,RelLib.Person.CHILD_REL_BIRTH) rel1 = RelLib.Person.CHILD_REL_BIRTH
if randint(0,2) == 1:
rel1 = choice( self.random_childrel_list)
rel2 = RelLib.Person.CHILD_REL_BIRTH
if randint(0,2) == 1:
rel2 = choice( self.random_childrel_list)
child.add_parent_family_handle(fam_h, rel1, rel2)
self.db.commit_person(child,self.trans) self.db.commit_person(child,self.trans)
if randint(0,3) > 0: if randint(0,3) > 0:
self.persons_todo.append(child_h) self.persons_todo.append(child_h)
@@ -868,7 +904,6 @@ class TestcaseGenerator(Tool.Tool):
return return
lastname = child.get_primary_name().get_surname() lastname = child.get_primary_name().get_surname()
if child_h in self.person_dates: if child_h in self.person_dates:
(born,died) = self.person_dates[child_h] (born,died) = self.person_dates[child_h]
person1_h = self.generate_person(1,lastname, alive_in_year=born) person1_h = self.generate_person(1,lastname, alive_in_year=born)
@@ -883,9 +918,15 @@ class TestcaseGenerator(Tool.Tool):
self.parents_todo.append(person2_h) self.parents_todo.append(person2_h)
fam = RelLib.Family() fam = RelLib.Family()
self.add_defaults(fam)
fam.set_father_handle(person1_h) fam.set_father_handle(person1_h)
fam.set_mother_handle(person2_h) fam.set_mother_handle(person2_h)
fam.set_relationship(RelLib.Family.MARRIED) if randint(0,2) == 1:
fam.set_relationship( choice( self.random_marrtype_list))
else:
fam.set_relationship( RelLib.Family.MARRIED)
lds = self.rand_ldsord( const.lds_ssealing)
fam.set_lds_sealing( lds)
fam.add_child_handle(child_h) fam.add_child_handle(child_h)
fam_h = self.db.add_family(fam,self.trans) fam_h = self.db.add_family(fam,self.trans)
fam = self.db.commit_family(fam,self.trans) fam = self.db.commit_family(fam,self.trans)
@@ -895,8 +936,13 @@ class TestcaseGenerator(Tool.Tool):
person2 = self.db.get_person_from_handle(person2_h) person2 = self.db.get_person_from_handle(person2_h)
person2.add_family_handle(fam_h) person2.add_family_handle(fam_h)
self.db.commit_person(person2,self.trans) self.db.commit_person(person2,self.trans)
rel1 = RelLib.Person.CHILD_REL_BIRTH
child.add_parent_family_handle(fam_h,RelLib.Person.CHILD_REL_BIRTH,RelLib.Person.CHILD_REL_BIRTH) if randint(0,2) == 1:
rel1 = choice( self.random_childrel_list)
rel2 = RelLib.Person.CHILD_REL_BIRTH
if randint(0,2) == 1:
rel2 = choice( self.random_childrel_list)
child.add_parent_family_handle(fam_h, rel1, rel2)
self.db.commit_person(child,self.trans) self.db.commit_person(child,self.trans)
self.commit_transaction() # COMMIT TRANSACTION STEP self.commit_transaction() # COMMIT TRANSACTION STEP
@@ -904,25 +950,67 @@ class TestcaseGenerator(Tool.Tool):
object.add_source_reference(self.default_sourceref) object.add_source_reference(self.default_sourceref)
object.add_media_reference(self.default_mediaref) object.add_media_reference(self.default_mediaref)
e = RelLib.Event() e = RelLib.Event()
e.set_privacy( randint(0,5) == 1)
e.set_name("TestcaseGenerator") e.set_name("TestcaseGenerator")
e.set_note("Generated by TestcaseGenerator") e.set_note("Generated by TestcaseGenerator")
e.set_cause("Test cause") e.set_cause("Test cause")
e.set_description("Test description") e.set_description("Test description")
e.add_source_reference(self.default_sourceref) sref = RelLib.SourceRef(self.default_sourceref)
e.add_media_reference(self.default_mediaref) sref.set_privacy( randint(0,5) == 1)
e.add_source_reference(sref)
mref = RelLib.MediaRef(self.default_mediaref)
mref.set_privacy( randint(0,5) == 1)
e.add_media_reference(mref)
e.set_place_handle(self.default_place.get_handle()) e.set_place_handle(self.default_place.get_handle())
(year,d) = self.rand_date() (year,d) = self.rand_date()
e.set_date_object( d) e.set_date_object( d)
self.db.add_event(e, self.trans) self.db.add_event(e, self.trans)
object.add_event_handle(e.get_handle()) object.add_event_handle(e.get_handle())
object.add_attribute( self.default_attribute) attr = RelLib.Attribute(self.default_attribute)
attr.set_privacy( randint(0,5) == 1)
object.add_attribute( attr)
try: try:
object.add_url(self.default_url) url = RelLib.Url(self.default_url)
object.add_address(self.default_address) url.set_privacy( randint(0,5) == 1)
object.add_url(url)
addr = RelLib.Address(self.default_address)
addr.set_privacy( randint(0,5) == 1)
object.add_address(addr)
except AttributeError: except AttributeError:
pass # family does not have an url pass # family does not have an url
object.set_note("Generated by TestcaseGenerator") object.set_note("Generated by TestcaseGenerator")
def rand_name( self, lastname=None, gender=None):
syllables1 = ["sa","li","na","ma","no","re","mi","cha","ki","du","ba","ku","el"]
syllables2 = ["as","il","an","am","on","er","im","ach","ik","ud","ab","ul","le"]
firstname = ""
if not gender:
gender = choice((RelLib.Person.MALE, RelLib.Person.FEMALE))
maxnames = 5
maxsyllables = 5
if not self.options.handler.options_dict['long_names']:
maxnames = 2
maxsyllables = 3
for i in range(0,randint(1,maxnames)):
for j in range(0,randint(2,maxsyllables)):
firstname = firstname + choice(syllables2)
if gender == RelLib.Person.FEMALE:
firstname = firstname + choice(("a","e","i","o","u"))
firstname = firstname + " "
firstname = firstname.title().strip()
if not lastname:
lastname = ""
for i in range(0,randint(2,maxsyllables)):
lastname = lastname + choice(syllables1)
n = randint(0,2)
if n == 0:
lastname = lastname.title()
elif n == 1:
lastname = lastname.upper()
return (firstname,lastname)
def rand_date( self, start=None, end=None): def rand_date( self, start=None, end=None):
""" """
Generates a random date object between the given years start and end Generates a random date object between the given years start and end
@@ -981,15 +1069,49 @@ class TestcaseGenerator(Tool.Tool):
if randint(0,1) == 1: if randint(0,1) == 1:
e.set_description("Test description") e.set_description("Test description")
if randint(0,1) == 1: if randint(0,1) == 1:
e.add_source_reference(self.default_sourceref) sref = self.default_sourceref
sref.set_privacy( randint(0,5) == 1)
e.add_source_reference( sref)
if randint(0,1) == 1: if randint(0,1) == 1:
e.add_media_reference(self.default_mediaref) mref = self.default_mediaref
mref.set_privacy( randint(0,5) == 1)
e.add_media_reference( mref)
if randint(0,1) == 1: if randint(0,1) == 1:
e.set_place_handle(self.default_place.get_handle()) e.set_place_handle(self.default_place.get_handle())
(year, d) = self.rand_date( start, end) (year, d) = self.rand_date( start, end)
e.set_date_object( d) e.set_date_object( d)
if randint(0,5) == 1:
w = RelLib.Witness()
w.set_privacy( randint(0,5) == 1)
w.set_comment("Generated by TestcaseGenerator")
if randint(0,1) == 1:
w.set_type( RelLib.Event.ID)
wph = self.generate_person( alive_in_year=year)
w.set_value( wph)
else:
w.set_type( RelLib.Event.NAME)
w.set_value( "%s %s" % self.rand_name())
e.add_witness(w)
return (year, e) return (year, e)
def rand_ldsord( self, status_list):
lds = RelLib.LdsOrd()
if randint(0,1) == 1:
lds.set_status( randint(0,len(status_list)-1))
if randint(0,1) == 1:
lds.set_temple( choice( const.lds_temple_to_abrev.keys()))
if randint(0,1) == 1:
lds.set_place_handle( self.default_place.get_handle())
if randint(0,1) == 1:
sref = self.default_sourceref
sref.set_privacy( randint(0,5) == 1)
lds.add_source_reference( sref)
if randint(0,1) == 1:
lds.set_note("Generated by TestcaseGenerator")
if randint(0,1) == 1:
(year,d) = self.rand_date( )
lds.set_date_object( d)
return lds
def commit_transaction(self): def commit_transaction(self):