main.py

import helper
import random
import config
import sys
import json


class SymptomChecker:

    # initialize w/ username and password
    def __init__(self):
        username = config.username
        password = config.password
        authUrl = config.priaid_authservice_url
        healthUrl = config.priaid_healthservice_url
        language = config.language
        self._printRawOutput = config.pritnRawOutput

        self._diagnosisClient = helper.Helper(username, password, authUrl, language, healthUrl)

    # starts here
    def start(self):
        # load in bodyLocations and SubLocations
        bodyLocations = self._diagnosisClient.loadBodyLocations()
        bodySubLocations = self.getAllSubLocations(bodyLocations)



        # get user input
        text = input("Where are you experiencing pain?\n")
        self.getUserInfo(bodyLocations, bodySubLocations, text)


    # takes in a list of bodyLocations and all body subLocations and determines the user's affected areas
    # will return a sublocation
    def getUserInfo(self, bodyLocations, bodySubLocations, text):
        # get bodyLocation and subLocation based on inputif applicable
        bodyLocation = self.findBodyLocation(bodyLocations, text)
        bodySubLocation = self.findBodySubLocation(bodySubLocations, text)


        # bodyLocation = NULL, bodySubLocation = NULL
        # could not find the bodyLocation or subLocation specified
        if (bodyLocation is None and bodySubLocation is None):
            # keep asking for body location until they chose one of the selected
            while (bodyLocation is None):
                print("I'm not sure exactly where you are in pain. Could you pick from this list of body locations?\n")
                self.printBodyLocations(bodyLocations)
                text = input()
                bodyLocation = self.findBodySubLocation(bodyLocations, text)

        # bodyLocation = NULL, bodySubLocation = NOT NULL
        # user gave a subLocation but no bodyLocation
        #if(bodyLocation is None and bodySubLocation is not None):

        # keep bodyLocation as null, user just has sublocation

        # bodyLocation = NOT NULL, bodySubLocation = NULL
        # user gave a bodyLocation but no sublocation, must determine sublocation
        if(bodyLocation is not None and bodySubLocation is None):
            newBodySubLocations = self.getSubLocations(bodyLocation)  # get subLocations from selected bodyLocation
            newBodySubLocation = None
            # keep asking for body sublocation until they have chosen one of the selected
            while (newBodySubLocation is None):
                print("based on the body location that you chose: " + bodyLocation + " please select a subLocation from this list")
                self.printBodySubLocations(newBodySubLocations)
                newBodySubLocation = input()

            # update body sublocation
            bodySubLocation = self.findBodySubLocation(newBodySubLocations, newBodySubLocation)

        self.getSymptoms(bodySubLocation)

        # bodyLocation = NOT NULL, bodySubLocation = NOT NULL
        # user gave a location can go in either both a bodyLocation and subLocation
        #if (bodyLocation is not None and bodySubLocation is not None):




    # don't do anything, just use sublocation

    # get all sublocations for all bodylocations
    def getAllSubLocations(self, bodyLocations):
        bodySubLocations = []
        for bodyLocation in bodyLocations:
            bodySubLocations = bodySubLocations + self._diagnosisClient.loadBodySubLocations(bodyLocation["ID"])
        return bodySubLocations

    # get subLocations based on bodyLocation
    def getSubLocations(self, bodyLocation):
        return self._diagnosisClient.loadBodySubLocations(bodyLocation["ID"])


    # find body location / sublocation & ID based on user input
    def findBodyLocation(self, bodyLocations, text):
        # look for input in each bodyLocation name
        bodyLocation = ""
        for bodyL in bodyLocations:
            if text in bodyL["Name"]:
                bodyLocation = bodyL  # set the body location
        return bodyLocation

    # check for body sublocation on first user input
    def findBodySubLocation(self, bodySubLocations, text):
        # look for input use that instead of larger body Location
        for bodySL in bodySubLocations:
            if text in bodySL["Name"]:
                bodySubLocation = bodySL  # set body sub location
        return bodySubLocation



    # print out all body Locations
    def printBodyLocations(bodyLocations):
        for bodyLocation in bodyLocations:
            print(bodyLocation["Name\n"])

    # print out all sublocation based on body location
    def printBodySubLocations(bodySubLocations):
        for bodySubLocation in bodySubLocations:
            print(bodySubLocation["Name\n"])

    # get symptoms of sublocation
    def getSymptoms(self, bodySubLocation):
        print("Experienced pain at subLocation: " + bodySubLocation["Name"])
        symptoms = self._diagnosisClient.loadSublocationSymptoms(bodySubLocation["ID"], helper.SelectorStatus.Man)

        self.printSymptoms(symptoms)

        user_input = input()
        selectedSymptomsIDs = list(map(int, user_input.split(',')))

        diagnosis = self.getDiagnosis(selectedSymptomsIDs)
        for d in diagnosis:
            print("{0} - {1}% \n".format(d["Issue"]["Name"], d["Issue"]["Accuracy"]))

        #print(json.dumps(diagnosis))

    # print all symptoms based on bodyLocation
    def printSymptoms(self, symptoms):
        print("Which of these symptoms are you experiencing?")
        print("Choose IDs separated by a comma")
        for symptom in symptoms:
            print(symptom["ID"], symptom["Name"])



    # get diagnosis based on selected symptoms and user info like gender and age
    def getDiagnosis(self, selectedSymptoms):
        return self._diagnosisClient.loadDiagnosis(selectedSymptoms, helper.Gender.Male, 1988)



    def _loadSublocationSymptoms(self, subLocId):
        symptoms = self._diagnosisClient.loadSublocationSymptoms(subLocId, helper.SelectorStatus.Man)

        self._writeRawOutput("loadSublocationSymptoms", symptoms)

        if not symptoms:
            raise Exception("Empty body sublocations symptoms results")

        self._writeHeaderMessage("Body sublocations symptoms:")

        for symptom in symptoms:
            print(symptom["Name"])

        randomSymptom = random.choice(symptoms)

        self._writeHeaderMessage("Randomly selected symptom: {0}".format(randomSymptom["Name"]))

        self._loadRedFlag(randomSymptom)

        selectedSymptoms = [randomSymptom]
        return selectedSymptoms



    def _loadDiagnosis(self, selectedSymptoms):
        self._writeHeaderMessage("Diagnosis")
        selectedSymptomsIds = []
        for symptom in selectedSymptoms:
            selectedSymptomsIds.append(symptom["ID"])

        diagnosis = self._diagnosisClient.loadDiagnosis(selectedSymptomsIds, helper.Gender.Male, 1988)
        self._writeRawOutput("loadDiagnosis", diagnosis)

        if not diagnosis:
            self._writeHeaderMessage("No diagnosis results for symptom {0}".format(selectedSymptoms[0]["Name"]))

        for d in diagnosis:
            specialisations = []
            for specialisation in d["Specialisation"]:
                specialisations.append(specialisation["Name"])
            print("{0} - {1}% \nICD: {2}{3}\nSpecialisations : {4}\n".format(d["Issue"]["Name"], d["Issue"]["Accuracy"],
                                                                             d["Issue"]["Icd"], d["Issue"]["IcdName"],
                                                                             ",".join(x for x in specialisations)))

        diagnosisIds = []
        for diagnose in diagnosis:
            diagnosisIds.append(diagnose["Issue"]["ID"])

        return diagnosisIds

    def _loadSpecialisations(self, selectedSymptoms):
        self._writeHeaderMessage("Specialisations")
        selectedSymptomsIds = []
        for symptom in selectedSymptoms:
            selectedSymptomsIds.append(symptom["ID"])

        specialisations = self._diagnosisClient.loadSpecialisations(selectedSymptomsIds,
                                                                    helper.Gender.Male, 1988)
        self._writeRawOutput("loadSpecialisations", specialisations)

        if not specialisations:
            self._writeHeaderMessage("No specialisations for symptom {0}".format(selectedSymptoms[0]["Name"]))

        for specialisation in specialisations:
            print("{0} - {1}%".format(specialisation["Name"], specialisation["Accuracy"]))

    def _loadRedFlag(self, selectedSymptom):
        redFlag = "Symptom {0} has no red flag".format(selectedSymptom["Name"])

        if selectedSymptom["HasRedFlag"]:
            redFlag = self._diagnosisClient.loadRedFlag(selectedSymptom["ID"])
            self._writeRawOutput("loadRedFlag", redFlag)

        self._writeHeaderMessage(redFlag);

    def _loadIssueInfo(self, issueId):
        issueInfo = self._diagnosisClient.loadIssueInfo(issueId)
        self._writeRawOutput("issueInfo", issueInfo)

        self._writeHeaderMessage("Issue info")
        print("Name: {0}".format(issueInfo["Name"]).encode("utf-8"))
        print("Professional Name: {0}".format(issueInfo["ProfName"]).encode("utf-8"))
        print("Synonyms: {0}".format(issueInfo["Synonyms"]).encode("utf-8"))
        print("Short Description: {0}".format(issueInfo["DescriptionShort"]).encode("utf-8"))
        print("Description: {0}".format(issueInfo["Description"]).encode("utf-8"))
        print("Medical Condition: {0}".format(issueInfo["MedicalCondition"]).encode("utf-8"))
        print("Treatment Description: {0}".format(issueInfo["TreatmentDescription"]).encode("utf-8"))
        print("Possible symptoms: {0} \n\n".format(issueInfo["PossibleSymptoms"]).encode("utf-8"))

    def _loadProposedSymptoms(self, selectedSymptoms):
        selectedSymptomsIds = []
        for symptom in selectedSymptoms:
            selectedSymptomsIds.append(symptom["ID"])

        proposedSymptoms = self._diagnosisClient.loadProposedSymptoms(selectedSymptomsIds,
                                                                      helper.Gender.Male, 1988)
        self._writeRawOutput("proposedSymptoms", proposedSymptoms)

        if not proposedSymptoms:
            self._writeHeaderMessage(
                "No proposed symptoms for selected symptom {0}".format(selectedSymptoms[0]["Name"]))
            return

        proposedSymptomsIds = []
        for proposeSymptom in proposedSymptoms:
            proposedSymptomsIds.append(proposeSymptom["ID"])

        self._writeHeaderMessage("Proposed symptoms: {0}".format(",".join(str(x) for x in proposedSymptomsIds)))


symptomChecker = SymptomChecker()
symptomChecker.start()