- added deposits

This commit is contained in:
Bernhard Stampfer
2016-02-26 18:07:31 +01:00
parent 70774e892b
commit 2ee9089583
6 changed files with 51 additions and 3 deletions

View File

@@ -4,6 +4,7 @@ from app import app
from user import User
from product import Product
from consumption import Consumption
from deposit import Deposit
import random as rand
import datetime
@@ -183,6 +184,24 @@ def add_consume(username, productid):
return
def get_deposits(userid = None):
#ID|USERID|AMOUNT|TIME
if userid == None:
rows = query_db("SELECT * FROM DEPOSITS")
else:
rows = query_db("SELECT * FROM DEPOSITS WHERE USERID = ?", [str(userid)])
deposits = []
if rows == None:
return deposits
for row in rows:
d = Deposit()
d.id = row[0]
d.userid = row[1]
d.amount = row[2]
d.time = datetime.datetime.strptime(row[3], "%Y-%m-%d %H:%M:%S")
deposits.append(d)
return deposits
##for testing only
def generate_test_users():