flaschengeist/geruecht/user/routes.py

from flask import Blueprint, request, jsonify
from geruecht.decorator import login_required
import geruecht.controller.mainController as mc
import geruecht.controller.accesTokenController as ac
from geruecht.model import USER
from datetime import datetime, time, date
from geruecht.exceptions import DayLocked
from geruecht.logger import getDebugLogger, getCreditLogger, getJobsLogger
from geruecht.model.accessToken import AccessToken

user = Blueprint("user", __name__)

mainController = mc.MainController()
accesTokenController = ac.AccesTokenController()

debug = getDebugLogger()
creditL = getCreditLogger()
jobL = getJobsLogger()


@user.route("/user/main")
@login_required(groups=[USER])
def _main(**kwargs):
    debug.info("/user/main")
    try:
        if 'accToken' in kwargs:
            accToken = kwargs['accToken']
            accToken.user = mainController.getUser(accToken.user.uid)
            retVal = accToken.user.toJSON()
            retVal['creditList'] = {credit.year: credit.toJSON()
                                    for credit in accToken.user.geruechte}
            debug.debug("return {{ {} }}".format(retVal))
            return jsonify(retVal)
    except Exception:
        debug.debug("exception", exc_info=True)
        return jsonify("error", "something went wrong"), 500


@user.route("/user/addAmount", methods=['POST'])
@login_required(groups=[USER])
def _addAmount(**kwargs):
    debug.info("/user/addAmount")
    try:
        if 'accToken' in kwargs:
            accToken = kwargs['accToken']
            data = request.get_json()
            amount = int(data['amount'])
            date = datetime.now()
            mainController.addAmount(
                accToken.user.uid, amount, year=date.year, month=date.month)
            accToken.user = mainController.getUser(accToken.user.uid)
            retVal = accToken.user.toJSON()
            retVal['creditList'] = {credit.year: credit.toJSON()
                                    for credit in accToken.user.geruechte}
            debug.debug("return {{ {} }}".format(retVal))
            creditL.info("{} {} {} fügt sich selbst {} € Schulden hinzu".format(
                date, accToken.user.firstname, accToken.user.lastname, amount/100))
            return jsonify(retVal)
    except Exception:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": "something went wrong"}), 500


@user.route("/user/saveConfig", methods=['POST'])
@login_required(groups=[USER])
def _saveConfig(**kwargs):
    debug.info("/user/saveConfig")
    try:
        if 'accToken' in kwargs:
            accToken = kwargs['accToken']
            data = request.get_json()
            accToken.user = mainController.modifyUser(
                accToken.user, accToken.ldap_conn, data)
            retVal = accToken.user.toJSON()
            retVal['creditList'] = {credit.year: credit.toJSON()
                                    for credit in accToken.user.geruechte}
            debug.debug("return {{ {} }}".format(retVal))
            return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 409


@user.route("/user/jobs", methods=['POST'])
@login_required(groups=[USER])
def _getUsers(**kwrags):
    debug.info("/user/jobs")
    try:
        data = request.get_json()
        from_date = data['from_date']
        to_date = data['to_date']
        from_date = datetime(
            from_date['year'], from_date['month'], from_date['day'])
        to_date = datetime(to_date['year'], to_date['month'], to_date['day'])
        lockedDays = mainController.getLockedDays(from_date, to_date)
        retVal = []
        for lockedDay in lockedDays:
            day = datetime.combine(lockedDay['daydate'], time(12))
            retDay = {
                "worker": mainController.getWorker(day),
                "day": {
                    "date": {
                        "year": day.year,
                        "month": day.month,
                        "day": day.day
                    },
                    "locked": lockedDay['locked']
                },
                "jobkinddate": mainController.getJobKindDates(day.date())
            }
            retVal.append(retDay)

        debug.debug("return {{ {} }}".format(retVal))
        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500

@user.route("/user/jobsOnDates", methods=['POST'])
@login_required(groups=[USER])
def _getJobsOnDates(**kwargs):
    debug.info("/user/jobsOnDates")
    try:
        data = request.get_json()
        lockedDays = mainController.getLockedDaysFromList(data)
        retVal = []
        for lockedDay in lockedDays:
            day = datetime.combine(lockedDay['daydate'], time(12))
            retDay = {
                "worker": mainController.getWorker(day),
                "day": {
                    "date": {
                        "year": day.year,
                        "month": day.month,
                        "day": day.day
                    },
                    "locked": lockedDay['locked']
                },
                "jobkinddate": mainController.getJobKindDates(day.date())
            }
            retVal.append(retDay)

        debug.debug("return {{ {} }}".format(retVal))
        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500

@user.route("/user/job", methods=['POST'])
@login_required(groups=[USER])
def _getUser(**kwargs):
    debug.info("/user/job")
    try:
        data = request.get_json()
        day = data['day']
        month = data['month']
        year = data['year']
        date = datetime(year, month, day, 12)
        lockedDay = mainController.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': mainController.getWorker(date),
            'day': lockedDay
        }
        debug.debug("retrun {{ {} }}".format(retVal))
        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500


@user.route("/user/addJob", methods=['POST'])
@login_required(groups=[USER])
def _addUser(**kwargs):
    debug.info("/user/addJob")
    try:
        if 'accToken' in kwargs:
            accToken = kwargs['accToken']
            user = accToken.user
            data = request.get_json()
            day = data['day']
            month = data['month']
            year = data['year']
            date = datetime(year, month, day, 12)
            job_kind = None
            if 'job_kind' in data:
                job_kind = data['job_kind']
            mainController.addWorker(user.uid, date, job_kind=job_kind, userExc=True)
            retVal = mainController.getWorker(date)
            debug.debug("return {{ {} }}".format(retVal))
            jobL.info("Mitglied {} {} schreib sich am {} zum Dienst ein.".format(
                user.firstname, user.lastname, date.date()))
            return jsonify(retVal)
    except DayLocked as err:
        debug.debug("exception", exc_info=True)
        return jsonify({'error': str(err)}), 403
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({'error': str(err)}), 409


@user.route("/user/deleteJob", methods=['POST'])
@login_required(groups=[USER])
def _deletJob(**kwargs):
    debug.info("/user/deleteJob")
    try:
        if 'accToken' in kwargs:
            accToken = kwargs['accToken']
            user = accToken.user
            data = request.get_json()
            day = data['day']
            month = data['month']
            year = data['year']
            date = datetime(year, month, day, 12)
            mainController.deleteWorker(user.uid, date, True)
            retVal = mainController.getWorker(date)
            debug.debug("return ok")
            jobL.info("Mitglied {} {} entfernt sich am {} aus dem Dienst".format(
                user.firstname, user.lastname, date.date()))
            return jsonify(retVal)
    except DayLocked as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 403
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 409

@user.route("/user/storno", methods=['POST'])
@login_required(groups=[USER])
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
        """

    debug.info("/user/storno")
    try:
        if 'accToken' in kwargs:
            accToken = kwargs['accToken']
            user = accToken.user
            data = request.get_json()
            amount = int(data['amount'])

            date = datetime.now()
            mainController.addCredit(
                user.uid, amount, year=date.year, month=date.month)
            accToken.user = mainController.getUser(accToken.user.uid)
            retVal = accToken.user.toJSON()
            retVal['creditList'] = {credit.year: credit.toJSON()
                                    for credit in accToken.user.geruechte}
            debug.debug("return {{ {} }}".format(retVal))
            creditL.info("{} {} {} storniert {} €".format(
                date, user.firstname, user.lastname, amount/100))
            return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500


@user.route("/user/getJobInvites", methods=['POST'])
@login_required(groups=[USER])
def _getJobInvites(**kwargs):
    try:
        debug.info("/user/getJobInvites")
        from_user = None
        to_user = None
        on_date = None

        data = request.get_json()

        if 'from_user' in data:
            from_user = data['from_user']
        if 'to_user' in data:
            to_user = data['to_user']
        on_date = date(data['date']['year'], data['date']['month'], data['date']['day'])
        retVal = mainController.getJobInvites(from_user, to_user, on_date)
        debug.debug("return {{ {} }}".format(retVal))
        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500

@user.route("/user/JobInvites", methods=['PUT', 'POST'])
@login_required(groups=[USER])
def _JobInvites(**kwargs):
    try:
        debug.info("/user/JobInvites")
        data = request.get_json()
        if request.method == 'PUT':
            mainController.setJobInvites(data)
            retVal = mainController.getJobInvites(kwargs['accToken'].user.toJSON(), None, datetime.now().date())
            debug.debug("return {{ {} }}".format(retVal))
        if request.method == 'POST':
            retVal = mainController.updateJobInvites(data)

        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500

@user.route("/user/deleteJobInvite", methods=['POST'])
@login_required(groups=[USER])
def _deleteJobInvite(**kwargs):
    try:
        debug.info("/user/deleteJobInvite")
        data = request.get_json()
        mainController.deleteJobInvite(data)
        retVal = mainController.getJobInvites(data['from_user'], None, datetime.now().date())
        debug.debug("return {{ {} }}".format(retVal))
        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500


@user.route("/user/getJobRequests", methods=['POST'])
@login_required(groups=[USER])
def _getJobRequests(**kwargs):
    try:
        debug.info("/user/getJobRequests")
        from_user = None
        to_user = None
        on_date = None

        data = request.get_json()

        if 'from_user' in data:
            from_user = data['from_user']
        if 'to_user' in data:
            to_user = data['to_user']
        on_date = date(data['date']['year'], data['date']['month'], data['date']['day'])
        retVal = mainController.getJobRequests(from_user, to_user, on_date)
        debug.debug("return {{ {} }}".format(retVal))
        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500

@user.route("/user/JobRequests", methods=['PUT', 'POST'])
@login_required(groups=[USER])
def _JobRequests(**kwargs):
    try:
        debug.info("/user/JobRequests")
        data = request.get_json()
        if request.method == 'PUT':
            mainController.setJobRequests(data)
            retVal = mainController.getJobRequests(kwargs['accToken'].user.toJSON(), None, datetime.now().date())
            debug.debug("return {{ {} }}".format(retVal))
        if request.method == 'POST':
            data['on_date'] = date(data['on_date']['year'], data['on_date']['month'], data['on_date']['day'])
            retVal = mainController.updateJobRequests(data)

        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500

@user.route("/user/deleteJobRequest", methods=['POST'])
@login_required(groups=[USER])
def _deleteJobRequest(**kwargs):
    try:
        debug.info("/user/deleteJobRequest")
        data = request.get_json()
        mainController.deleteJobRequest(data)
        retVal = mainController.getJobRequests(data['from_user'], None, datetime.now().date())
        debug.debug("return {{ {} }}".format(retVal))
        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500


@user.route("/user/getAccessTokens", methods=['GET', 'POST'])
@login_required(groups=[USER])
def _getAccessTokens(**kwargs):
    try:
        debug.info("/user/getAccessTokens")
        if request.method == 'POST':
            data = request.get_json()
            delAccToken = AccessToken(data['id'], kwargs['accToken'].user, None, None, None)
            accesTokenController.deleteAccessToken(delAccToken)
        tokens = accesTokenController.getAccessTokensFromUser(kwargs['accToken'].user)
        retVal = []
        for token in tokens:
            retVal.append(token.toJSON())
        debug.debug("return {{ {} }}".format(retVal))
        return jsonify(retVal)
    except Exception as err:
        debug.debug("exception", exc_info=True)
        return jsonify({"error": str(err)}), 500