def search_by_str():
search = raw_input("Enter your search term: ")
drugs = start()
drugs = return_match(drugs, search)
if len(drugs) > 0:
drugs = add_FDA_info(drugs)
count = 1
for drug in drugs:
print str(count) + "."
Drug.printer(drugs[drug])
print
count += 1
else:
print "No results."
def search_by_num():
print "Enter minimum and maximum current price and percent increase in two years."
print "You can just hit enter if you don't want a minimum or maximum."
raw_min_price = raw_input("MINIMUM CURRENT PRICE: ")
raw_min_percent = raw_input("MINIMUM PERCENT INCREASE: ")
raw_max_price = raw_input("MAXIMUM CURRENT PRICE: ")
raw_max_percent = raw_input("MAXIMUM PERCENT INCREASE: ")
print
if raw_min_percent:
min_percent = float(remove_stuff(raw_min_percent)) / 100
else:
min_percent = -10000000
if raw_min_price:
min_price = float(remove_stuff(raw_min_price))
else:
min_price = 0
if raw_max_percent:
max_percent = float(remove_stuff(raw_max_percent)) / 100
else:
max_percent = 1000000
if raw_max_price:
max_price = float(remove_stuff(raw_max_price))
else:
max_price = 100000000
drugs = start()
drugs = return_highest(drugs, min_percent, min_price, max_percent, max_price)
if len(drugs) > 0:
drugs = add_FDA_info(drugs)
count = 1
print
for drug in drugs:
print str(count) + "."
Drug.printer(drugs[drug])
print
count += 1
ask_for_graph(drugs)
else:
print "No results."
def write(drugs):
fieldnames = ["ID", "Name", "Scientific Name", "Unit", "OTC", "Brand/Generic", "Vendor", "Package", "Description"]
dates = get_date_list()
for i in range(len(dates)):
fieldnames.append(dates[i])
with open("FullPrices.csv","a") as prices_csv:
writer = csv.writer(prices_csv, delimiter='|')
writer.writerow(fieldnames)
for entry in drugs:
drug = drugs[entry]
drugrow = [drug.id, drug.name, drug.scientific_name, drug.unit, drug.otc, drug.b_or_g, drug.vendor, drug.package, drug.desc]
for i in range(len(dates)):
try:
drugrow.append(drug.prices[dates[i]])
except KeyError:
drugrow.append(None)
writer.writerow(drugrow)
prices_csv.close()
print "Done"
if __name__ == '__main__':
drugs = create_list()
count = 0
for drug in drugs:
print "%i." % count
Drug.printer(drugs[drug])
print
count += 1
write(drugs)
