BrickTracker/bricktracker/views/set.py

import logging

from flask import (
    Blueprint,
    jsonify,
    render_template,
    redirect,
    request,
    url_for,
)
from flask_login import login_required
from werkzeug.wrappers.response import Response

from .exceptions import exception_handler
from ..minifigure import BrickMinifigure
from ..part import BrickPart
from ..set import BrickSet
from ..set_checkbox_list import BrickSetCheckboxList
from ..set_list import BrickSetList

logger = logging.getLogger(__name__)

set_page = Blueprint('set', __name__, url_prefix='/sets')


# List of all sets
@set_page.route('/', methods=['GET'])
@exception_handler(__file__)
def list() -> str:
    return render_template(
        'sets.html',
        collection=BrickSetList().all(),
        brickset_checkboxes=BrickSetCheckboxList().list(),
    )


# Change the status of a checkbox
@set_page.route('/<id>/status/<checkbox_id>', methods=['POST'])
@login_required
@exception_handler(__file__, json=True)
def update_status(*, id: str, checkbox_id: str) -> Response:
    value: bool = request.json.get('value', False)  # type: ignore

    brickset = BrickSet().select_light(id)
    checkbox = BrickSetCheckboxList().get(checkbox_id)

    brickset.update_status(checkbox, value)

    # Info
    logger.info('Set {number} ({id}): status "{status}" changed to "{state}"'.format(  # noqa: E501
        number=brickset.fields.set,
        id=brickset.fields.id,
        status=checkbox.fields.name,
        state=value,
    ))

    return jsonify({'value': value})


# Ask for deletion of a set
@set_page.route('/<id>/delete', methods=['GET'])
@login_required
@exception_handler(__file__)
def delete(*, id: str) -> str:
    return render_template(
        'delete.html',
        item=BrickSet().select_specific(id),
        error=request.args.get('error'),
    )


# Actually delete of a set
@set_page.route('/<id>/delete', methods=['POST'])
@exception_handler(__file__, post_redirect='set.delete')
def do_delete(*, id: str) -> Response:
    brickset = BrickSet().select_light(id)
    brickset.delete()

    # Info
    logger.info('Set {number} ({id}): deleted'.format(
        number=brickset.fields.set,
        id=brickset.fields.id,
    ))

    return redirect(url_for('set.deleted', id=id))


# Set is deleted
@set_page.route('/<id>/deleted', methods=['GET'])
@exception_handler(__file__)
def deleted(*, id: str) -> str:
    return render_template(
        'success.html',
        message='Set "{id}" has been successfuly deleted.'.format(id=id),
    )


# Details of one set
@set_page.route('/<id>/details', methods=['GET'])
@exception_handler(__file__)
def details(*, id: str) -> str:
    return render_template(
        'set.html',
        item=BrickSet().select_specific(id),
        open_instructions=request.args.get('open_instructions'),
        brickset_checkboxes=BrickSetCheckboxList().list(all=True),
    )


# Update the missing pieces of a minifig part
@set_page.route('/<id>/minifigures/<minifigure_id>/parts/<part_id>/missing', methods=['POST'])  # noqa: E501
@login_required
@exception_handler(__file__, json=True)
def missing_minifigure_part(
    *,
    id: str,
    minifigure_id: str,
    part_id: str
) -> Response:
    brickset = BrickSet().select_specific(id)
    minifigure = BrickMinifigure().select_specific(brickset, minifigure_id)
    part = BrickPart().select_specific(
        brickset,
        part_id,
        minifigure=minifigure,
    )

    missing = request.json.get('missing', '')  # type: ignore

    part.update_missing(missing)

    # Info
    logger.info('Set {number} ({id}): updated minifigure ({minifigure}) part ({part}) missing count to {missing}'.format(  # noqa: E501
        number=brickset.fields.set,
        id=brickset.fields.id,
        minifigure=minifigure.fields.fig_num,
        part=part.fields.id,
        missing=missing,
    ))

    return jsonify({'missing': missing})


# Update the missing pieces of a part
@set_page.route('/<id>/parts/<part_id>/missing', methods=['POST'])
@login_required
@exception_handler(__file__, json=True)
def missing_part(*, id: str, part_id: str) -> Response:
    brickset = BrickSet().select_specific(id)
    part = BrickPart().select_specific(brickset, part_id)

    missing = request.json.get('missing', '')  # type: ignore

    part.update_missing(missing)

    # Info
    logger.info('Set {number} ({id}): updated part ({part}) missing count to {missing}'.format(  # noqa: E501
        number=brickset.fields.set,
        id=brickset.fields.id,
        part=part.fields.id,
        missing=missing,
    ))

    return jsonify({'missing': missing})