def get_product(id, type): if (type == "costumes"): return session.query(Kostym)\ .filter(Kostym.id == id).first() elif (type == "accessories"): return session.query(Doplnek)\ .filter(Doplnek.id == id).first()
def get_product_amount(type, id): if (type == "costumes"): return session.query(Kostym.pocet) \ .filter(Kostym.id == id).first() elif (type == "accessories"): return session.query(Doplnek.pocet) \ .filter(Doplnek.id == id).first()
def get_all_orders(): return ( session.query(Vypujcka, Osoba).outerjoin(Osoba, Vypujcka.klient == Osoba.rc).all(), session.query(VypujckaKostym, Kostym).outerjoin(Kostym, VypujckaKostym.kostym_id == Kostym.id).all(), session.query(DoplnekVypujcka, Doplnek)\ .outerjoin(Doplnek, DoplnekVypujcka.doplnek_id == Doplnek.id).all() )
def get_products_data(limit, offset, url): if (url == '/costumes_list'): return (session.query(Kostym).order_by( Kostym.id.asc()).limit(limit).offset(offset).all(), session.query(KostymVyuziti, Vyuziti).outerjoin( Vyuziti, KostymVyuziti.vyuziti_id == Vyuziti.id).all()) else: return (session.query(Doplnek).order_by( Doplnek.id.asc()).limit(limit).offset(offset).all())
def set_product_amount(type, id, amount): if (type == "costumes"): update_costume = session.query(Kostym)\ .filter(Kostym.id == id).first() update_costume.pocet = amount session.commit() elif (type == "accessories"): update_costume = session.query(Doplnek) \ .filter(Doplnek.id == id).first() update_costume.pocet = amount session.commit()
def update_product_amount(id, amount, type): if type == "costume": costume = session.query(Kostym) \ .filter(Kostym.id == id).first() costume.pocet += amount session.add(costume) elif type == "accessory": accessory = session.query(Doplnek) \ .filter(Doplnek.id == id).first() accessory.pocet += amount session.add(accessory) session.commit()
def get_user_profile(email): # TODO pridat objednavky return session.query(Osoba, Zamestnanec.pozice, Klient.clenstvi) \ .outerjoin(Zamestnanec, Osoba.rc == Zamestnanec.osoba_rc) \ .outerjoin(Klient, Osoba.rc == Klient.osoba_rc) \ .filter(Osoba.email == email)\ .first()
def update_users_data(data): user = session.query(Osoba).filter(Osoba.rc == data['rc']).first() user.jmeno = data['jmeno'] user.prijmeni = data['prijmeni'] user.obec = data['obec'] user.ulice = data['ulice'] user.cislo_popisne = data['cislo_popisne'] user.tel_cislo = data['tel_cislo'] user.email = data['email'] session.commit()
def add_or_update_costume(image, *args, **kwargs): id = kwargs.get('id', None) cz_datetime = datetime.strptime(kwargs['datum_vyroby'], '%d.%m.%Y') costume = Kostym.query.filter_by(id=id).first() if not costume: costume = Kostym(nazev=kwargs['nazev'], vyrobce=kwargs['vyrobce'], material=kwargs['material'], popis=kwargs['popis'], velikost=kwargs['velikost'], datum_vyroby=cz_datetime.strftime("%Y-%m-%d"), opotrebeni=kwargs['opotrebeni'], pocet=kwargs['pocet'], cena=kwargs['cena'], obrazek=image, barva=kwargs['barva']) session.add(costume) costume = session.query(Kostym).order_by(Kostym.id.desc()).first() else: costume.nazev = kwargs['nazev'] costume.vyrobce = kwargs['vyrobce'] costume.material = kwargs['material'] costume.popis = kwargs['popis'] costume.velikost = kwargs['velikost'] costume.datum_vyroby = cz_datetime.strftime("%Y-%m-%d") costume.opotrebeni = kwargs['opotrebeni'] costume.pocet = kwargs['pocet'] costume.cena = kwargs['cena'] costume.barva = kwargs['barva'] if image: costume.obrazek = image session.add(costume) session.query(KostymVyuziti).filter_by(kostym_id=costume.id).delete() for usage in kwargs['vyuziti']: session.add(KostymVyuziti(vyuziti_id=usage, kostym_id=costume.id)) session.commit()
def create_order(*args, **kwargs): cz_datetime = datetime.strptime(kwargs['datum_vraceni'], '%d.%m.%Y') stmt = Vypujcka( nazev_akce=kwargs['nazev_akce'], vracen=0, datum_vypujceni=time.strftime("%Y-%m-%d"), datum_vraceni=cz_datetime.strftime("%Y-%m-%d"), klient=current_user.get_id(), zamestnanec=current_user.get_id(), ) session.add(stmt) new_order_id = session.query(Vypujcka).order_by(Vypujcka.id.desc()).first() costumes_occurences = Counter(args[0]['costumes']) for item, occurence in costumes_occurences.items(): stmt = VypujckaKostym(kostym_id=item, vypujcka_id=new_order_id.id, pocet=occurence) costume = session.query(Kostym).filter_by(id=item).first() costume.pocet -= occurence session.add(costume) session.add(stmt) accessories_occurences = Counter(args[0]['accessories']) for item, occurence in accessories_occurences.items(): stmt = DoplnekVypujcka(doplnek_id=item, vypujcka_id=new_order_id.id, pocet=occurence) session.add(stmt) accessoory = session.query(Doplnek).filter_by(id=item).first() accessoory.pocet -= occurence session.add(accessoory) session.commit()
def update_user(**kwargs): session.query(Klient).filter(Klient.osoba_rc==kwargs['rc']).\ update({'clenstvi': kwargs['membership'][0]}) if kwargs['role'][0] == 'zakaznik': session.query(Zamestnanec).filter( Zamestnanec.osoba_rc == kwargs['rc'][0]).delete() else: if session.query(Zamestnanec).filter( Zamestnanec.osoba_rc == kwargs['rc']).first(): session.query(Zamestnanec).filter(Zamestnanec.osoba_rc == kwargs['rc']). \ update({'pozice': kwargs['role'][0]}) else: stmt = Zamestnanec(osoba_rc=kwargs['rc'][0], pozice=kwargs['role'][0]) session.add(stmt) session.commit()
def delete_order(id): session.query(Vypujcka).filter(Vypujcka.id == id).delete() session.commit()
def get_order_products(id): return (session.query(VypujckaKostym).filter( VypujckaKostym.vypujcka_id == id).all(), session.query(DoplnekVypujcka).filter( DoplnekVypujcka.vypujcka_id == id).all())
def update_order(**kwargs): order = session.query(Vypujcka).filter_by(id=kwargs['id'][0]).first() order.vracen = kwargs['returned'][0] session.commit()
def get_user_orders(rc): return (session.query(Vypujcka).filter(Vypujcka.klient == rc), session.query(VypujckaKostym, Kostym).outerjoin( Kostym, VypujckaKostym.kostym_id == Kostym.id).all(), session.query(DoplnekVypujcka, Doplnek).outerjoin( Doplnek, DoplnekVypujcka.doplnek_id == Doplnek.id).all())
def get_users_data(): return session.query(Osoba, Zamestnanec.pozice, Klient.clenstvi)\ .outerjoin(Zamestnanec, Osoba.rc == Zamestnanec.osoba_rc)\ .outerjoin(Klient, Osoba.rc == Klient.osoba_rc)\ .all()
def get_prize(pruduct, id): if pruduct == "costumes": return session.query(Kostym.cena).filter_by(id=id).first() elif pruduct == "accessories": return session.query(Doplnek.cena).filter_by(id=id).first()
def get_use(use): return session.query(Vyuziti).filter_by(druh_akce=use).first()
def get_uses(): return session.query(Vyuziti).all()
def delete_user(rc): session.query(Osoba).filter(Osoba.rc == rc).delete() session.commit()
def delete_accessory(id): if id: session.query(Doplnek).filter_by(id=id).delete() session.commit()
def get_costume_order(id): return session.query(VypujckaKostym, Vypujcka).join(Vypujcka, Vypujcka.id == VypujckaKostym.vypujcka_id)\ .filter(VypujckaKostym.kostym_id == id).first()
def delete_costume(id): if id: session.query(Kostym).filter_by(id=id).delete() session.commit()
def get_accessories_data(): return (session.query(Doplnek).all())
def get_costumes_data(): return (session.query(Kostym).all())