import peewee


database = peewee.SqliteDatabase(None)


class BddInsee:

    def __init__(self, chemin_base_donnees):
        database.init(chemin_base_donnees)

    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())))