piket/piket_server/routes/aardbei.py

from typing import Any, Dict, List, Tuple, Union

from flask import request

from piket_server.aardbei_sync import (
    AARDBEI_ENDPOINT,
    ActivityId,
    get_activity,
    AardbeiLink,
    AardbeiSyncError,
    create_missing,
    get_aardbei_people,
    match_activity,
    get_activities,
    link_matches,
    match_local_aardbei,
    update_names,
)
from piket_server.flask import app, db


def common_prepare_aardbei_sync(
    token: str, endpoint: str
) -> Union[AardbeiSyncError, AardbeiLink]:
    aardbei_people = get_aardbei_people(token, endpoint)

    if isinstance(aardbei_people, AardbeiSyncError):
        return aardbei_people

    aardbei_activities = get_activities(token, endpoint)

    if isinstance(aardbei_activities, AardbeiSyncError):
        return aardbei_activities

    return match_local_aardbei(aardbei_people)


@app.route("/people/aardbei_diff", methods=["POST"])
def aardbei_diff() -> Tuple[Dict[str, Any], int]:
    data: Dict[str, str] = request.json
    link = common_prepare_aardbei_sync(
        data["token"], data.get("endpoint", AARDBEI_ENDPOINT)
    )

    if isinstance(link, AardbeiSyncError):
        return {"error": link.value}, 503

    return (
        {
            "num_changes": link.num_changes,
            "new_people": [member.person.full_name for member in link.remote_only],
            "link_existing": [match.local.name for match in link.matches],
            "altered_name": [match.local.name for match in link.matches],
        },
        200,
    )


@app.route("/people/aardbei_apply", methods=["POST"])
def aardbei_apply() -> Union[Tuple[Dict[str, Any], int]]:
    data: Dict[str, str] = request.json
    link = common_prepare_aardbei_sync(
        data["token"], data.get("endpoint", AARDBEI_ENDPOINT)
    )

    if isinstance(link, AardbeiSyncError):
        return {"error": link.value}, 503

    link_matches(link.matches)
    create_missing(link.remote_only)
    update_names(link.altered_name)

    db.session.commit()

    return (
        {
            "num_changes": link.num_changes,
            "new_people": [member.person.full_name for member in link.remote_only],
            "link_existing": [match.local.name for match in link.matches],
            "altered_name": [match.local.name for match in link.matches],
        },
        200,
    )


@app.route("/aardbei/get_activities", methods=["POST"])
def aardbei_get_activities() -> Tuple[Dict[str, object], int]:
    data: Dict[str, str] = request.json
    activities = get_activities(data["token"], data.get("endpoint", AARDBEI_ENDPOINT))

    if isinstance(activities, AardbeiSyncError):
        return {"error": activities.value}, 503

    return {"activities": [x.as_json_dict for x in activities]}, 200


@app.route("/aardbei/apply_activity", methods=["POST"])
def aardbei_apply_activity() -> Tuple[Dict[str, Any], int]:
    data: Dict[str, Union[str, int]] = request.json
    aid = data["activity_id"]
    token = data["token"]
    endpoint = data["endpoint"]

    if not isinstance(aid, int):
        return {"error": "nonnumeric_activity_id"}, 400

    if not isinstance(token, str):
        return {"error": "illtyped_token"}, 400

    if not isinstance(endpoint, str):
        return {"error": "illtyped_endpoint"}, 400

    activity = get_activity(activity_id=ActivityId(aid), token=token, endpoint=endpoint)

    if isinstance(activity, AardbeiSyncError):
        return {"error": activity.value}, 503

    match_activity(activity)
    db.session.commit()

    return (activity.as_json_dict, 200)