-
Notifications
You must be signed in to change notification settings - Fork 173
/
favorite_product.py
75 lines (54 loc) · 2.39 KB
/
favorite_product.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
from flask import Blueprint, request, current_app
from sqlalchemy import and_
from tblib.model import session
from tblib.handler import json_response, ResponseCode
from ..models import FavoriteProduct, FavoriteProductSchema
favorite_product = Blueprint(
'favorite_product', __name__, url_prefix='/favorite_products')
@favorite_product.route('', methods=['POST'])
def create_favorite_product():
"""收藏商品
"""
data = request.get_json()
favorite_product = FavoriteProductSchema().load(data)
session.add(favorite_product)
session.commit()
return json_response(favorite_product=FavoriteProductSchema().dump(favorite_product))
@favorite_product.route('', methods=['GET'])
def favorite_product_list():
"""收藏的商品列表,可根据用户 ID 和商品 ID 筛选
"""
user_id = request.args.get('user_id', type=int)
product_id = request.args.get('product_id', type=int)
order_direction = request.args.get('order_direction', 'desc')
limit = request.args.get(
'limit', current_app.config['PAGINATION_PER_PAGE'], type=int)
offset = request.args.get('offset', 0, type=int)
order_by = FavoriteProduct.id.asc(
) if order_direction == 'asc' else FavoriteProduct.id.desc()
query = FavoriteProduct.query
if user_id is not None:
query = query.filter(FavoriteProduct.user_id == user_id)
if product_id is not None:
query = query.filter(FavoriteProduct.product_id == product_id)
total = query.count()
query = query.order_by(order_by).limit(limit).offset(offset)
return json_response(favorite_products=FavoriteProductSchema().dump(query, many=True), total=total)
@favorite_product.route('/<int:id>', methods=['DELETE'])
def delete_favorite_product(id):
"""取消收藏商品
"""
favorite_product = FavoriteProduct.query.get(id)
if favorite_product is None:
return json_response(ResponseCode.NOT_FOUND)
session.delete(favorite_product)
session.commit()
return json_response(favorite_product=FavoriteProductSchema().dump(favorite_product))
@favorite_product.route('/<int:id>', methods=['GET'])
def favorite_product_info(id):
"""查询收藏商品
"""
favorite_product = FavoriteProduct.query.get(id)
if favorite_product is None:
return json_response(ResponseCode.NOT_FOUND)
return json_response(favorite_product=FavoriteProductSchema().dump(favorite_product))