164 lines
5 KiB
Python
164 lines
5 KiB
Python
from functools import wraps
|
|
from flask import Blueprint, jsonify, abort, request
|
|
from flask_login import current_user
|
|
from db.models import database, OrdersPool, Order, OrderAddOn
|
|
from db.queries import user_get, domsubusers_list, orders_pool_list, orders_pool
|
|
|
|
api = Blueprint('api', __name__)
|
|
|
|
@api.route('/subs')
|
|
def subs():
|
|
return jsonify(
|
|
[
|
|
{"sub_username": dsu.sub.telegram_username}
|
|
for dsu
|
|
in domsubusers_list(current_user.db_user)
|
|
]
|
|
)
|
|
|
|
def authorized_sub(func):
|
|
@wraps(func)
|
|
def wrapper(*args, **kwargs):
|
|
try:
|
|
sub = user_get(request.view_args['username'])
|
|
except:
|
|
abort(403)
|
|
return
|
|
|
|
if sub.telegram_username not in [dsu.sub.telegram_username for dsu in domsubusers_list(current_user.db_user)]:
|
|
abort(403)
|
|
return
|
|
|
|
kwargs['sub'] = sub
|
|
return func(*args, **kwargs)
|
|
return wrapper
|
|
|
|
@api.route('/subs/<username>/sets')
|
|
@authorized_sub
|
|
def sub_order_sets(username, sub):
|
|
return jsonify([
|
|
{
|
|
'id': op.id,
|
|
'name': op.name,
|
|
'scheduled': op.scheduled,
|
|
'time': op.time,
|
|
'weekends': op.weekends,
|
|
'weekdays': op.weekdays,
|
|
'orders': [{
|
|
'id': order.id,
|
|
} for order in op.orders]
|
|
}
|
|
for op
|
|
in orders_pool_list(sub.id)
|
|
])
|
|
|
|
@api.route('/subs/<username>/sets/', methods=['POST'])
|
|
@authorized_sub
|
|
def sub_order_set_create(username, sub):
|
|
# Create new
|
|
with database.atomic() as transaction:
|
|
try:
|
|
new_order_pool = OrdersPool.create(
|
|
user=sub,
|
|
name=request.json['name'],
|
|
scheduled=request.json['scheduled'],
|
|
probability=request.json['probability'] if 'probability' in request.json else None,
|
|
time=request.json['time'] if 'time' in request.json else None,
|
|
weekdays=request.json['weekdays'] if 'weekdays' in request.json else None,
|
|
weekends=request.json['weekends'] if 'weekends' in request.json else None,
|
|
confirm_delay=request.json['confirm_delay'] if 'confirm_delay' in request.json else None,
|
|
)
|
|
|
|
for order_to_create in request.json['orders']:
|
|
new_order = Order.create(
|
|
pool=new_order_pool,
|
|
name=order_to_create['name'],
|
|
weight=order_to_create['weight'],
|
|
repeat=order_to_create['repeat']
|
|
)
|
|
|
|
for add_on_to_create in order_to_create['add_ons']:
|
|
OrderAddOn.create(
|
|
order=new_order,
|
|
name=add_on_to_create['name'],
|
|
probability=add_on_to_create['probability']
|
|
)
|
|
except:
|
|
transaction.rollback()
|
|
abort(500)
|
|
|
|
return jsonify(new_order_pool.to_dict())
|
|
|
|
@api.route('/subs/<username>/sets/<set_id>', methods = ['GET', 'POST'])
|
|
@authorized_sub
|
|
def sub_order_set(username, set_id, sub):
|
|
op = orders_pool(sub.id, set_id)
|
|
|
|
if request.method == 'POST':
|
|
def update_add_ons(order, add_ons):
|
|
for updated_add_on in add_ons:
|
|
if isinstance(updated_add_on['id'], int):
|
|
add_on_to_update = order.add_ons.where(OrderAddOn.id == updated_add_on['id']).get()
|
|
|
|
if '_delete' in updated_add_on:
|
|
add_on_to_update.delete_instance(recursive=True)
|
|
else:
|
|
add_on_to_update.name = updated_add_on['name']
|
|
add_on_to_update.probability = updated_add_on['probability']
|
|
|
|
add_on_to_update.save()
|
|
else:
|
|
OrderAddOn.create(
|
|
order=order,
|
|
name=updated_add_on['name'],
|
|
probability=updated_add_on['probability']
|
|
)
|
|
|
|
with database.atomic() as transaction:
|
|
try:
|
|
op.name = request.json['name']
|
|
|
|
op.scheduled = request.json['scheduled']
|
|
if op.scheduled:
|
|
op.probability = request.json['probability']
|
|
op.weekdays = request.json['weekdays']
|
|
op.weekends = request.json['weekends']
|
|
op.time = request.json['time']
|
|
op.confirm_delay = request.json['confirm_delay']
|
|
else:
|
|
op.probability = None
|
|
op.weekdays = None
|
|
op.weekends = None
|
|
op.time = None
|
|
op.confirm_delay = None
|
|
|
|
op.save()
|
|
|
|
for updated_order in request.json['orders']:
|
|
if isinstance(updated_order['id'], int):
|
|
order_to_update = op.orders.where(Order.id == updated_order['id']).get()
|
|
|
|
if '_delete' in updated_order:
|
|
order_to_update.delete_instance(recursive=True)
|
|
else:
|
|
order_to_update.name = updated_order['name']
|
|
order_to_update.weight = updated_order['weight']
|
|
order_to_update.repeat = updated_order['repeat']
|
|
|
|
order_to_update.save()
|
|
|
|
update_add_ons(order_to_update, updated_order['add_ons'])
|
|
else:
|
|
order_to_update = Order.create(
|
|
pool=op,
|
|
name=updated_order['name'],
|
|
weight=updated_order['weight'],
|
|
repeat=updated_order['repeat']
|
|
)
|
|
|
|
update_add_ons(order_to_update, updated_order['add_ons'])
|
|
except:
|
|
transaction.rollback()
|
|
abort(500)
|
|
|
|
return jsonify(op.to_dict())
|