90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
from database import Database
|
|
from flask import Flask, abort, request
|
|
from flask.json import jsonify
|
|
from gevent.pywsgi import WSGIServer
|
|
import logging
|
|
|
|
|
|
LOGGER = logging.getLogger(__name__)
|
|
LOGGER.setLevel(logging.DEBUG)
|
|
logFormatter = logging.Formatter(
|
|
"%(asctime)s [%(threadName)s] [%(levelname)s] %(message)s")
|
|
|
|
fileHandler = logging.FileHandler("../logs/server.log")
|
|
fileHandler.setFormatter(logFormatter)
|
|
fileHandler.setLevel(logging.INFO)
|
|
LOGGER.addHandler(fileHandler)
|
|
|
|
consoleHandler = logging.StreamHandler()
|
|
consoleHandler.setLevel(logging.DEBUG)
|
|
LOGGER.addHandler(consoleHandler)
|
|
|
|
app = Flask(__name__)
|
|
DATABASE = Database()
|
|
|
|
|
|
@app.route('/')
|
|
def index():
|
|
return "<h1>Hello, World!</h>", 200
|
|
|
|
|
|
@app.route('/scan2kasse/login')
|
|
def login():
|
|
if not request.json or 'login' not in request.json:
|
|
abort(400)
|
|
if not DATABASE.get_user(login = request.json['login']):
|
|
abort(403)
|
|
return jsonify({}), 200
|
|
|
|
|
|
@app.route('/scan2kasse/insert', methods=['POST'])
|
|
def insert():
|
|
match request.json:
|
|
case {'user': user, 'items': items, 'date': date}:
|
|
failed = DATABASE.insert_bought_items(user, items, date)
|
|
if failed:
|
|
return jsonify(failed), 400
|
|
return jsonify({'inserted': True}), 201
|
|
case {'user': user, 'items': items}:
|
|
failed = DATABASE.insert_bought_items(user, items)
|
|
if failed:
|
|
return jsonify(failed), 400
|
|
return jsonify({'inserted': True}), 201
|
|
case _:
|
|
abort(400)
|
|
|
|
@app.route('/scan2kasse/<string:user>', methods=['GET'])
|
|
@app.route('/scan2kasse/<int:year>/<int:month>', methods=['GET'])
|
|
@app.route('/scan2kasse/<string:user>/<int:year>', methods=['GET'])
|
|
@app.route('/scan2kasse/<string:user>/<int:year>/<int:month>', methods=['GET'])
|
|
def get_monthly_report_from_user(user: str = None, year: int = None, month: int = None):
|
|
if month and (month > 12 or month < 1):
|
|
abort(400)
|
|
LOGGER.info("Getting results.")
|
|
results = DATABASE.get_report(user=user, year=year, month=month)
|
|
LOGGER.debug(f"Results received: {results}")
|
|
if results:
|
|
result_dict = group_results(results)
|
|
else:
|
|
result_dict = {}
|
|
return jsonify(result_dict)
|
|
|
|
|
|
def group_results(results: tuple) -> dict:
|
|
result_dict = {}
|
|
LOGGER.debug("Grouping...")
|
|
for result in results:
|
|
if result[0] not in result_dict:
|
|
result_dict[result[0]] = {}
|
|
if str(result[1]) not in result_dict[result[0]]:
|
|
result_dict[result[0]][str(result[1])] = {}
|
|
result_dict[result[0]][str(result[1])][result[2]] = (
|
|
result[3], result[4])
|
|
LOGGER.debug("Grouped.")
|
|
return result_dict
|
|
|
|
|
|
if __name__ == '__main__':
|
|
http_server = WSGIServer(('', 5000), app)
|
|
http_server.serve_forever()
|