from core.error import ArcError
from core.present import UserPresent, UserPresentList
from core.sql import Connect
from core.user import UserOnline
from flask import Blueprint, request

from .auth import auth_required
from .func import error_return, success_return

bp = Blueprint('present', __name__, url_prefix='/present')


@bp.route('/me', methods=['GET'])  # 用户奖励信息
@auth_required(request)
def present_info(user_id):
    with Connect() as c:
        try:
            x = UserPresentList(c, UserOnline(c, user_id))
            x.select_user_presents()

            return success_return(x.to_dict_list())
        except ArcError as e:
            return error_return(e)
    return error_return()


@bp.route('/me/claim/<present_id>', methods=['POST'])  # 礼物确认
@auth_required(request)
def claim_present(user_id, present_id):
    with Connect() as c:
        try:
            x = UserPresent(c, UserOnline(c, user_id))
            x.claim_user_present(present_id)

            return success_return()
        except ArcError as e:
            return error_return(e)
    return error_return()