54 lines
2.4 KiB
Python
Executable File
54 lines
2.4 KiB
Python
Executable File
import os
|
|
import peewee
|
|
|
|
|
|
database = peewee.SqliteDatabase(None)
|
|
|
|
|
|
class BddInsee:
|
|
|
|
def __init__(self, chemin_base_donnees):
|
|
new = not os.path.isfile(chemin_base_donnees)
|
|
database.init(chemin_base_donnees)
|
|
if new:
|
|
database.create_tables([self.Person])
|
|
|
|
class Person(peewee.Model):
|
|
last_name = peewee.CharField()
|
|
first_name = peewee.CharField()
|
|
is_woman = peewee.BooleanField()
|
|
annee_naissance = peewee.IntegerField()
|
|
mois_naissance = peewee.IntegerField()
|
|
jour_naissance = peewee.IntegerField()
|
|
code_lieu_naissance = peewee.CharField()
|
|
commune_naissance = peewee.CharField()
|
|
pays_naissance = peewee.CharField()
|
|
annee_deces = peewee.IntegerField()
|
|
mois_deces = peewee.IntegerField()
|
|
jour_deces = peewee.IntegerField()
|
|
code_lieu_deces = peewee.CharField()
|
|
numero_act_deces = peewee.CharField()
|
|
|
|
class Meta:
|
|
database = database
|
|
|
|
def find_person(self, first_name, last_name, maiden_name, annee_naissance, mois_naissance, jour_naissance):
|
|
if maiden_name:
|
|
return self.Person.select().where((self.Person.annee_naissance == int(annee_naissance))
|
|
& (self.Person.mois_naissance == int(mois_naissance))
|
|
& (self.Person.jour_naissance == int(jour_naissance))
|
|
& (self.Person.first_name.contains(first_name.upper()))
|
|
& ((self.Person.last_name.contains(last_name.upper()))
|
|
| self.Person.last_name.contains(maiden_name.upper())))
|
|
else:
|
|
return self.Person.select().where((self.Person.annee_naissance == int(annee_naissance))
|
|
& (self.Person.mois_naissance == int(mois_naissance))
|
|
& (self.Person.jour_naissance == int(jour_naissance))
|
|
& (self.Person.first_name.contains(first_name.upper()))
|
|
& (self.Person.last_name.contains(last_name.upper())))
|
|
|
|
def import_data_list(self, data_list):
|
|
with database.atomic():
|
|
for batch in peewee.chunked(data_list, 70):
|
|
self.Person.insert_many(batch).execute()
|