from datetime import datetime
from typing import Optional

from apps.common.schemas import BaseAuditModel
from core.uid_gen import FelicityIDType
from pydantic import EmailStr

#
#  Patient Schema
#

# Shared properties


class PatientBase(BaseAuditModel):
    client_patient_id: Optional[str] = None
    client_uid: Optional[FelicityIDType] = None
    patient_id: Optional[str] = None
    first_name: Optional[str] = None
    middle_name: Optional[str] = None
    last_name: Optional[str] = None
    gender: Optional[str] = None
    age: Optional[int] = None
    date_of_birth: Optional[datetime] = None
    age_dob_estimated: Optional[bool] = None
    phone_mobile: Optional[str] = None
    phone_home: Optional[str] = None
    consent_sms: Optional[bool] = None
    email: Optional[EmailStr] = None
    internal_use: Optional[bool] = False
    active: Optional[bool] = None
    district_uid: Optional[FelicityIDType] = None
    province_uid: Optional[FelicityIDType] = None
    country_uid: Optional[FelicityIDType] = None


# Properties to receive via API on creation
class PatientCreate(PatientBase):
    client_patient_id: str
    first_name: str
    last_name: str
    client_uid: FelicityIDType
    active: bool = True


# Properties to receive via API on update
class PatientUpdate(PatientBase):
    pass


class PatientInDBBase(PatientBase):
    uid: Optional[FelicityIDType] = None

    class Config:
        orm_mode = True


# Additional properties to return via API
class Patient(PatientInDBBase):
    pass


# Additional properties stored in DB
class PatientInDB(PatientInDBBase):
    pass


#
#  Identification Schema
#

# Shared properties


class IdentificationBase(BaseAuditModel):
    name: str


# Properties to receive via API on creation
class IdentificationCreate(IdentificationBase):
    pass


# Properties to receive via API on update
class IdentificationUpdate(IdentificationBase):
    pass


class IdentificationInDBBase(IdentificationBase):
    uid: Optional[FelicityIDType] = None

    class Config:
        orm_mode = True


# Additional properties to return via API
class Identification(IdentificationInDBBase):
    pass


# Additional properties stored in DB
class IdentificationInDB(IdentificationInDBBase):
    pass


#
#  PatientIdentification Schema
#

# Shared properties


class PatientIdentificationBase(BaseAuditModel):
    patient_uid: str
    identification_uid: str
    value: str


# Properties to receive via API on creation
class PatientIdentificationCreate(PatientIdentificationBase):
    pass


# Properties to receive via API on update
class PatientIdentificationUpdate(PatientIdentificationBase):
    pass


class PatientIdentificationInDBBase(PatientIdentificationBase):
    uid: Optional[FelicityIDType] = None

    class Config:
        orm_mode = True


# Additional properties to return via API
class PatientIdentification(PatientIdentificationInDBBase):
    pass


# Additional properties stored in DB
class PatientIdentificationInDB(PatientIdentificationInDBBase):
    pass