matemat/preisanzeige/main.py

#!/usr/bin/env python3
__author__ = 'progamler'

from os import system
import curses
from sqlalchemy import Table, Column, Integer, String, MetaData, ForeignKey, create_engine
from sqlalchemy.sql import select

#produkte = Table('produkte', metadata,
 #   Column('barcode', Integer, primary_key=True),
 #   Column('name', String(40)),
 #    Column('preis', Integer),)


def create_db():
    db = create_engine('sqlite:///./db.sqlite')
    metadata = MetaData()
    produkte = Table('produkte', metadata,
    Column('barcode', Integer, primary_key=True),
    Column('name', String(40)),
    Column('preis', Integer),)
    users = 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')))

    metadata.create_all(db)

    i = produkte.insert()
    db.execute(i, name='Wasser', preis=0.5, barcode='42600370288')
    db.execute(i, name='Over 9000', preis=9000.01, barcode='2')
    db.execute(i, name='l33t', preis=1337, barcode='3')
    db.execute(i, name='Clubmate 0.5', preis=1.5, barcode='4029764001807')

    u = users.insert()
    db.execute(u, barcode='23', name='proGamler')


def return_price(produkt):
    db = create_engine('sqlite:///./db.sqlite')
    metadata = MetaData(db)
    metadata.reflect()

    produkte = Table('produkte', metadata, autoload=True)
    s = select(['name', 'preis']).where(produkte.c.barcode == produkt)

    result = db.execute(s)
    return result.fetchone()

def isproduct(code):
    foo = return_price(code)
    try :
        print(foo.name)
        return True
    except:
        return False

def user(code):
    db = create_engine('sqlite:///./db.sqlite')
    metadata = MetaData(db)
    metadata.reflect()

    user = Table('user', metadata, autoload=True)
    s = select(['name']).where(user.c.barcode == code)

    result = db.execute(s)
    x = result.fetchone()
    return x.name
def isuser(code):
    try:
        name = user(code)
        return True
    except:
        return False
def logout(x):
    if x == 666:
        return True
    else:
        return False

x = 0
name = ''
list = []
while x != int('01'):
    screen = curses.initscr()

    screen.clear()
    screen.border(0)
    screen.addstr(2, 2, "Please Scan Code")
    screen.refresh()
    if x != 0:
        if isproduct(x):
            pro = return_price(x)
            screen.addstr(4, 4, pro.name)
            list.append(x)
        elif isuser(x):
            name = user(x)
            screen.addstr(4, 4, "Du wurdest eingeloggt")
        elif logout(x):
            name = ''
            list = []


    screen.addstr(2,30, name)
    preis = 0
    y = 6
    for a in list:
        foo = return_price(a)
        screen.addstr(y,30, str(foo.preis))
        screen.addstr(y,4, foo.name)
        y = y+1
        preis = preis+foo.preis
    screen.addstr(y+1,4, "----------------------------------------------")
    screen.addstr(y+2,4, "Gesamtsumme:")
    screen.addstr(y+2,30, str(preis))

    k = screen.getstr(3,3,60).decode('utf-8').rstrip()
    if k != '':
        x = int(k)
    else:
        x = 0






curses.endwin()