39 lines
1.5 KiB
Python
39 lines
1.5 KiB
Python
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 |