variable_descriptions.py

from pydantic import BaseModel, Field
from typing import Optional
from enum import Enum

var_description = dict()


# -------------  enums  -----------------

class SurgeryType(Enum):
    none = "not scheduled for surgery"
    minor = "minor surgery"
    major = "major surgery (not lower extremity)"
    major_lower_extremity = "major lower extremity surgery"


class Mobility(Enum):
    ambulatory = "normal/ambulatory"
    bedrest = "bedrest or only walking in room"
    confined = "confined to bed >72 hours"


class Sex(Enum):
    male = "M"
    female = "F"

var_description['success'] = Field(
    default=False,
    title='Success',
    description='Describes whether the calculation was successful or not.'
)
var_description['message'] = Field(
    default="message",
    title='Result message',
    description='Summarizes the results of the calculation'
)
var_description['additional_info'] = Field(
    default="info",
    title='Result information',
    description='Summarizes important information about the calculation'
)

var_description['inr'] = Field(
    default=1.0,
    title='International Normalized Ratio',
    description='This is the patient\'s INR value and should be between 0 and 10',
    gt=0,
    lt=10
)
var_description['is_on_dialysis'] = Field(
    default=False,
    title='Is the patient on dialysis?',
    description='Dialysis at least twice in the past seven days or 24-hours of continuous hemodialysis within the '
                'prior 7 days.'
)
var_description['bilirubin'] = Field(
    default=1.0,
    title='Serum bilirubin',
    description='This is the patient\'s total bilirubin value in mg/dL and should be between 0 and 10',
    gt=0,
    lt=10
)
var_description['creatinine'] = Field(
    default=1.0,
    title='Serum creatinine',
    description='This is the patient\'s serum creatinine in mg/dL and should be between 0 and 10',
    gt=0,
    lt=10
)
var_description['sodium'] = Field(
    default=137.0,
    title='Serum sodium',
    description='Also known as \'Na.\' This is the patient\'s serum sodium in mg/dL and should be in a range between '
                '110 and 160',
    gt=110,
    lt=160
)

# caprini ------------------------------------------------------

var_description['age'] = Field(
    default=55,
    title="Age",
    description="Age in years",
    gt=18,
    lt=120
)
var_description['sex'] = Field(
    default=Sex.male,
    title="Sex",
    description="Recognized sex of the patient, M or F"
)
var_description['type_of_surgery'] = Field(
    default=SurgeryType.none,
    title="Type of surgery",
    description="Minor surgery does not refer to type of surgery but rather length of anesthesia <45 minutes. Major "
                "surgery refers to procedures with general or regional anesthesia time exceeding 45 minutes. Major"
                "lower extremity surgery refers to procedures anesthesia time exceeding 45 minutes and involving "
                "the hip, knee, thigh, or ankle. The only accepted inputs are: 'not scheduled for surgery', 'minor "
                "surgery', 'major lower extremity surgery', or 'major surgery (not lower extremity)'."
)
var_description['recent_major_surgery'] = Field(
    default=False,
    title="Recent major surgery",
    description="In last 30 days, has patient had major surgery, defined as anesthesia time >45 minutes?",
    points=1
)
var_description['recent_chf'] = Field(
    default=False,
    title="Recent congestive heart failure",
    description="In last 30 days, has patient had heart failure?",
    points=1
)
var_description['recent_sepsis'] = Field(
    default=False,
    title="Recent sepsis",
    description="In last 30 days, has patient had sepsis, been hospitalized for an infection, or required IV "
                "antibiotics. Do not count pneumonia under this category?",
    points=1
)
var_description['recent_pna'] = Field(
    default=False,
    title="Recent pneumonia",
    description="In last 30 days, has patient had pneumonia or recieved IV antibiotics for a lung infection?",
    points=1
)
var_description['recent_preg'] = Field(
    default=False,
    title="Recent pregnancy or postpartum",
    description="In last 30 days, has patient been pregnant or postpartum?",
    points=1
)

var_description['recent_cast'] = Field(
    default=False,
    title="Recent immobilizing plaster cast",
    description="In last 30 days, has patient had an immobilizing plaster cast?",
    points=2
)
var_description['recent_fracture'] = Field(
    default=False,
    title="Recent hip, pelvis, or leg fracture",
    description="In last 30 days, has patient had a hip, pelvis, or leg fracture?",
    points=5
)
var_description['recent_stroke'] = Field(
    default=False,
    title="Recent stroke",
    description="In last 30 days, has patient had stroke?",
    points=5
)
var_description['recent_trauma'] = Field(
    default=False,
    title="Recent multiple trauma",
    description="In last 30 days, has patient had multiple trauma?",
    points=5
)
var_description['recent_spinal_cord_injury'] = Field(
    default=False,
    title="Recent acute spinal cord injury causing paralysis",
    description="In last 30 days, has patient had a acute spinal cord injury causing paralysis?",
    points=5
)
var_description['varicose_veins'] = Field(
    default=False,
    title="Varicose veins",
    description="Does the patient have varicose veins?",
    points=1
)
var_description['current_swollen_legs'] = Field(
    default=False,
    title="Current swollen legs",
    description="Does the patient have a current swollen legs?",
    points=1
)
var_description['current_central_venous_access'] = Field(
    default=False,
    title="Current central venous access",
    description="Does the patient have a current central venous access?",
    points=2
)
var_description['history_of_dvt_pe'] = Field(
    default=False,
    title="History of DVT/PE",
    description="Does the patient have a history of DVT/PE?",
    points=3
)
var_description['family_history_of_thrombosis'] = Field(
    default=False,
    title="Family history of thrombosis",
    description="Does the patient have a family history of thrombosis?",
    points=3
)
var_description['positive_factor_v_leiden'] = Field(
    default=False,
    title="Positive Factor V Leiden",
    description="Does the patient have a positive Factor V Leiden?",
    points=3
)
var_description['positive_prothrombin_20210a'] = Field(
    default=False,
    title="Positive prothrombin 20210A",
    description="Does the patient have a positive prothrombin 20210A?",
    points=3
)
var_description['elevated_serum_homocysteine'] = Field(
    default=False,
    title="Elevated serum homocysteine",
    description="Does the patient have a elevated serum homocysteine?",
    points=3
)
var_description['positive_lupus_anticoagulant'] = Field(
    default=False,
    title="Positive lupus anticoagulant",
    description="Does the patient have a positive lupus anticoagulant?",
    points=3
)
var_description['elevated_anticardiolipin_antibody'] = Field(
    default=False,
    title="Elevated anticardiolipin antibody",
    description="Does the patient have a elevated anticardiolipin antibody?",
    points=3
)
var_description['heparin_induced_thrombocytopenia'] = Field(
    default=False,
    title="Heparin-induced thrombocytopenia",
    description="Does the patient have heparin-induced thrombocytopenia?",
    points=3
)
var_description['other_congenital_or_acquired_thrombophilia'] = Field(
    default=False,
    title="Other congenital or acquired thrombophilia",
    description="Does the patient have any other congenital or acquired thrombophilia?",
    points=3
)
var_description['mobility'] = Field(
    default=Mobility.ambulatory,
    title="Mobility",
    description="There are three categories of mobility: 'normal/ambulatory', 'bedrest or only walking in room',"
                " or 'confined to bed >72 hours'."
)
var_description['history_of_inflammatory_bowel_disease'] = Field(
    default=False,
    title="History of inflammatory bowel disease",
    description="Does the patient have a history of inflammatory bowel disease?",
    points=1
)
var_description['bmi'] = Field(
    default=22,
    title="BMI",
    description="Body mass index (BMI) of patient in kg/m2?"
)
var_description['acute_mi'] = Field(
    default=False,
    title="Acute MI",
    description="Does the patient have an acute MI?",
    points=1
)
var_description['copd'] = Field(
    default=False,
    title="COPD",
    description="Does the patient have COPD?",
    points=1
)
var_description['present_or_previous_malignancy'] = Field(
    default=False,
    title="Present or previous malignancy",
    description="Does the patient have a present or previous malignancy?",
    points=1
)
var_description['other_risk_factors'] = Field(
    default=False,
    title="Other risk factors",
    description="Does the patient have any other risk factors for VTE?",
    points=1
)

# wells dvt ------------------------------------------------------

var_description['active_cancer'] = Field(
    default=False,
    title="Active cancer",
    description="Does the patient have active cancer (treatment or palliation within 6 months)?"
)
var_description['bedridden_recently'] = Field(
    default=False,
    title="Bedridden recently",
    description="Has the patient been bedridden recently (greater than 3 days) or had major surgery within 12 weeks?"
)
var_description['calf_swelling'] = Field(
    default=False,
    title="Calf swelling",
    description="Does the patient have calf swelling greater than 3 cm compared to the other leg? (measured 10 cm below tibial tuberosity)"
)
var_description['collateral_veins'] = Field(
    default=False,
    title="Collateral veins",
    description="Does the patient have collateral (nonvaricose) superficial veins?"
)
var_description['entire_leg_swollen'] = Field(
    default=False,
    title="Entire leg swollen",
    description="Does that patient have an entire leg which is swollen?"
)
var_description['localized_tenderness'] = Field(
    default=False,
    title="Localized tenderness",
    description="Does the patient have localized tenderness along a deep venous system?"
)
var_description['pitting_edema'] = Field(
    default=False,
    title="Pitting edema",
    description="Does the patient have pitting edema, confined to symptomatic leg?"
)
var_description['paralysis_paresis_or_plaster'] = Field(
    default=False,
    title="Paralysis, paresis, or plaster immobilization",
    description="Does the patient have paralysis, paresis, or recent plaster immobilization of the lower extremity?"
)
var_description['previous_dvt'] = Field(
    default=False,
    title="Previously documented DVT",
    description="Does the patient have previously-documented DVT?"
)
var_description['alternative_diagnosis_as_likely'] = Field(
    default=False,
    title="Alternative diagnosis to DVT",
    description="Does the patient have alternative diagnosis to DVT as likely or more likely?"
)

# PSI / PORT ------------------------------------------------------
var_description['nursing_home_resident'] = Field(
    default=False,
    title="Nursing home resident",
    description="Is the patient a nursing home resident?"
)
var_description['neoplastic_disease'] = Field(
    default=False,
    title="Neoplastic disease",
    description="Does the patient have neoplastic disease?"
)
var_description['liver_disease'] = Field(
    default=False,
    title="Liver disease",
    description="Does the patient have liver disease?"
)
var_description['congestive_heart_failure'] = Field(
    default=False,
    title="Congestive heart failure",
    description="Does the patient have congestive heart failure?"
)
var_description['cerebrovascular_disease'] = Field(
    default=False,
    title="Cerebrovascular disease",
    description="Does the patient have cerebrovascular disease?"
)
var_description['renal_disease'] = Field(
    default=False,
    title="Renal disease",
    description="Does the patient have renal disease?"
)
var_description['altered_mental_status'] = Field(
    default=False,
    title="Altered mental status",
    description="Does the patient have altered mental status?"
)
var_description['respiratory_rate'] = Field(
    default=20,
    title="Respiratory rate",
    description="Respiratory rate in breaths per minute",
    gt=0,
    lt=100
)
var_description['systolic_bp'] = Field(
    default=90,
    title="Systolic blood pressure",
    description="Systolic blood pressure in mmHg",
    gt=0,
    lt=300
)
var_description['temperature'] = Field(
    default=37.8,
    title="Temperature",
    description="Temperature in degrees Celsius",
    gt=0,
    lt=50
)
var_description['pulse'] = Field(
    default=125,
    title="Pulse",
    description="Pulse in beats per minute",
    gt=0,
    lt=300
)
var_description['ph'] = Field(
    default=7.35,
    title="pH",
    description="pH",
    gt=0,
    lt=10
)
var_description['glucose'] = Field(
    default=250,
    title="Glucose",
    description="Glucose in mg/dL",
    gt=0,
    lt=1000
)

var_description['bun'] = Field(
    default=1,
    title="BUN",
    description="Blood urea nitrogen in mg/dL",
    gt=0,
    lt=100
)
var_description['sodium_mmol_L'] = Field(
    default=136.0,
    title='Serum sodium in mmol/L',
    description='Also known as \'Na.\' This is the patient\'s serum sodium in mmol/L and should be in a range between '
                '136 and 145 mmol/L',
)
var_description['hematocrit'] = Field(
    default=30,
    title="Hematocrit",
    description="Hematocrit in %",
    gt=0,
    lt=100
)
var_description['pao2'] = Field(
    default=60,
    title="PaO2",
    description="PaO2 in mmHg",
    gt=0,
    lt=1000
)
var_description['pleural_effusion'] = Field(
    default=False,
    title="Pleural effusion",
    description="Does the patient have a pleural effusion?"
)

# -------------  response and request models  -----------------


# basic response model
class CalcResponse(BaseModel):
    success: bool = var_description['success']
    score: Optional[int]
    message: Optional[str] = var_description['message']
    additional_info: Optional[str] = var_description['additional_info']


# original meld
class CalcRequestMeld(BaseModel):
    is_on_dialysis: bool = var_description['is_on_dialysis']
    creatinine: float = var_description['creatinine']
    bilirubin: float = var_description['bilirubin']
    inr: float = var_description['inr']


# meld sodium
class CalcRequestMeldNa(BaseModel):
    is_on_dialysis: bool = var_description['is_on_dialysis']
    creatinine: float = var_description['creatinine']
    bilirubin: float = var_description['bilirubin']
    inr: float = var_description['inr']
    sodium: float = var_description['sodium']


class CalcRequestCapriniVte(BaseModel):
    age: int = var_description['age']
    sex: Sex = var_description['sex']
    type_of_surgery: SurgeryType = var_description['type_of_surgery']
    recent_major_surgery: bool = var_description['recent_major_surgery']
    recent_chf: bool = var_description['recent_chf']
    recent_sepsis: bool = var_description['recent_sepsis']
    recent_pna: bool = var_description['recent_pna']
    recent_preg: bool = var_description['recent_preg']
    recent_cast: bool = var_description['recent_cast']
    recent_fracture: bool = var_description['recent_fracture']
    recent_stroke: bool = var_description['recent_stroke']
    recent_trauma: bool = var_description['recent_trauma']
    recent_spinal_cord_injury: bool = var_description['recent_spinal_cord_injury']
    varicose_veins: bool = var_description['varicose_veins']
    current_swollen_legs: bool = var_description['current_swollen_legs']
    current_central_venous_access: bool = var_description['current_central_venous_access']
    history_of_dvt_pe: bool = var_description['history_of_dvt_pe']
    family_history_of_thrombosis: bool = var_description['family_history_of_thrombosis']
    positive_factor_v_leiden: bool = var_description['positive_factor_v_leiden']
    positive_prothrombin_20210a: bool = var_description['positive_prothrombin_20210a']
    elevated_serum_homocysteine: bool = var_description['elevated_serum_homocysteine']
    positive_lupus_anticoagulant: bool = var_description['positive_lupus_anticoagulant']
    elevated_anticardiolipin_antibody: bool = var_description['elevated_anticardiolipin_antibody']
    heparin_induced_thrombocytopenia: bool = var_description['heparin_induced_thrombocytopenia']
    other_congenital_or_acquired_thrombophilia: bool = var_description['other_congenital_or_acquired_thrombophilia']
    mobility: Mobility = var_description['mobility']
    history_of_inflammatory_bowel_disease: bool = var_description['history_of_inflammatory_bowel_disease']
    bmi: int = var_description['bmi']
    acute_mi: bool = var_description['acute_mi']
    copd: bool = var_description['copd']
    present_or_previous_malignancy: bool = var_description['present_or_previous_malignancy']
    other_risk_factors: bool = var_description['other_risk_factors']


# -------------  wells DVT -----------------
class CalcRequestWellsDvt(BaseModel):
    active_cancer: bool = var_description['active_cancer']
    bedridden_recently: bool = var_description['bedridden_recently']
    calf_swelling: bool = var_description['calf_swelling']
    collateral_veins: bool = var_description['collateral_veins']
    entire_leg_swollen: bool = var_description['entire_leg_swollen']
    localized_tenderness: bool = var_description['localized_tenderness']
    pitting_edema: bool = var_description['pitting_edema']
    paralysis_paresis_or_plaster: bool = var_description['paralysis_paresis_or_plaster']
    previous_dvt: bool = var_description['previous_dvt']
    alternative_diagnosis_as_likely: bool = var_description['alternative_diagnosis_as_likely']

# -------------  PSI/PORT  -----------------
class CalcRequestPsiPort(BaseModel):
    age: int = var_description['age']
    sex: Sex = var_description['sex']
    nursing_home_resident: bool = var_description['nursing_home_resident']
    neoplastic_disease: bool = var_description['neoplastic_disease']
    liver_disease: bool = var_description['liver_disease']
    congestive_heart_failure: bool = var_description['congestive_heart_failure']
    cerebrovascular_disease: bool = var_description['cerebrovascular_disease']
    renal_disease: bool = var_description['renal_disease']
    altered_mental_status: bool = var_description['altered_mental_status']
    respiratory_rate: int = var_description['respiratory_rate']
    systolic_bp: int = var_description['systolic_bp']
    temperature: float = var_description['temperature']
    pulse: int = var_description['pulse']
    ph: float = var_description['ph']
    bun: int = var_description['bun']
    sodium_mmol_L: int = var_description['sodium_mmol_L']
    glucose: int = var_description['glucose']
    hematocrit: int = var_description['hematocrit']
    pao2: int = var_description['pao2']
    pleural_effusion: int = var_description['pleural_effusion']

if __name__ == "__main__":
    print("you've reached this message in error")