from datetime import datetime from time import sleep import escpos.printer from flask import Flask, redirect, request, render_template, flash from flask_login import LoginManager, login_user, logout_user, login_required from pos.config import Config from pos.logging import init_logging, get_logger from pos.database import Database, User, Product, Event, Transaction, Order config = Config() debug = config.core['GENERAL'].getboolean('Debug', False) conf_db = config.core['DATABASE'] conf_printer = config.core['PRINTER'] conf_flask = config.core['FLASK'] init_logging(config.logging) log = get_logger('web') printer = escpos.printer.Network( conf_printer['Host'], port=conf_printer.getint('Port'), timeout=15) db = Database(**conf_db) app = Flask(__name__) app.config.update( debug=debug, SECRET_KEY=conf_flask['SECRET_KEY'] ) login_manager = LoginManager() login_manager.init_app(app) login_manager.login_view = 'login' login_manager.login_message_category = 'error' @login_manager.user_loader def load_user(uid): user = None with db.get_session() as session: user = session.query(User).get(uid) return user @app.route('/') def home(): return redirect('/sell') @app.route('/login', methods=['GET']) def login_page(): return render_template('login.html', title='Login') @app.route('/login', methods=['POST']) def login(): username = request.form['username'] password = request.form['password'] if not all([username, password]): flash("Fill all fields.", 'error') return login_page(), 400 with db.get_session() as session: user = session.query(User)\ .filter(User.username == username)\ .filter(User.is_active == 1)\ .one_or_none() if user is None: flash("User not found.", 'error') return login_page(), 400 if user.password == password: login_user(user) session.add(user) user.is_authenticated = True flash("Succesfully logged in.", 'success') return redirect('/') else: flash("Wrong password.", 'error') return login_page(), 400 @app.route('/logout') def logout(): logout_user() return redirect('/') def get_products(session): return session.query(Product)\ .filter(Product.is_active == 1)\ .order_by(Product.sort.asc())\ .all() def get_event(session): return session.query(Event)\ .filter(Event.starts_at < datetime.now())\ .filter((Event.ends_at.is_(None)) | (Event.ends_at > datetime.now()))\ .one_or_none() @app.route('/sell', methods=['GET']) @login_required def sell_page(): with db.get_session() as session: products = get_products(session) event = get_event(session) return render_template('sell.html', title='Sell', event=event, products=products) def print_orders(event, orders): printer.open() printer.set(align='CENTER') printer.image('static/img/macao-logo-printer.png', impl='bitImageColumn') printer.set(align='CENTER', text_type='B') printer.text(event.name.upper()) printer.text("\n\n") for o in orders: printer.set(align='LEFT', width=2, height=2) printer.text("{} x {}".format(o.quantity, o.product.name.upper())) printer.text("\n") printer.text("\n") printer.cut() printer.close() sleep(0.7) def print_transaction(transaction): categorized_orders = {} for o in transaction.orders: uid = o.product.category_uid if not categorized_orders.get(uid, False): categorized_orders[uid] = [] categorized_orders[uid].append(o) for cat, orders in categorized_orders.items(): print_orders(transaction.event, orders) @app.route('/sell', methods=['POST']) @login_required def sell(): with db.get_session() as session: event = get_event(session) if event is None: flash("Can't perform order without an ongoing event!", 'error') return sell_page() quantities = request.form.values() if not any(int(qty) > 0 for qty in quantities): flash("Can't perform empty order!", 'error') return sell_page() with db.get_session() as session: items = request.form.items() orders = [ Order(product_uid=uid, quantity=qty) for uid, qty in items if int(qty) > 0 ] transaction = Transaction(event_uid=event.uid, orders=orders) session.add(transaction) session.flush() print_transaction(transaction) flash("Success!", 'success') return sell_page() if __name__ == '__main__': app.run(host='0.0.0.0', port=8080, debug=debug)