lambda_function

import time
import os
import logging
import requests

"""--- logger create log object which stores all infor---"""
logger = logging.getLogger()
logger.setLevel(logging.DEBUG)

""""---Django url ---"""


def by_nric_name(patientNRIC, patientName):
    if patientNRIC is None or patientName is None:
        return False
    patientName = patientName.capitalize()
    patientNRIC = patientNRIC.upper()
    params = {'patientNRIC': patientNRIC, 'patientName': patientName}
    django_url = "https://1f01434759ce.ngrok.io/bynricandname/"
    if requests.get(django_url, params=params):
        return True
    else:
        return False


def by_name_code(patientName, patientAddressCode):
    if patientName is None or patientAddressCode is None:
        return False
    patientsName = patientName.capitalize()
    params = {'patientName': patientsName, 'patientAddress': patientAddressCode}
    django_url = "https://1f01434759ce.ngrok.io/bynameandaddress/"
    if requests.get(django_url, params=params):
        return True
    else:
        return False


def by_nric_code(patientNRIC, patientAddressCode):
    # if patientsNRIC is None or patientsAddressCode is None:
    #     print("Both nric and code not none")
    patientNRIC = patientNRIC.upper()
    params = {'patientNRIC': patientNRIC, 'patientAddress': patientAddressCode}
    django_url = "https://1f01434759ce.ngrok.io/bynricandaddress/"
    if requests.get(django_url, params=params):
        return True
    else:
        return False


''''---  ---'''


def get_slots(intent_request):
    return intent_request['currentIntent']['slots']


def elicit_slot(session_attributes, intent_name, slots, slot_to_elicit, message):
    return {
        'sessionAttributes': session_attributes,
        'dialogAction': {
            'type': 'ElicitSlot',
            'intentName': intent_name,
            'slots': slots,
            'slotToElicit': slot_to_elicit,
            'message': message
        }
    }


def close(session_attributes, fulfillment_state, message):
    response = {
        'sessionAttributes': session_attributes,
        'dialogAction': {
            'type': 'Close',
            'fulfillmentState': fulfillment_state,
            'message': message
        }
    }

    return response


"""--- Helper function---"""


def build_validation_result(is_valid, violated_slot, message_content):
    if message_content is None:
        return {
            "isValid": is_valid,
            "violatedSlot": violated_slot,
        }

    return {
        'isValid': is_valid,
        'violatedSlot': violated_slot,
        'message': {'contentType': 'PlainText', 'content': message_content}
    }


def validate_nric(patientNRIC):
    prefixList = ["S", "T", "F", "G"]
    if patientNRIC:
        patientNRIC = patientNRIC.upper()
        if (len(patientNRIC) == 9) and (patientNRIC[0] in prefixList) and (patientNRIC[8].isalpha()):
            return build_validation_result(True, None, None)

        else:
            return build_validation_result(False, 'patientNRIC',
                                           'Please make sure your NRIC entered is 9 digits, the first letter should should contain one of the alphabets in (STFG)'
                                           'and end with alphabet.')


def validate_code(patientAddressCode):
    # if patientAddressCode:
    #     return build_validation_result(True,None,None)
    if len(patientAddressCode) == 6:
        return build_validation_result(True, None, None)
    else:
        return build_validation_result(False, 'patientAddressCode', 'Enter a valid postal code, it should be 6 digits')



def validate_name(patientName):
    if patientName:
        return build_validation_result(True, None, None)
    else:
        return build_validation_result(False, 'patientName', 'Please make sure you have entered a correct name')


def authentication(intent_request):
    slots = get_slots(intent_request)
    patientName = slots["patientName"]
    patientNRIC = slots["patientNRIC"]
    patientAddressCode = slots["patientAddressCode"]

    output_session_attributes = intent_request['sessionAttributes'] if intent_request[
                                                                           'sessionAttributes'] is not None else {}
    if "retryCount" in output_session_attributes:
        retryCount = int(output_session_attributes["retryCount"])
    else:
        retryCount = 0
    source = intent_request['invocationSource']

    if source == 'DialogCodeHook':
        if not patientNRIC:
            return elicit_slot(intent_request['sessionAttributes'],
                               intent_request['currentIntent']['name'],
                               slots,
                               'patientNRIC',
                               {"contentType": "PlainText", "content": "Please enter your NRIC, thank you."})
        nric_validation_result = validate_nric(patientNRIC)
        if not nric_validation_result['isValid']:
            retryCount += 1
            output_session_attributes["retryCount"] = "{}".format(retryCount)
            slots[nric_validation_result["violatedSlot"]] = None
            if (retryCount < 3):
                return elicit_slot(intent_request['sessionAttributes'],
                                   intent_request['currentIntent']['name'],
                                   slots,
                                   nric_validation_result["violatedSlot"],
                                   nric_validation_result["message"])
            else:
                return close(intent_request['sessionAttributes'],
                             'Fulfilled',
                             {'contentType': 'PlainText',
                              'content': 'Sorry you have exceeded the number of attempts.Please try again. Goodbye'})

        if not patientName:
            return elicit_slot(intent_request['sessionAttributes'],
                               intent_request['currentIntent']['name'],
                               slots,
                               'patientName',
                               {"contentType": "PlainText", "content": "Please enter your Name, thank you."})
        name_validation_result = validate_name(patientName)
        if not name_validation_result['isValid']:
            slots[name_validation_result["violatedSlot"]] = None
            return elicit_slot(intent_request['sessionAttributes'],
                               intent_request['currentIntent']['name'],
                               slots,
                               name_validation_result["violatedSlot"],
                               name_validation_result["message"])

        if not patientAddressCode:
            return elicit_slot(intent_request['sessionAttributes'],
                               intent_request['currentIntent']['name'],
                               slots,
                               'patientAddressCode',
                               {"contentType": "PlainText",
                                "content": "Please enter your postal code and it should be 6 digits."})
        code_validation_result = validate_code(patientAddressCode)
        if not code_validation_result['isValid']:
            slots[code_validation_result["violatedSlot"]] = None
            return elicit_slot(intent_request['sessionAttributes'],
                               intent_request['currentIntent']['name'],
                               slots,
                               code_validation_result["violatedSlot"],
                               code_validation_result["message"])

        if 'validation' not in output_session_attributes.keys():
            check_nric_name = by_nric_name(patientNRIC, patientName)
            check_nric_postalcode = by_nric_code(patientNRIC, patientAddressCode)
            check_name_postalcode = by_name_code(patientName, patientAddressCode)
            checks = [check_nric_name, check_nric_postalcode, check_name_postalcode]
            if any(checks):
                output_session_attributes['validation'] = 'success'
                if check_nric_name:
                    patientNRIC = patientNRIC.upper()
                    patientName = patientName.capitalize()
                    params = {'patientNRIC': patientNRIC, 'patientName': patientName}
                    django_url = "https://1f01434759ce.ngrok.io/bynricandname/"
                    patient_data = requests.get(django_url, params=params).json()
                    name = patient_data[0]['patientName']
                    allergy = patient_data[0]['patientAllergies']
                    medication = patient_data[0]['patientMedicationName']
                    dosage = patient_data[0]['dosage']
                    sideEffect = patient_data[0]['sideEffects']
                    return close(intent_request['sessionAttributes'],
                                 'Fulfilled',
                                 {'contentType': 'PlainText',
                                  'content': 'Hi {},your credentials are verified. Your medication prescribed is {}, the dosage is {}. Be reminded that'
                                             'you are allergy to {} and the side effect for the medication is {}'.format(
                                      name, medication,dosage,allergy, sideEffect)})

                if check_nric_postalcode:
                    patientNRIC = patientNRIC.upper()
                    params = {'patientNRIC': patientNRIC, 'patientAddress': patientAddressCode}
                    django_url = "https://1f01434759ce.ngrok.io/bynricandaddress/"
                    patient_data = requests.get(django_url, params=params).json()
                    name = patient_data[0]['patientName']
                    allergy = patient_data[0]['patientAllergies']
                    medication =patient_data[0]['patientMedicationName']
                    dosage =patient_data[0]['dosage']
                    sideEffect =patient_data[0]['sideEffects']
                    return close(intent_request['sessionAttributes'],
                                 'Fulfilled',
                                 {'contentType': 'PlainText',
                                  'content': 'Hi {},your credentials are verified. Your medication prescribed is {}, the dosage is {}. Be reminded that'
                                             'you are allergy to {} and the side effect for the medication is {}'.format(
                                      name, medication,dosage,allergy, sideEffect)})

                if check_name_postalcode:
                    patientName = patientName.capitalize()
                    params = {'patientName': patientName, 'patientAddress': patientAddressCode}
                    django_url = "https://1f01434759ce.ngrok.io/bynameandaddress/"
                    patient_data = requests.get(django_url, params=params).json()
                    name = patient_data[0]['patientName']
                    allergy = patient_data[0]['patientAllergies']
                    medication = patient_data[0]['patientMedicationName']
                    dosage = patient_data[0]['dosage']
                    sideEffect = patient_data[0]['sideEffects']
                    return close(intent_request['sessionAttributes'],
                                 'Fulfilled',
                                 {'contentType': 'PlainText',
                                  'content': 'Hi {},your credentials are verified. Your medication prescribed is {}, the dosage is {}. Be reminded that'
                                             'you are allergy to {} and the side effect for the medication is {}'.format(
                                     name, medication,dosage,allergy, sideEffect)})

            else:
                #seting slot values to none for re-entering
                patientNRIC = None
                slots["patientName"] = None
                slots["patientNRIC"] = None
                slots["patientAddressCode"] = None

        if (patientNRIC is None) and ('validation' not in output_session_attributes.keys()):
            return elicit_slot(intent_request['sessionAttributes'],
                               intent_request['currentIntent']['name'],
                               slots,
                               'patientNRIC',
                               {"contentType": "PlainText", "content": "Records don't match. Please enter your NRIC."})

    """ --- Main handler --- """


def lambda_handler(event, context):
    """
    Route the incoming request based on intent.
    The JSON body of the request is provided in the event slot.
    """
    # By default, treat the user request as coming from the America/New_York time zone.
    os.environ["TZ"] = "Asia/Singapore"
    time.tzset()
    logger.debug('event.bot.name={}'.format(event['bot']['name']))
    logger.debug(event)
    return dispatch(event)


def dispatch(intent_request):
    """
    Called when the user specifies an intent for this bot.
    """

    logger.debug(
        'dispatch userId={}, intentName={}'.format(intent_request['userId'], intent_request['currentIntent']['name']))

    intent_name = intent_request['currentIntent']['name']

    # Dispatch to your bot's intent handlers
    if intent_name == 'Authentication':
        return authentication(intent_request)

    raise Exception('Intent with name ' + intent_name + ' not supported')