Arcaea-server/latest version/server/purchase.py

from time import time

from core.error import ArcError, ItemUnavailable
from core.item import ItemFactory, Stamina6
from core.purchase import Purchase, PurchaseList
from core.redeem import UserRedeem
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('purchase', __name__, url_prefix='/purchase')


@bp.route('/bundle/pack', methods=['GET'])  # 曲包信息
@auth_required(request)
def bundle_pack(user_id):
    with Connect() as c:
        try:
            x = PurchaseList(c, UserOnline(c, user_id)
                             ).select_from_type('pack')
            return success_return(x.to_dict_list())
        except ArcError as e:
            return error_return(e)
    return error_return()


@bp.route('/bundle/single', methods=['GET'])  # 单曲购买信息获取
@auth_required(request)
def get_single(user_id):
    with Connect() as c:
        try:
            x = PurchaseList(c, UserOnline(c, user_id)
                             ).select_from_type('single')
            return success_return(x.to_dict_list())
        except ArcError as e:
            return error_return(e)
    return error_return()


@bp.route('/bundle/bundle', methods=['GET'])  # 捆绑包
def bundle_bundle():
    return success_return([])
    # 感觉上是定死的，没意义啊
    # return success_return([{
    #     "name": "chronicles",
    #     "items": [{
    #         "type": "bundle",
    #         "id": "core"
    #     }, {
    #         "type": "bundle",
    #         "id": "prelude"
    #     }, {
    #         "type": "bundle",
    #         "id": "rei"
    #     }, {
    #         "type": "bundle",
    #         "id": "yugamu"
    #     }, {
    #         "type": "bundle",
    #         "id": "vs"
    #     }],
    #     "orig_price": 2000,
    #     "price": 2000,
    #     "discount_from": 1657152000000,
    #     "discount_to": 1758447999000,
    #     "discount_reason": "chronicle"
    # }])


@bp.route('/me/pack', methods=['POST'])  # 曲包和单曲购买
@auth_required(request)
def buy_pack_or_single(user_id):
    with Connect() as c:
        try:
            if 'pack_id' in request.form:
                purchase_name = request.form['pack_id']
            elif 'single_id' in request.form:
                purchase_name = request.form['single_id']
            else:
                return success_return()

            x = Purchase(c, UserOnline(c, user_id)).select(purchase_name)
            x.buy()

            return success_return({
                'user_id': x.user.user_id,
                'ticket': x.user.ticket,
                'packs': x.user.packs,
                'singles': x.user.singles,
                'characters': x.user.characters_list
            })
        except ArcError as e:
            return error_return(e)
    return error_return()


@bp.route('/me/item', methods=['POST'])  # 特殊购买，world模式boost和stamina
@auth_required(request)
def buy_special(user_id):
    with Connect() as c:
        try:
            if 'item_id' not in request.form:
                return error_return()
            item_id = request.form['item_id']

            x = Purchase(c, UserOnline(c, user_id))
            x.purchase_name = item_id
            x.price = 50
            x.orig_price = 50
            x.discount_from = -1
            x.discount_to = -1
            x.items = [ItemFactory(c).get_item(item_id)]
            x.buy()

            r = {'user_id': x.user.user_id, 'ticket': x.user.ticket}
            if item_id == 'stamina6':
                r['stamina'] = x.user.stamina.stamina
                r['max_stamina_ts'] = x.user.stamina.max_stamina_ts
                r['world_mode_locked_end_ts'] = -1
            return success_return(r)
        except ArcError as e:
            return error_return(e)
    return error_return()


@bp.route('/me/stamina/<buy_stamina_type>', methods=['POST'])  # 购买体力
@auth_required(request)
def purchase_stamina(user_id, buy_stamina_type):
    with Connect() as c:
        try:
            if buy_stamina_type != 'fragment':
                return error_return()

            user = UserOnline(c, user_id)
            user.select_user_one_column('next_fragstam_ts', -1)
            now = int(time()*1000)
            if user.next_fragstam_ts > now:
                return ItemUnavailable('Buying stamina by fragment is not available yet.', 905)

            user.update_user_one_column(
                'next_fragstam_ts', now + 24 * 3600 * 1000)
            s = Stamina6(c)
            s.user_claim_item(user)
            return success_return({
                "user_id": user.user_id,
                "stamina": user.stamina.stamina,
                "max_stamina_ts": user.stamina.max_stamina_ts,
                "next_fragstam_ts": user.next_fragstam_ts,
                'world_mode_locked_end_ts': -1
            })
        except ArcError as e:
            return error_return(e)
    return error_return()


@bp.route('/me/redeem', methods=['POST'])  # 兑换码
@auth_required(request)
def redeem(user_id):
    with Connect() as c:
        try:
            x = UserRedeem(c, UserOnline(c, user_id))
            x.claim_user_redeem(request.form['code'])

            return success_return({"coupon": "fragment" + str(x.fragment) if x.fragment > 0 else ""})
        except ArcError as e:
            return error_return(e)
    return error_return()