from flask import Blueprint, request, jsonify import geruecht.controller as gc import geruecht.controller.userController as uc import geruecht.controller.accesTokenController as ac from geruecht.model import USER from datetime import datetime user = Blueprint("user", __name__) accesTokenController = ac.AccesTokenController() userController = uc.UserController() @user.route("/user/main") def _main(): token = request.headers.get("Token") accToken = accesTokenController.validateAccessToken(token, [USER]) if accToken: accToken.user = userController.getUser(accToken.user.uid) retVal = accToken.user.toJSON() retVal['creditList'] = {credit.year: credit.toJSON() for credit in accToken.user.geruechte} return jsonify(retVal) return jsonify({"error": "permission denied"}), 401 @user.route("/user/addAmount", methods=['POST']) def _addAmount(): token = request.headers.get("Token") accToken = accesTokenController.validateAccessToken(token, [USER]) if accToken: data = request.get_json() amount = int(data['amount']) date = datetime.now() userController.addAmount(accToken.user.uid, amount, year=date.year, month=date.month) accToken.user = userController.getUser(accToken.user.uid) retVal = accToken.user.toJSON() retVal['creditList'] = {credit.year: credit.toJSON() for credit in accToken.user.geruechte} return jsonify(retVal) return jsonify({"error": "permission denied"}), 401