"""Class definitions used in ehrkit.py"""

from datetime import date

class Patient:

    """Patient with biographical and medical history data

    Attributes:

        id (int): unique patient ID

        sex (str): patient sex ('M' or 'F')

        dob (datetime): date of birth

        dod (datetime): date of death

        alive (bool): True if patient alive, else False

    """

    def __init__(self, data):

        # Biographical Data:

        if "id" in data:  #int

            self.id = data["id"] #should we use pid in case of conflict with python keyword?

        else:

            self.id = None

        if "sex" in data:

            self.sex = data["sex"]

        else:

            self.sex = None

        if "dob" in data:

            self.dob = data["dob"]

        else:

            self.dob = None

        if "dod" in data:

            self.dod = data["dod"]

        else:

            self.dod = None

        if "alive" in data:

            self.alive = data["alive"]

        else:

            self.alive = None

        ### MEDICAL DATA: ###

        # Current patient diagnoses (Array of diagnosis objects)

        if "diagnosis" in data:

            self.diagnosis = data["diagnosis"]

        else:

            self.diagnosis = None

        # Prescriptions patient is currently taking (Array of prescription objects)

        if "current_prescriptions" in data:

            self.prescriptions = data["current_prescriptions"]

        else:

            self.prescriptions = None

        # Primary care provider (Foreign Key on Provider object):

        if "primary_provider" in data:

            self.primary_provider = data["primary_provider"]

        else:

            self.primary_provider = None

        ### NOTEEVENTS data(list of nltk separated sentences):###

        if "Text" in data:

            self.note_events = data["Text"]

        else:

            self.note_events = None

        ### MEDICAL HISTORY: ###

        # 1.) Past prescriptions (Array of prescription objects)

        if "past_prescriptions" in data:

            self.prescriptHistory = data["past_prescriptions"]

        else:

            self.prescriptHistory = None

        # 2.) Past diagnoses (Array of diagnosis objects)

        if "past_diagnoses" in data:

            self.diagHistory = data["past_diagnoses"]

        else:

            self.diagHistory = None

        # 3) Procedures (Array of procedure objects)

        if "procedures" in data:

            self.procedures = data["procedures"]

        else:

            self.procedures = None

    def addNE(self, textlist):

        """List of all note events for patient, in tuple format (docID, content).

        """

        self.note_events = textlist

    def addPrescriptions(self, textlist):

        self.prescriptions = textlist

    def diagnose(self, disease, start = None, end = None):

        """Adds diagnosis to Patient history:"""

        data = {"name": disease, "id": None, "start": start, "end": end}

        newDiag = Diagnosis(data)

        if not self.diagnosis: self.diagnosis = []

        self.diagnosis.append(newDiag)

    def add_procedure(self, procedure):

        """Adds procedure to Patient history"""

        data = {"name": procedure}

        new_proc = Procedure(data)

        if not self.procedures: self.procedures = []

        self.procedures.append(new_proc)

    def age(self):

        """Compute patient age using birthdate"""

        today = date.today()

        birthdate = self.dob

        return today.year - birthdate.year - ((today.month, today.day) < (birthdate.month, birthdate.day))

# Disease Object

class Disease:

    def __init__(self, data):

        self.name = data["name"]

        # Foreign key for diagnosis object

        self.id = data["id"]

        # Symptoms — array of strings

        self.symptoms = data["symptoms"]

        self.abbreviation = data["abbreviations"]

# Diagnosis Object

# - Foreign Key on Disease Object

class Diagnosis:

    def __init__(self, data):

        # Foreign key on disease object

        self.name = data["name"]

        # Date of diagnosis

        self.start = data["start"]

        # End date, if patient cured:

        self.end = data["end"]

# Prescription object

class Prescription:

    def __init__(self, data):

        # Generic or Brand Name or Compound Name:

        self.name = data["name"]

        # Is there a standardized drug ID?

        # This is important b/c patient could have

        # been prescribed a generic

        self.id = data["id"]

        #Foreign key on patient object

        self.patient_id = data["patient_id"]

        self.indication = data["indication"]

        # How often medicine is taken:

        self.freq = data["frequency"]

        #Unit of time for frequency (daily, weekly, monthly)

        self.freq = data["frequency_unit"]

        # Numerical quantity of dose — float

        self.dosage = data["dosage"]

        # Units of dosage (e.g. mg, ml, etc.) — str

        self.units = data["dosage_units"]

        self.start = data["start_date"]

        self.end = data["end_date"]

        # Not sure if need to record drug manufacturer — str

        self.manufacturer = data["manufacturer"]

        # Foreign key on Provider() object for prescriber:

        self.prescriber_id = data["prescriber_id"]

    # Length of prescription in days:

    def duration(self):

        if self.start == None and self.end == None:

            return None

        if self.end == None:

            end = date.today()

        else:

            end = self.end

        duration = end - self.start

        return duration.days

# Unsure about this class; is it too generic for our data?

class Procedure:

    def __init__(self, data):

        # Can be a foreign key on charge list,

        # enabling us to capture procedure cost data

        self.name = data["name"]

# Medical facility class:

class Facility:

    def __init__(self, data):

        # Unique facility ID:

        self.id = data["id"]

        self.name = data["name"]

        self.state = data["state"]

        self.city = data["city"]

        self.address = data["address"]

        # For instance "hospital", "out-patient clinic", etc.

        self.type = data["type"]

# Healthcare provider class:

class Provider:

    def __init__(self, data):

        # Unique physician ID:

        self.id = data["id"]

        # Primary facility for physician, foreign key on Facility objects

        self.facility_id = data["facility_id"]

        self.first_name = data["first_name"]

        self.last_name = data["last_name"]

        # Provider type: RN, EMT, MD, etc.

        self.type = data["type"]

        # Provider title:

        self.title = data["title"]