import os
from peewee import Model, CharField, BigAutoField, DateTimeField, IntegerField, fn, Field, Expression, TextField, ForeignKeyField, DateField
from playhouse.sqlite_ext import CSqliteExtDatabase, FTS5Model, AutoIncrementField, SearchField, RowIDField

var_dir = os.path.join(os.path.abspath(os.environ['HOME']), '.eldoc')
db_dir = os.path.join(var_dir, 'db')
db_file = os.path.join(db_dir, 'eldoc.sqlite3')

if not os.path.exists(var_dir):
    os.mkdir(var_dir)
if not os.path.exists(db_dir):
    os.mkdir(db_dir)

db = CSqliteExtDatabase(db_file, pragmas={
    'journal_mode': 'wal',
    'cache_size': -1 * 64000,  # 64MB
    'foreign_keys': 'on',
    'ignore_check_constraints': 'off',
    'synchronous': 'off'})

class BaseModel(Model):
    'Базовая таблица'
    class Meta:
        database = db
class BaseFTSModel(FTS5Model):
    class Meta:
        database = db
        options = {'tokenize': 'porter'}

class Patient(BaseModel):
    id = AutoIncrementField()
    last_name = TextField()
    first_name = TextField()
    middle_name = TextField(null=True)
    birth_date = DateField()
    gender = CharField(6)
    doc_type = CharField(32, null=True)
    doc_serial = TextField(null=True)
    doc_number = TextField(null=True)
    policy_number = TextField(null=True)
    policy_company = TextField(null=True)
    snils_number = TextField(null=True)
    notes = TextField(null=True)
Patient.add_index(
    Patient.index(Patient.last_name, Patient.first_name, Patient.birth_date, unique=True)
    )
class Reception(BaseModel):
    id = AutoIncrementField()
    patient = ForeignKeyField(Patient, backref='receptions', on_delete='CASCADE')
    time = DateTimeField()
class Diagnosis(BaseModel):
    id = AutoIncrementField()
    code = CharField()
    title = TextField()
    description = TextField(null=True)
class ReceptionDiagnosis(BaseModel):
    id = AutoIncrementField()
    reception = ForeignKeyField(Reception, backref='reception_diagnosisses', on_delete='CASCADE')
    diagnosis = ForeignKeyField(Diagnosis, backref='reception_diagnosisses', on_delete='CASCADE')
class ReceptionAnamnesis(BaseModel):
    id = AutoIncrementField()
    reception = ForeignKeyField(Reception, backref='reception_anamnesisses', on_delete='CASCADE')
    text = TextField()
class AnamnesisTemplate(BaseModel):
    id = AutoIncrementField()
    text = TextField()
class PatientIndex(BaseFTSModel):
    rowid = RowIDField()
    fio = SearchField()

def search_patients(q):
    with db.atomic():
        return (Patient.select()
                .join(
                    PatientIndex,
                    on=(Patient.id == PatientIndex.rowid))
                .where(PatientIndex.match(q))
                .order_by(PatientIndex.bm25()))
def store_patient_index(pat):
    fio_list = [pat.last_name, pat.first_name]
    if pat.middle_name:
        fio_list.append(pat.middle_name)
    PatientIndex.insert(
        {
            PatientIndex.rowid: pat.id,
            PatientIndex.fio: ' '.join(fio_list)
            }
        ).execute()
def update_patient_index(pat):
    fio_list = [pat.last_name, pat.first_name]
    if pat.middle_name:
        fio_list.append(pat.middle_name)
    PatientIndex.update(
        {
            PatientIndex.fio: ' '.join(fio_list)
            }
        ).where(PatientIndex.rowid == pat.id).execute()

db.connect()
db.create_tables([
    Patient,
    PatientIndex
    ])