def add_receipt(num_receipts=1):
user = User.query.first()
days_ago_to_start = num_receipts/2
date = datetime.now() - timedelta(days=days_ago_to_start)
even_receipt = True
for x in range(num_receipts):
SHOP_NAME = "Countdown Birkenhead"
SHOP_CATEGORY = "Groceries"
TOTAL = 123.81
r = Receipt(SHOP_NAME, SHOP_CATEGORY, TOTAL, date)
r.purchased_items = [
PurchasedItem("Chocolate Milk", SHOP_CATEGORY, 3.99, 2),
PurchasedItem("Garlic Pita Breads", SHOP_CATEGORY, 4.50, 1),
PurchasedItem("Old Spice Deoderant", SHOP_CATEGORY, 6.21, 4),
PurchasedItem("Eye Fillet Steak", SHOP_CATEGORY, 21.4, 1),
PurchasedItem("Obikwa Wine 750mL", SHOP_CATEGORY, 6.99, 4),
PurchasedItem("North Shore Rubbish Sack", SHOP_CATEGORY, 4.99, 5),
PurchasedItem("Nutri-Grain 700g", SHOP_CATEGORY, 6.23, 1),
PurchasedItem("Olivio Butter", SHOP_CATEGORY, 5.95, 1)
]
user.receipts.append(r)
even_receipt = not even_receipt
if even_receipt:
date = date + timedelta(days=1)
print len(user.receipts)
db.add(user)
db.commit()
def rand_receipt():
store = choice(store_names)
# print store
store_name = store.split("-")[0].strip()
category = store.split("-")[1].strip()
# tax_rate = randint(100, 150)/1000.0
items = rand_items(store_name, category, randint(1, 7))
total_transaction = 0.0
for item in items:
total_transaction += item.price_per_item * item.quantity
total_transaction = round(total_transaction, 2)
days = randint(0, 21)
hours = randint(0,23)
minutes = randint(0, 59)
time_ago = timedelta(days=days, hours=hours, minutes=minutes)
receipt = Receipt(store_name, category, total_transaction, datetime.now()-time_ago)
receipt.purchased_items = items
return receipt
