from flask import Blueprint, request, jsonify from datetime import datetime import geruecht.controller.userController as uc from geruecht.decorator import login_required from geruecht.model import MONEY, GASTRO import time vorstand = Blueprint("vorstand", __name__) userController = uc.UserController() @vorstand.route("/sm/addUser", methods=['POST', 'GET']) @login_required(groups=[MONEY, GASTRO]) def _addUser(**kwargs): if request.method == 'GET': return "

HEllo World

" data = request.get_json() user = data['user'] day = data['day'] month = data['month'] year = data['year'] date = datetime(year,month,day,12) retVal = userController.addWorker(user['username'], date) print(retVal) return jsonify(retVal) @vorstand.route("/sm/getUser", methods=['POST']) @login_required(groups=[MONEY, GASTRO]) def _getUser(**kwargs): data = request.get_json() day = data['day'] month = data['month'] year = data['year'] date = datetime(year, month, day, 12) lockedDay = userController.getLockedDay(date) if not lockedDay: lockedDay = { 'date': { 'year': year, 'month': month, 'day': day }, 'locked': False } else: lockedDay = { 'date': { 'year': year, 'month': month, 'day': day }, 'locked': lockedDay['locked'] } retVal = { 'worker': userController.getWorker(date), 'day': lockedDay } print(retVal) return jsonify(retVal) @vorstand.route("/sm/deleteUser", methods=['POST']) @login_required(groups=[MONEY, GASTRO]) def _deletUser(**kwargs): data = request.get_json() user = data['user'] day = data['day'] month = data['month'] year = data['year'] date = datetime(year, month, day, 12) userController.deleteWorker(user['username'], date) return jsonify({"ok": "ok"}) @vorstand.route("/sm/lockDay", methods=['POST']) @login_required(groups=[MONEY, GASTRO]) def _lockDay(**kwargs): try: data = request.get_json() year = data['year'] month = data['month'] day = data['day'] locked = data['locked'] date = datetime(year, month, day, 12) lockedDay = userController.setLockedDay(date, locked, True) if not lockedDay: retVal = { 'date': { 'year': year, 'month': month, 'day': day }, 'locked': False } else: retVal = { 'date': { 'year': year, 'month': month, 'day': day }, 'locked': lockedDay['locked'] } print(retVal) return jsonify(retVal) except Exception as err: return jsonify({'error': err}), 409