import squish
import test
import object
from squish import *
import names
from leahi_dialin.ui.td_messaging import TD_Messaging
from leahi_dialin.common.td_defs import TDOpModes,TDTreatmentStates

td =TD_Messaging()


ACID_CONCENTRATE            = ["1.0 K, 2.50 Ca, 1 Mg","2.0 K, 2.50 Ca, 1 Mg","3.0 K, 2.50 Ca, 1 Mg"]
DIALYZER_TYPE               = ["Diacap Pro 13H","Diacap Pro 16H", "Diacap Pro 19H", "Optiflux F160NRe", "Optiflux F180NRe", "Optiflux F200NRe","Optiflux F250NRe"]
VITALS                      = ["OFF","5","10","15","20","30","60"]
BICARBONATE                 = ["Sodium Bicarbonate"]
   
def get_object_from_names(names_dict, error_message = "Missing object", timeout_ms = 200):
    """
        To get an object with try..except catching to prevent script errors when the object is not found on the GUI
        @param names_dict - the dictionary element from the names.py file (ie: names.some_variable_name_of_element)
        @returns the object with corresponding dictionary, otherwise "None"
    """     
    try:
        return squish.waitForObject(names_dict, timeout_ms)
    except LookupError:
        test.fail("ERROR : " + error_message)
        return None   

def setObjectText(obj, text):
    obj["text"] = text
    return obj

def findObjectById(parent, id):
    """
    Recursively searches for a child object by its id.
    Returns the found object or None if not found.
    """
    if str(parent.id) == id:
        return parent

    for child in object.children(parent):
        found = findObjectById(child, id)
        if found:
            return found
    return None

def findChildByText(parent_object, target_text):
    """Recursively finds a child object by its text property."""
    for child in object.children(parent_object):
        if hasattr(child, "text") and str(child.text) == target_text:
            return child
        found = findChildByText(child, target_text)
        if found:
            return found
        
def set_value_based_on_target(obj, target_value):
    """
    obj: dictionary containing object paths
         Example:
         {
             "value_obj": ":mainTreatmentScreen.PressureText",
             "left_arrow": ":mainTreatmentScreen.LeftArrow",
             "right_arrow": ":mainTreatmentScreen.RightArrow"
         }
 
    target_value: integer or string number, e.g. 220
    """
    target_value = target_value
 
    # Wait for all objects
    parent_obj = squish.waitForObjectExists(obj)
    # change range as per your screen count
    
    left_arrow = findObjectById(parent_obj, "_leftArrow")
    right_arrow =findObjectById(parent_obj, "_rightArrow")
 
    # Read current value (supports invisible text too)
    try:
        current_value = round(float(findObject(obj).value),1)
    except LookupError:
        current_value = float(findObject(obj).property("value"))
 
    # Determine direction
    while current_value != float(target_value):
        if current_value < float(target_value):
            squish.mouseClick(squish.waitForObject(right_arrow))

        elif current_value > float(target_value):
            squish.mouseClick(squish.waitForObject(left_arrow))
        # Update current value after click
        try:
            current_value = round(float(findObject(obj).value),1)
        except Exception:
            current_value = float(findObject(obj).property("value"))
 
        test.log(f"Updated value: {current_value}")
 
    test.log(f"✅ Target value reached: {current_value}")
    
def select_different_dropdown(object,type,whichTypeIndex):
    type_combo_box = get_object_from_names(object, error_message="Combo box object is missing")
    if type_combo_box is not None: 
        squish.mouseClick(squish.waitForObjectExists(object))
    
        type_option = get_object_from_names(setObjectText(obj = names.o_option_combo_box,text = type[whichTypeIndex]),error_message=f"Option {type[whichTypeIndex]} object is missing",timeout_ms=5000)
        if type_option is not None:
            squish.mouseClick(type_option)
            return True
    return False # default return if not successful

def verify_create_treatment_parameters():
    test.startSection("Pre treatment parameters")
    Button = setObjectText(names.o_preTreatmentStack_Text, "Next")
    
    for i in range(4):
        squish.mouseClick(waitForObject(Button))
    squish.mouseClick(waitForObject(names.o_PreTreatmentCreate_pretreatmentPatientIDEntry_TextEntry))
    squish.waitForObject(names.o_PreTreatmentCreate_pretreatmentPatientIDEntry_TextEntry).text ="abcd"
    set_value_based_on_target(names.o_PreTreatmentCreate_bloodFlowRateControl_ValueAdjuster, 60)
    set_value_based_on_target(names.o_PreTreatmentCreate_dialysateFlowRateControl_ValueAdjuster, 75)
    set_value_based_on_target(names.o_PreTreatmentCreate_durationControl_ValueAdjuster, 75)
    set_value_based_on_target(names.o_PreTreatmentCreate_heparinBolusVolumeControl_ValueAdjuster, 0.4)
    set_value_based_on_target(names.o_PreTreatmentCreate_heparinDispensingRateControl_ValueAdjuster, 0.5)
    set_value_based_on_target(names.o_PreTreatmentCreate_heparinStopTimeControl_ValueAdjuster, 60)
    select_different_dropdown(names.o_PreTreatmentCreate_acidConcentrateComboBox_BaseComboBox,ACID_CONCENTRATE,2)
    set_value_based_on_target(names.o_PreTreatmentCreate_dialysateTemperatureControl_ValueAdjuster,37.0)
    select_different_dropdown(names.o_PreTreatmentCreate_dialyzerTypeComboBox_BaseComboBox,DIALYZER_TYPE,2)
    set_value_based_on_target(names.o_PreTreatmentCreate_salineBolusVolumeControl_ValueAdjuster, 200)
    select_different_dropdown(names.o_PreTreatment_vitalsCombobox_BaseCombobox,VITALS,1)
    select_different_dropdown(names.o_PreTreatmentCreate_bicarbonateConcentrateComboBox_BaseComboBox,BICARBONATE,0)
    Validatebutton = setObjectText(names.o_preTreatmentStack_Text, "VALIDATE")
    squish.mouseClick(waitForObject(Validatebutton))
    td.td_Treatment_Parameters_Validation(  vAccepted                                    = 1,
                                            vBloodFlowRateRejectReason                   = 0,
                                            vDialysateFlowRateRejectReason               = 0,
                                            vTreatmentDurationRejectReason               = 0,
                                            vSalineBolusVolumeRejectReason               = 0,
                                            vHeparinStopTimeRejectReason                 = 0,
                                            vHeparinTypeRejectReason                     = 0,
                                            vAcidConcentrateRejectReason                 = 0,
                                            vBicarbonateConcentrateRejectReason          = 0,
                                            vDialyzerTypeRejectReason                    = 0,
                                            vBloodPressureMeasureIntervalRejectReason    = 0,
                                            vRinsebackFlowRateRejectReason               = 0,
                                            vRinsebackVolumeRejectReason                 = 0,
                                            vArterialPressureLimitWindowRejectReason     = 0,
                                            vVenousPressureLimitWindowRejectReason       = 0,
                                            vVenousPressureLimitAsymtrcRejectReason      = 0,
                                            vTransmembranePressureLimitWindowRejectReason  = 0,
                                            vDialysateTempRejectReason                   = 0,
                                            vHeparinDispensingRateRejectReason           = 0,
                                            vHeparinBolusVolumeRejectReason              = 0
                                           )
    
    confirmButton = setObjectText(names.o_preTreatmentStack_Text, "CONFIRM")
    squish.mouseClick(waitForObject(confirmButton))
    td.td_operation_mode(TDOpModes.MODE_PRET.value, 0)
    test.endSection()