119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
from flask import Blueprint, request
|
||
|
||
from core.error import DataExist, InputError, NoData
|
||
from core.item import ItemFactory
|
||
from core.redeem import Redeem
|
||
from core.sql import Connect, Query, Sql
|
||
|
||
from .api_auth import api_try, request_json_handle, role_required
|
||
from .api_code import success_return
|
||
from .constant import Constant
|
||
|
||
|
||
bp = Blueprint('redeems', __name__, url_prefix='/redeems')
|
||
|
||
|
||
@bp.route('', methods=['GET'])
|
||
@role_required(request, ['select'])
|
||
@request_json_handle(request, optional_keys=Constant.QUERY_KEYS)
|
||
@api_try
|
||
def redeems_get(data, user):
|
||
'''查询全redeem信息'''
|
||
with Connect() as c:
|
||
query = Query(['code', 'type'], ['code'], ['code']).from_dict(data)
|
||
x = Sql(c).select('redeem', query=query)
|
||
r = [Redeem().from_list(i) for i in x]
|
||
|
||
if not r:
|
||
raise NoData(api_error_code=-2)
|
||
|
||
return success_return([x.to_dict(has_items=False) for x in r])
|
||
|
||
|
||
@bp.route('', methods=['POST'])
|
||
@role_required(request, ['insert'])
|
||
@request_json_handle(request, required_keys=['code', 'type'], optional_keys=['items'])
|
||
@api_try
|
||
def redeems_post(data, user):
|
||
'''添加redeem，注意可以有items，不存在的item会自动创建'''
|
||
with Connect() as c:
|
||
r = Redeem(c).from_dict(data)
|
||
if r.select_exists():
|
||
raise DataExist(
|
||
f'redeem `{r.code}` already exists')
|
||
r.insert_all()
|
||
return success_return(r.to_dict(has_items='items' in data))
|
||
|
||
|
||
@bp.route('/<string:code>', methods=['GET'])
|
||
@role_required(request, ['select'])
|
||
@api_try
|
||
def redeems_redeem_get(user, code: str):
|
||
'''查询单个redeem信息'''
|
||
with Connect() as c:
|
||
r = Redeem(c).select(code)
|
||
r.select_items()
|
||
return success_return(r.to_dict())
|
||
|
||
|
||
@bp.route('/<string:code>', methods=['DELETE'])
|
||
@role_required(request, ['delete'])
|
||
@api_try
|
||
def redeems_redeem_delete(user, code: str):
|
||
'''删除redeem，会连带删除redeem_item'''
|
||
with Connect() as c:
|
||
Redeem(c).select(code).delete_all()
|
||
return success_return()
|
||
|
||
|
||
@bp.route('/<string:code>', methods=['PUT'])
|
||
@role_required(request, ['change'])
|
||
@request_json_handle(request, optional_keys=['type'], must_change=True)
|
||
@api_try
|
||
def redeems_redeem_put(data, user, code: str):
|
||
'''更新redeem信息，注意不能有items'''
|
||
with Connect() as c:
|
||
r = Redeem(c).select(code)
|
||
if 'type' in data:
|
||
r.redeem_type = int(data['type'])
|
||
r.update()
|
||
return success_return(r.to_dict(has_items=False))
|
||
|
||
|
||
@bp.route('/<string:code>/items', methods=['GET'])
|
||
@role_required(request, ['select'])
|
||
@api_try
|
||
def redeems_redeem_items_get(user, code: str):
|
||
'''查询redeem的items'''
|
||
with Connect() as c:
|
||
r = Redeem(c)
|
||
r.code = code
|
||
r.select_items()
|
||
return success_return([x.to_dict(has_is_available=True) for x in r.items])
|
||
|
||
|
||
@bp.route('/<string:code>/items', methods=['PATCH'])
|
||
@role_required(request, ['change'])
|
||
@request_json_handle(request, is_batch=True)
|
||
@api_try
|
||
def redeems_redeem_items_patch(data, user, code: str):
|
||
'''增删改单个redeem的items'''
|
||
with Connect() as c:
|
||
r = Redeem(c)
|
||
r.code = code
|
||
r.select_items()
|
||
r.remove_items([ItemFactory.from_dict(x, c=c)
|
||
for x in data.get('remove', [])])
|
||
r.add_items([ItemFactory.from_dict(x, c=c)
|
||
for x in data.get('create', [])])
|
||
|
||
updates = data.get('update', [])
|
||
for x in updates:
|
||
if 'amount' not in x:
|
||
raise InputError('`amount` is required in `update`')
|
||
if not isinstance(x['amount'], int) or x['amount'] <= 0:
|
||
raise InputError(
|
||
'`amount` must be a positive integer', api_error_code=-101)
|
||
|
||
r.update_items([ItemFactory.from_dict(x, c=c) for x in updates])
|
||
return success_return([x.to_dict(has_is_available=True) for x in r.items])
|