def __init__(self):
# Disease name : Table disease object
self.bucket = dict()
# for algo1
self.symptom_score = dict()
# for algo2
self.symptom_critical_count = dict()
#for algo3
self.top_value_bucket_symptoms = dict()
self.disease_score = dict()
self.disease_fraction_done = dict()
self.done = 0
self.removed_questions_list = list()
self.finished_diseases_list = list()
self.disease_about_to_removed = list()
# stores the symptom that is not True / False
self.symptom_signature_needed = ['fever']
self.diseases_object = Disease()
self.disease_signature_object = Disease_Signature()
# self initialization of object in init
self.populate_diseases()
self.disease_top_score = dict(self.disease_score)
def __init__(self):
# Disease name : Table disease object
self.bucket = dict()
# for algo1
self.symptom_score = dict()
# for algo2
self.symptom_critical_count = dict()
#for algo3
self.top_value_bucket_symptoms = dict()
self.disease_score = dict()
self.disease_fraction_done = dict()
self.done = 0
self.removed_questions_list = list()
self.finished_diseases_list = list()
self.disease_about_to_removed = list()
# stores the symptom that is not True / False
self.symptom_signature_needed = ['fever']
self.diseases_object = Disease()
self.disease_signature_object = Disease_Signature()
# self initialization of object in init
self.populate_diseases()
self.disease_top_score = dict(self.disease_score)
def get_new_diseases(self):
diseases_section = self.settings['Diseases']
number_of_cubes = self.settings['Other'].getint('cubes')
for disease_id in diseases_section:
disease_colour = diseases_section[disease_id]
self.diseases[disease_colour] = Disease(disease_colour)
self.disease_cubes[disease_colour] = number_of_cubes
def __init__(self):
disease_object_home = Disease()
disease_dict_home = disease_object_home.get_disease()
self.hepA = "hepA"
self.dengue = "dengue"
self.malaria = "malaria"
self.typhoid_fever = "typhoid_fever"
self.leptospirosis = "leptospirosis"
self.fever_states = ["Yes, High (> 103 F)", "Yes, Mild (101-103 F)", "Yes, Very Mild (99 - 101 F)", "No"]
# print self.fever_states[0]
self.fever = {
self.dengue: self.fever_states[0],
self.hepA: self.fever_states[1],
self.malaria: self.fever_states[0],
self.typhoid_fever: self.fever_states[0],
self.leptospirosis: self.fever_states[1],
}
def __init__(self):
disease_object_home = Disease()
disease_dict_home = disease_object_home.get_disease()
self.hepA = 'hepA'
self.dengue = 'dengue'
self.malaria = 'malaria'
self.typhoid_fever = 'typhoid_fever'
self.leptospirosis = 'leptospirosis'
self.fever_states = ['Yes, High (> 103 F)', 'Yes, Mild (101-103 F)', 'Yes, Very Mild (99 - 101 F)', 'No']
# print self.fever_states[0]
self.fever = {
self.dengue: self.fever_states[0],
self.hepA: self.fever_states[1],
self.malaria : self.fever_states[0],
self.typhoid_fever : self.fever_states[0],
self.leptospirosis : self.fever_states[1]
}
def test_toy_model():
random.seed(42)
s = Society(episodes_per_day=5, encounter_size=2)
d = Disease(days_infectious=10, pr_transmission_per_day=0.2)
# seed size is the number of people in the population who we seed as being infected:
o = Outbreak(s,
d,
pop_size=1000,
seed_size=2,
n_days=ALL_TIME_DAYS,
population_type=Population,
person_type=Person)
# so, this is a village of 10000 people with 2 starting off infected
o.simulate()
assert o.recorder.story[90:95] == \
[[18.199999999999967, 0.619, 0.57, 0.0, 0.0, 0.0],
[18.399999999999967, 0.643, 0.592, 0.0, 0.0, 0.0],
[18.599999999999966, 0.656, 0.603, 0.0, 0.0, 0.0],
[18.799999999999965, 0.673, 0.618, 0.0, 0.0, 0.0],
[18.999999999999964, 0.69, 0.635, 0.0, 0.0, 0.0]]
class Buckets:
def __init__(self):
# Disease name : Table disease object
self.bucket = dict()
# for algo1
self.symptom_score = dict()
# for algo2
self.symptom_critical_count = dict()
#for algo3
self.top_value_bucket_symptoms = dict()
self.disease_score = dict()
self.disease_fraction_done = dict()
self.done = 0
self.removed_questions_list = list()
self.finished_diseases_list = list()
self.disease_about_to_removed = list()
# stores the symptom that is not True / False
self.symptom_signature_needed = ['fever']
self.diseases_object = Disease()
self.disease_signature_object = Disease_Signature()
# self initialization of object in init
self.populate_diseases()
self.disease_top_score = dict(self.disease_score)
"""
<------------------------PUBLIC METHODS TO BY CALLED BY OUTSIDE MODULES----------------->
"""
"""
returns the list of asked questions , used in -1 , 0 as they use db no this module
"""
def get_asked_questions(self):
return self.removed_questions_list
"""
returns a list of [Disease Name , Fraction Done ] for doc mail
"""
def get_bucket_fraction_list(self):
lists = list()
for disease_name in self.bucket.keys():
self.calculate_fraction(disease_name)
fraction = self.disease_fraction_done[disease_name]
lists.append([disease_name, fraction])
return lists
"""
calculates avg score of all active buckets.
"""
def get_avg_fraction(self):
for disease_name in self.bucket.keys():
self.calculate_fraction(disease_name)
number = 0
fraction = float(0)
for fractions in self.disease_fraction_done.values():
number = number + 1
fraction = fraction + fractions
if number == 0:
self.done = 1
return None
return fraction / float(number)
"""
get_top_symptoms()
returns a ordered list of symptom(s) with the highest score across all diseases
for algo1
"""
def get_popular_symptoms(self, number_of_symptoms=1):
"""
top_symptoms = heapq.nlargest(number_of_symptoms, self.symptom_score,
key=self.symptom_score.get)
"""
top_symptoms = keywithmaxval(self.symptom_score)
return top_symptoms
"""
get_top_critical_symptoms()
returns a ordered list of symptom(s) with the highest critical count across all diseases
for algo2
"""
def get_top_critical_symptoms(self, number_of_symptoms=1):
"""
top_symptoms = heapq.nlargest(number_of_symptoms, self.symptom_critical_count,
key=self.symptom_critical_count.get)
"""
top_symptoms = keywithmaxval(self.symptom_critical_count)
return top_symptoms
"""
get_buckets_top_symptom()
returns a list of the top symptom in each bucket
for algo3
"""
def get_buckets_top_symptom(self, number_of_symptoms=1):
self.calculate_highest_symptom()
# print self.top_value_bucket_symptoms
"""
list_symptom = heapq.nlargest(number_of_symptoms, self.top_value_bucket_symptoms,
key=self.top_value_bucket_symptoms.get)
"""
list_symptom = keywithmaxval(self.top_value_bucket_symptoms)
return list_symptom
"""
question_asked()
removes symptom from all diseases in symptom_validity_table objects.
removes symptom from all scores
also removes buckets based on a single critical Mismatch or a double Important
Mismatch.
re-calculates score for all disease buckets
response can be True, False or 'Yes'
"""
def answered_question_True(self, symptom, response, dbconnection, chat_ID):
# print self.disease_score
if self.done == 1:
print "No More Stuff"
if response == 'Yes':
# print "Response is now True"
response = True
if symptom in self.removed_questions_list:
return None
for table_disease_name in self.bucket.keys():
table_disease_object = self.bucket[table_disease_name]
symptom_state = table_disease_object.get(symptom)
# print symptom_state
if response == symptom_state:
table_disease_object.set_score(symptom, 0)
table_disease_object.set(symptom, False)
dbconnection.set_symptom_data_for_chat_id(chat_ID, symptom)
else:
table_disease_object.set(symptom, False)
if table_disease_object.get_score(symptom) == CRITICAL:
self.remove_disease(table_disease_name)
print bcolors.FAIL + table_disease_name + " removed"
if len(self.bucket) == 0:
self.done = 1
if table_disease_object.get_score(symptom) == IMPORTANT:
if table_disease_name in self.disease_about_to_removed:
self.remove_disease(table_disease_name)
print bcolors.FAIL + table_disease_name + " removed"
print bcolors.OKBLUE
if len(self.bucket) == 0:
self.done = 1
else:
self.disease_about_to_removed.append(table_disease_name)
# print "Mismatch:" + str(symptom)
# print "-------------------------"
# print table_disease_object.get_score('fever')
# print"------------------------------------"
if symptom in self.symptom_score.keys():
self.symptom_score.pop(symptom)
if symptom in self.top_value_bucket_symptoms.keys():
self.top_value_bucket_symptoms.pop(symptom)
if symptom in self.symptom_critical_count.keys():
self.symptom_critical_count.pop(symptom)
for disease_name in self.bucket.keys():
self.calculate_current_score(disease_name)
self.removed_questions_list.append(symptom)
"""
<---------------------------------------INTERNAL METHODS-------------------------------->
"""
'''
remove_disease()
removes disease and it's score from all dicts
'''
def remove_disease(self, name):
if name in self.bucket.keys():
diseases_dict = self.diseases_object.get_disease()
for specific_disease_name in diseases_dict.keys():
if name == specific_disease_name:
disease_dict = diseases_dict[specific_disease_name]
for symptom in disease_dict.keys():
if symptom in self.removed_questions_list:
pass
else:
self.remove_symptom_score(symptom, disease_dict)
self.bucket.pop(disease_dict['name'])
self.disease_score.pop(disease_dict['name'])
self.disease_fraction_done.pop(disease_dict['name'])
pass
def get_score_by_symptom(self, symptom):
if symptom in self.removed_questions_list:
return None
elif symptom in self.symptom_score.keys():
return self.symptom_score[symptom]
else:
return None
def get_symptom_by_disease(self, symptom, disease):
table = symptom_validity_table
table = self.bucket[disease]
return table.get(symptom)
def get_score_by_disease(self, disease_name):
if disease_name in self.bucket.keys():
self.calculate_current_score(disease_name)
return self.disease_score[disease_name]
else:
return "INVALID DISEASE"
pass
"""
calculate_highest_symptom()
populates the top_value_bucket dict
runs : get_highest_symptom
: calculate_highest_symptom
"""
def calculate_highest_symptom(self):
for disease_name in self.bucket:
symptom = self.get_highest_symptom(disease_name)
if symptom is None:
pass
else:
score = self.bucket[disease_name].get_score(symptom)
self.update_top_value(symptom, score)
'''
updates the symptom top_value dict
'''
def update_top_value(self, symptom, score):
if symptom in self.top_value_bucket_symptoms.keys():
self.top_value_bucket_symptoms[
symptom] = self.top_value_bucket_symptoms[symptom] + score
else:
self.top_value_bucket_symptoms[symptom] = score
"""
get_highest_symptom()
returns the highest unanswered symptom in a given disease
"""
def get_highest_symptom(self, disease_name):
diseases_name_dict = self.diseases_object.get_disease()
disease_dict = diseases_name_dict[disease_name]
name = None
max = 0
for symptom in disease_dict.keys():
if symptom in self.removed_questions_list:
pass
elif symptom == 'name':
pass
else:
score = disease_dict[symptom]
if score > max:
max = score
name = symptom
return name
"""
calculate_current_score()
calculates total score for the given disease
"""
def calculate_current_score(self, disease_name):
if disease_name in self.bucket.keys():
list_disease_dict = self.diseases_object.get_disease()
specific_disease_dict = list_disease_dict[disease_name]
table_disease_object = self.bucket[disease_name]
score = 0
for symptom in specific_disease_dict.keys():
if symptom == 'name':
pass
else:
temp = table_disease_object.get_score(symptom)
score = score + temp
self.disease_score[disease_name] = score
"""
add_symptom_score()
adds score across diseases by symptoms
"""
def add_symptom_score(self, symptom, score):
if symptom in self.symptom_score.keys():
self.symptom_score[symptom] = self.symptom_score[symptom] + score
else:
self.symptom_score[symptom] = score
"""
update_critical_count()
increments the critical count of a critical symptom
"""
def update_critical_count(self, symptom):
if symptom in self.symptom_critical_count.keys():
self.symptom_critical_count[symptom] += 1
else:
self.symptom_critical_count[symptom] = 1
'''
remove_symptom_score()
removes symptom score from disease dict
removes symptom count from critical_count_dict
'''
def remove_symptom_score(self, symptom, disease_dict):
if disease_dict[symptom] is None:
print "symptom already asked , hence ignore"
elif symptom == 'name':
pass
else:
self.symptom_score[symptom] -= disease_dict[symptom]
if disease_dict[symptom] == CRITICAL:
self.symptom_critical_count[symptom] -= 1
"""
calculates the fractional completion of a disease
"""
def calculate_fraction(self, disease_name):
done = self.disease_top_score[disease_name] - self.disease_score[
disease_name]
todo = self.disease_top_score[disease_name]
fractiondone = done / float(todo)
self.disease_fraction_done[disease_name] = fractiondone
"""
set_table():
works on per symptom basis
adds symptom score to total symptom score ie
sets symptom to True in symptom validity table
set symptom score in symptom validity table
calls : add_symptom_score()
: update_critical_count()
: set_score()
: set()
called by
: populate_diseases
"""
def set_table(self,
symptom,
score,
table_disease_object,
disease_name,
arg=True):
table_disease_object.set_score(symptom, score)
if symptom in self.symptom_signature_needed:
"""
as the fever signature is not just yes no :)
"""
table_disease_object.set(
symptom, self.disease_signature_object.get_fever(disease_name))
#print table_disease_object.get_dict()
else:
table_disease_object.set(symptom, arg)
if table_disease_object.get_score(symptom) == CRITICAL:
self.update_critical_count(symptom)
self.add_symptom_score(symptom, score)
"""
populate_diseases()
poplulates diseases from db and sets in the symptom validity table
also adds the sypmtom validity object to the bucket dict
"""
def populate_diseases(self):
diseases_dict = self.diseases_object.get_disease()
for specific_disease_name in diseases_dict.keys():
new_table_obj = symptom_validity_table()
specific_disease_dict = diseases_dict[specific_disease_name]
for symptom in specific_disease_dict.keys():
#print symptom
if symptom == 'name':
print specific_disease_dict[symptom] + " : LOADED"
else:
self.set_table(symptom, specific_disease_dict[symptom],
new_table_obj, specific_disease_name)
self.bucket[specific_disease_name] = new_table_obj
#print new_table_obj.get_dict()
self.calculate_current_score(specific_disease_dict['name'])
def test_init(self):
disease = Disease('Blue')
self.assertEqual('Blue', disease.colour)
self.assertFalse(disease.cured)
egg = Product('Egg')
tomato = Product('Tomato')
cucumber = Product('Cucumber')
print(orange)
dish1 = Dish('Dish1', [orange, chocolate, milk], 34.5, 10)
dish2 = Dish('Dish2', [lemon, strawberry, orange], 23, 5)
dish3 = Dish('Dish3', [egg, milk, chocolate], 50, 8)
dish4 = Dish('Dish4', [tomato, cucumber, orange], 30, 7)
print(dish1)
restaurant.menu = [dish1, dish2, dish3, dish4]
for dish in restaurant.menu:
print(dish)
disease1 = Disease('Citrus allergy', [orange, lemon])
disease2 = Disease('Strawberry allergy', [strawberry])
disease3 = Disease('Lactose intolerance', [milk, egg])
disease4 = Disease('Chocolate allergy', [chocolate])
print(disease1)
customer = Customer('Robert', 'Smith', '*****@*****.**',
[disease4])
print(customer)
menu_customer = customer.get_personal_menu(restaurant)
for dish in menu_customer:
print(dish)
waiter = Waiter('Elliot', 'Smith', 35.7)
print(waiter)
class Buckets:
def __init__(self):
# Disease name : Table disease object
self.bucket = dict()
# for algo1
self.symptom_score = dict()
# for algo2
self.symptom_critical_count = dict()
#for algo3
self.top_value_bucket_symptoms = dict()
self.disease_score = dict()
self.disease_fraction_done = dict()
self.done = 0
self.removed_questions_list = list()
self.finished_diseases_list = list()
self.disease_about_to_removed = list()
# stores the symptom that is not True / False
self.symptom_signature_needed = ['fever']
self.diseases_object = Disease()
self.disease_signature_object = Disease_Signature()
# self initialization of object in init
self.populate_diseases()
self.disease_top_score = dict(self.disease_score)
"""
<------------------------PUBLIC METHODS TO BY CALLED BY OUTSIDE MODULES----------------->
"""
"""
returns the list of asked questions , used in -1 , 0 as they use db no this module
"""
def get_asked_questions(self):
return self.removed_questions_list
"""
returns a list of [Disease Name , Fraction Done ] for doc mail
"""
def get_bucket_fraction_list(self):
lists = list()
for disease_name in self.bucket.keys():
self.calculate_fraction(disease_name)
fraction = self.disease_fraction_done[disease_name]
lists.append([disease_name, fraction])
return lists
"""
calculates avg score of all active buckets.
"""
def get_avg_fraction(self):
for disease_name in self.bucket.keys():
self.calculate_fraction(disease_name)
number = 0
fraction = float(0)
for fractions in self.disease_fraction_done.values():
number = number + 1
fraction = fraction + fractions
if number == 0:
self.done = 1
return None
return fraction / float(number)
"""
get_top_symptoms()
returns a ordered list of symptom(s) with the highest score across all diseases
for algo1
"""
def get_popular_symptoms(self, number_of_symptoms=1):
"""
top_symptoms = heapq.nlargest(number_of_symptoms, self.symptom_score,
key=self.symptom_score.get)
"""
top_symptoms = keywithmaxval(self.symptom_score)
return top_symptoms
"""
get_top_critical_symptoms()
returns a ordered list of symptom(s) with the highest critical count across all diseases
for algo2
"""
def get_top_critical_symptoms(self, number_of_symptoms=1):
"""
top_symptoms = heapq.nlargest(number_of_symptoms, self.symptom_critical_count,
key=self.symptom_critical_count.get)
"""
top_symptoms = keywithmaxval(self.symptom_critical_count)
return top_symptoms
"""
get_buckets_top_symptom()
returns a list of the top symptom in each bucket
for algo3
"""
def get_buckets_top_symptom(self, number_of_symptoms=1):
self.calculate_highest_symptom()
# print self.top_value_bucket_symptoms
"""
list_symptom = heapq.nlargest(number_of_symptoms, self.top_value_bucket_symptoms,
key=self.top_value_bucket_symptoms.get)
"""
list_symptom = keywithmaxval(self.top_value_bucket_symptoms)
return list_symptom
"""
question_asked()
removes symptom from all diseases in symptom_validity_table objects.
removes symptom from all scores
also removes buckets based on a single critical Mismatch or a double Important
Mismatch.
re-calculates score for all disease buckets
response can be True, False or 'Yes'
"""
def answered_question_True(self, symptom, response, dbconnection, chat_ID):
# print self.disease_score
if self.done == 1:
print "No More Stuff"
if response == 'Yes':
# print "Response is now True"
response = True
if symptom in self.removed_questions_list:
return None
for table_disease_name in self.bucket.keys():
table_disease_object = self.bucket[table_disease_name]
symptom_state = table_disease_object.get(symptom)
# print symptom_state
if response == symptom_state:
table_disease_object.set_score(symptom, 0)
table_disease_object.set(symptom, False)
dbconnection.set_symptom_data_for_chat_id(chat_ID, symptom)
else:
table_disease_object.set(symptom, False)
if table_disease_object.get_score(symptom) == CRITICAL:
self.remove_disease(table_disease_name)
print bcolors.FAIL + table_disease_name + " removed"
if len(self.bucket) == 0:
self.done = 1
if table_disease_object.get_score(symptom) == IMPORTANT:
if table_disease_name in self.disease_about_to_removed:
self.remove_disease(table_disease_name)
print bcolors.FAIL + table_disease_name + " removed"
print bcolors.OKBLUE
if len(self.bucket) == 0:
self.done = 1
else:
self.disease_about_to_removed.append(table_disease_name)
# print "Mismatch:" + str(symptom)
# print "-------------------------"
# print table_disease_object.get_score('fever')
# print"------------------------------------"
if symptom in self.symptom_score.keys():
self.symptom_score.pop(symptom)
if symptom in self.top_value_bucket_symptoms.keys():
self.top_value_bucket_symptoms.pop(symptom)
if symptom in self.symptom_critical_count.keys():
self.symptom_critical_count.pop(symptom)
for disease_name in self.bucket.keys():
self.calculate_current_score(disease_name)
self.removed_questions_list.append(symptom)
"""
<---------------------------------------INTERNAL METHODS-------------------------------->
"""
'''
remove_disease()
removes disease and it's score from all dicts
'''
def remove_disease(self, name):
if name in self.bucket.keys():
diseases_dict = self.diseases_object.get_disease()
for specific_disease_name in diseases_dict.keys():
if name == specific_disease_name:
disease_dict = diseases_dict[specific_disease_name]
for symptom in disease_dict.keys():
if symptom in self.removed_questions_list:
pass
else:
self.remove_symptom_score(symptom, disease_dict)
self.bucket.pop(disease_dict['name'])
self.disease_score.pop(disease_dict['name'])
self.disease_fraction_done.pop(disease_dict['name'])
pass
def get_score_by_symptom(self, symptom):
if symptom in self.removed_questions_list:
return None
elif symptom in self.symptom_score.keys():
return self.symptom_score[symptom]
else:
return None
def get_symptom_by_disease(self, symptom, disease):
table = symptom_validity_table
table = self.bucket[disease]
return table.get(symptom)
def get_score_by_disease(self, disease_name):
if disease_name in self.bucket.keys():
self.calculate_current_score(disease_name)
return self.disease_score[disease_name]
else:
return "INVALID DISEASE"
pass
"""
calculate_highest_symptom()
populates the top_value_bucket dict
runs : get_highest_symptom
: calculate_highest_symptom
"""
def calculate_highest_symptom(self):
for disease_name in self.bucket:
symptom = self.get_highest_symptom(disease_name)
if symptom is None:
pass
else:
score = self.bucket[disease_name].get_score(symptom)
self.update_top_value(symptom, score)
'''
updates the symptom top_value dict
'''
def update_top_value(self, symptom, score):
if symptom in self.top_value_bucket_symptoms.keys():
self.top_value_bucket_symptoms[symptom] = self.top_value_bucket_symptoms[symptom] + score
else:
self.top_value_bucket_symptoms[symptom] = score
"""
get_highest_symptom()
returns the highest unanswered symptom in a given disease
"""
def get_highest_symptom(self, disease_name):
diseases_name_dict = self.diseases_object.get_disease()
disease_dict = diseases_name_dict[disease_name]
name = None
max = 0
for symptom in disease_dict.keys():
if symptom in self.removed_questions_list:
pass
elif symptom == 'name':
pass
else:
score = disease_dict[symptom]
if score > max:
max = score
name = symptom
return name
"""
calculate_current_score()
calculates total score for the given disease
"""
def calculate_current_score(self,disease_name):
if disease_name in self.bucket.keys():
list_disease_dict=self.diseases_object.get_disease()
specific_disease_dict=list_disease_dict[disease_name]
table_disease_object=self.bucket[disease_name]
score=0
for symptom in specific_disease_dict.keys():
if symptom == 'name':
pass
else:
temp=table_disease_object.get_score(symptom)
score=score+temp
self.disease_score[disease_name]=score
"""
add_symptom_score()
adds score across diseases by symptoms
"""
def add_symptom_score(self,symptom,score):
if symptom in self.symptom_score.keys():
self.symptom_score[symptom]=self.symptom_score[symptom]+score
else :
self.symptom_score[symptom] = score
"""
update_critical_count()
increments the critical count of a critical symptom
"""
def update_critical_count(self, symptom):
if symptom in self.symptom_critical_count.keys():
self.symptom_critical_count[symptom] += 1
else:
self.symptom_critical_count[symptom] = 1
'''
remove_symptom_score()
removes symptom score from disease dict
removes symptom count from critical_count_dict
'''
def remove_symptom_score(self,symptom,disease_dict):
if disease_dict[symptom] is None :
print "symptom already asked , hence ignore"
elif symptom == 'name':
pass
else:
self.symptom_score[symptom] -= disease_dict[symptom]
if disease_dict[symptom] == CRITICAL:
self.symptom_critical_count[symptom] -= 1
"""
calculates the fractional completion of a disease
"""
def calculate_fraction(self, disease_name):
done = self.disease_top_score[disease_name] - self.disease_score[disease_name]
todo = self.disease_top_score[disease_name]
fractiondone = done / float(todo)
self.disease_fraction_done[disease_name] = fractiondone
"""
set_table():
works on per symptom basis
adds symptom score to total symptom score ie
sets symptom to True in symptom validity table
set symptom score in symptom validity table
calls : add_symptom_score()
: update_critical_count()
: set_score()
: set()
called by
: populate_diseases
"""
def set_table(self, symptom, score, table_disease_object, disease_name, arg=True):
table_disease_object.set_score(symptom, score)
if symptom in self.symptom_signature_needed:
"""
as the fever signature is not just yes no :)
"""
table_disease_object.set(symptom, self.disease_signature_object.get_fever(disease_name))
#print table_disease_object.get_dict()
else:
table_disease_object.set(symptom, arg)
if table_disease_object.get_score(symptom) == CRITICAL:
self.update_critical_count(symptom)
self.add_symptom_score(symptom, score)
"""
populate_diseases()
poplulates diseases from db and sets in the symptom validity table
also adds the sypmtom validity object to the bucket dict
"""
def populate_diseases(self):
diseases_dict=self.diseases_object.get_disease()
for specific_disease_name in diseases_dict.keys():
new_table_obj = symptom_validity_table()
specific_disease_dict = diseases_dict[specific_disease_name]
for symptom in specific_disease_dict.keys():
#print symptom
if symptom == 'name':
print specific_disease_dict[symptom] + " : LOADED"
else:
self.set_table(symptom, specific_disease_dict[symptom],
new_table_obj, specific_disease_name)
self.bucket[specific_disease_name] = new_table_obj
#print new_table_obj.get_dict()
self.calculate_current_score(specific_disease_dict['name'])
