some cleanup and rewrite

This commit is contained in:
Jan-Philipp Warmers 2014-04-19 14:31:04 +02:00
parent 9a1e944938
commit 0e3dd402b8
1 changed files with 104 additions and 59 deletions

View File

@ -1,12 +1,13 @@
#!/usr/bin/env python3
__author__ = 'progamler'
from os import system
import curses
from sqlalchemy import Table, Column, Integer, String, MetaData, ForeignKey, create_engine
import datetime
from sqlalchemy import Table, Column, Integer, String, MetaData, ForeignKey, create_engine, Date, func
from sqlalchemy.sql import select
# Config foo
db_uri = 'sqlite:///./db.sqlite'
@ -24,12 +25,14 @@ def create_db():
Column('barcode', Integer, primary_key=True),
Column('name', String(40)),
Column('preis', Integer), )
users = Table('user', metadata,
user = Table('user', metadata,
Column('barcode', Integer, primary_key=True),
Column('name', String(40)), )
#transaktion = Table('transaktion', metadata,
#Column('produkt', Integer, ForeignKey('produkte.barcode')),
#Column('user', Integer ,ForeignKey('users.barcode')))
transaktion = Table('transaktion', metadata,
Column('id', Integer, primary_key=True),
Column('produkt', Integer, ForeignKey('produkte.barcode')),
Column('user', Integer, ForeignKey('user.barcode')),
Column('time', Date, server_default=func.now()), sqlite_autoincrement=True)
metadata.create_all(db)
@ -39,22 +42,22 @@ def create_db():
db.execute(i, name='l33t', preis=1337, barcode='3')
db.execute(i, name='Clubmate 0.5', preis=1.5, barcode='4029764001807')
u = users.insert()
u = user.insert()
db.execute(u, barcode='23', name='proGamler')
def return_price(produkt):
db = create_engine(db_uri)
metadata = MetaData(db)
metadata.reflect()
produkte = Table('produkte', metadata, autoload=True)
s = select(['name', 'preis']).where(produkte.c.barcode == produkt)
s = select(['name', 'preis', 'barcode']).where(produkte.c.barcode == produkt)
result = db.execute(s)
return result.fetchone()
def isproduct(code):
foo = return_price(code)
try:
@ -63,32 +66,58 @@ def isproduct(code):
except:
return False
def user(code):
db = create_engine(db_uri)
metadata = MetaData(db)
metadata.reflect()
user = Table('user', metadata, autoload=True)
s = select(['name']).where(user.c.barcode == code)
s = select(['name', 'barcode']).where(user.c.barcode == code)
result = db.execute(s)
x = result.fetchone()
return x.name
return x
def isuser(code):
try:
name = user(code)
try:
print(name.name)
return True
except:
return False
def logout(x):
if x == 666:
return True
else:
return False
def count(date):
db = create_engine(db_uri)
metadata = MetaData(db)
metadata.reflect()
trans = Table('transaktion', metadata, autoload=True)
user = Table('transaktion', metadata, autoload=True)
current_time = datetime.timedelta(weeks=date)
s = select(['id','user']).where(trans.c.time > current_time)
result = db.execute(s)
for a in result:
print(a)
if __name__ == "__main__":
x = 0
name = ''
list = []
userbarcode = ''
while x != int('01'):
screen = curses.initscr()
@ -102,13 +131,29 @@ while x != int('01'):
screen.addstr(4, 4, pro.name)
list.append(x)
elif isuser(x):
name = user(x)
foo = user(x)
name = foo.name
userbarcode = foo.barcode
screen.addstr(4, 4, "Du wurdest eingeloggt")
elif logout(x):
if userbarcode:
db = create_engine(db_uri)
metadata = MetaData(db)
metadata.reflect()
trans = Table('transaktion', metadata, autoload=True)
trans_o = trans.insert()
for a in list:
foo = return_price(a)
db.execute(trans_o,
produkt=foo.barcode,
user=userbarcode
)
name = ''
list = []
screen.addstr(2, 30, name)
preis = 0
y = 6
@ -123,9 +168,9 @@ while x != int('01'):
screen.addstr(y + 2, 30, str(preis))
k = screen.getstr(3, 3, 60).decode('utf-8').rstrip()
if k != '':
try:
x = int(k)
else:
except:
x = 0
curses.endwin()