flaschengeist/geruecht/baruser/routes.py

from flask import Blueprint, request, jsonify
import geruecht.controller as gc
import geruecht.controller.ldapController as lc
import geruecht.controller.userController as uc
from datetime import datetime
from geruecht.model import BAR, MONEY, USER
from geruecht.decorator import login_required

baruser = Blueprint("baruser", __name__)

ldap= lc.LDAPController(gc.ldapConfig['URL'], gc.ldapConfig['dn'])
userController = uc.UserController()


@baruser.route("/bar")
@login_required(groups=[BAR])
def _bar(**kwargs):
    """ Main function for Baruser

        Returns JSON-file with all Users, who hast amounts in this month.

        Returns:
            JSON-File with Users, who has amounts in this month
            or ERROR 401 Permission Denied
    """
    dic = {}
    users = userController.getAllUsersfromDB()
    for user in users:
        geruecht = None
        geruecht = user.getGeruecht(datetime.now().year)
        if geruecht is not None:
            month = geruecht.getMonth(datetime.now().month)
            amount = month[0] - month[1]
            all = geruecht.getSchulden()
            if all != 0:
                if all >= 0:
                    type = 'credit'
                else:
                    type = 'amount'
                dic[user.uid] = {"username": user.uid,
                                    "firstname": user.firstname,
                                    "lastname": user.lastname,
                                    "amount": abs(all),
                                    "locked": user.locked,
                                    "type": type
                                    }
    return jsonify(dic)


@baruser.route("/baradd", methods=['POST'])
@login_required(groups=[BAR])
def _baradd(**kwargs):
    """ Function for Baruser to add amount

        This function added to the user with the posted userID the posted amount.

        Returns:
            JSON-File with userID and the amount
            or ERROR 401 Permission Denied
    """
    data = request.get_json()
    userID = data['userId']
    amount = int(data['amount'])

    date = datetime.now()
    userController.addAmount(userID, amount, year=date.year, month=date.month)
    user = userController.getUser(userID)
    geruecht = user.getGeruecht(year=date.year)
    month = geruecht.getMonth(month=date.month)
    amount = abs(month[0] - month[1])
    all = geruecht.getSchulden()
    if all >= 0:
        type = 'credit'
    else:
        type = 'amount'
    dic = user.toJSON()
    dic['amount'] = abs(all)
    dic['type'] = type

    return jsonify(dic)


@baruser.route("/barGetUsers")
@login_required(groups=[BAR, MONEY])
def _getUsers(**kwargs):
    """ Get Users without amount

        This Function returns all Users, who hasn't an amount in this month.

        Returns:
            JSON-File with Users
            or ERROR 401 Permission Denied
    """
    retVal = {}
    retVal = ldap.getAllUser()
    return jsonify(retVal)

@baruser.route("/bar/storno", methods=['POST'])
@login_required(groups=[BAR])
def _storno(**kwargs):
    """ Function for Baruser to storno amount

            This function added to the user with the posted userID the posted amount.

            Returns:
                JSON-File with userID and the amount
                or ERROR 401 Permission Denied
        """
    data = request.get_json()
    userID = data['userId']
    amount = int(data['amount'])

    date = datetime.now()
    userController.addCredit(userID, amount, year=date.year, month=date.month)
    user = userController.getUser(userID)
    geruecht = user.getGeruecht(year=date.year)
    month = geruecht.getMonth(month=date.month)
    amount = abs(month[0] - month[1])
    all = geruecht.getSchulden()
    if all >= 0:
        type = 'credit'
    else:
        type = 'amount'
    dic = user.toJSON()
    dic['amount'] = abs(all)
    dic['type'] = type

    return jsonify(dic)

@baruser.route("/barGetUser", methods=['POST'])
@login_required(groups=[BAR])
def _getUser(**kwargs):
    data = request.get_json()
    username = data['userId']
    user = userController.getUser(username)
    amount = user.getGeruecht(datetime.now().year).getSchulden()
    if amount >= 0:
        type = 'credit'
    else:
        type = 'amount'

    retVal = user.toJSON()
    retVal['amount'] = amount
    retVal['type'] = type
    return jsonify(retVal)


@baruser.route("/search", methods=['POST'])
@login_required(groups=[BAR, MONEY, USER])
def _search(**kwargs):
    data = request.get_json()
    searchString = data['searchString']
    retVal = ldap.searchUser(searchString)
    return jsonify(retVal)