#!/usr/bin/env python

# vim: set fileencoding=utf-8

import models as m

import shared

import string

from decimal import Decimal

from datetime import date, datetime

def get_date_and_shop(session):

return datetime.strptime(d, '%d-%m-%Y'), sklep

if __name__ == '__main__':

m.init_model()

m.meta.create_all(m.meta.engine)

session = m.Session()

(data, sklep) = get_date_and_shop(session)

while True:

try:

kategoria = shared.choice('Wybierz kategorię: ', session, m.Kategoria, shared.enter_new(m.Kategoria))

podkategoria = shared.choice(

'Wybierz podkategorię: ',

session,

m.Podkategoria,

shared.enter_new(m.Podkategoria, kategoria),

query_processor = lambda x: x.filter(m.Podkategoria.kategoria_id == kategoria.id)

)

produkt = shared.choice(

'Wybierz produkt: ',

session,

m.Produkt,

shared.enter_new(m.Produkt, podkategoria),

query_processor = lambda x: x.filter(m.Produkt.podkategoria_id == podkategoria.id).order_by(m.Produkt.nazwa)

)

cena = shared.get_answer(

'Cena: ',

None,

'[0-9]{0,5}(\.|,)[0-9]{0,2}'

)

cena = string.replace(cena, ',', '.')

ilosc = shared.get_answer(

'Ilosc',

'1',

'([1-9][0-9]*)?(\.|,)?[0-9]*'

)

ilosc = string.replace(ilosc, ',', '.')

zakup = m.Zakup(data, sklep, produkt, Decimal(cena), Decimal(ilosc))

session.add(zakup)

moar = shared.get_answer(

'Dalej?',

't',

't|n'

)

if moar is 'n':

break

except KeyboardInterrupt:

break

session.commit()