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)