feat: enhance media API with structured responses and error handling; add pagination support
This commit is contained in:
102
app.py
102
app.py
@@ -30,10 +30,13 @@ with app.app_context():
|
||||
db.create_all()
|
||||
|
||||
# API 路由
|
||||
@app.route('/api/media', methods=['GET'])
|
||||
@app.route('/api/media/list', methods=['GET'])
|
||||
def get_all_media():
|
||||
try:
|
||||
media_list = Media.query.all()
|
||||
return jsonify([{
|
||||
return jsonify({
|
||||
"code": 0,
|
||||
"data": [{
|
||||
'id': media.id,
|
||||
'title': media.title,
|
||||
'type': media.type,
|
||||
@@ -44,10 +47,15 @@ def get_all_media():
|
||||
'date': media.date.isoformat() if media.date else None,
|
||||
'created_at': media.created_at.isoformat(),
|
||||
'updated_at': media.updated_at.isoformat()
|
||||
} for media in media_list])
|
||||
} for media in media_list],
|
||||
"message": "Success"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"code": 1, "data": {}, "message": str(e)}), 500
|
||||
|
||||
@app.route('/api/media', methods=['POST'])
|
||||
@app.route('/api/media/create', methods=['POST'])
|
||||
def create_media():
|
||||
try:
|
||||
data = request.json
|
||||
new_media = Media(
|
||||
title=data['title'],
|
||||
@@ -61,6 +69,8 @@ def create_media():
|
||||
db.session.add(new_media)
|
||||
db.session.commit()
|
||||
return jsonify({
|
||||
"code": 0,
|
||||
"data": {
|
||||
'id': new_media.id,
|
||||
'title': new_media.title,
|
||||
'type': new_media.type,
|
||||
@@ -71,10 +81,16 @@ def create_media():
|
||||
'date': new_media.date.isoformat() if new_media.date else None,
|
||||
'created_at': new_media.created_at.isoformat(),
|
||||
'updated_at': new_media.updated_at.isoformat()
|
||||
},
|
||||
"message": "Created successfully"
|
||||
}), 201
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
return jsonify({"code": 2, "data": {}, "message": str(e)}), 500
|
||||
|
||||
@app.route('/api/media/<int:media_id>', methods=['PUT'])
|
||||
@app.route('/api/media/updateById/<int:media_id>', methods=['PUT'])
|
||||
def update_media(media_id):
|
||||
try:
|
||||
media = Media.query.get_or_404(media_id)
|
||||
data = request.json
|
||||
|
||||
@@ -89,6 +105,8 @@ def update_media(media_id):
|
||||
|
||||
db.session.commit()
|
||||
return jsonify({
|
||||
"code": 0,
|
||||
"data": {
|
||||
'id': media.id,
|
||||
'title': media.title,
|
||||
'type': media.type,
|
||||
@@ -99,14 +117,84 @@ def update_media(media_id):
|
||||
'date': media.date.isoformat() if media.date else None,
|
||||
'created_at': media.created_at.isoformat(),
|
||||
'updated_at': media.updated_at.isoformat()
|
||||
},
|
||||
"message": "Updated successfully"
|
||||
})
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
return jsonify({"code": 2, "data": {}, "message": str(e)}), 500
|
||||
|
||||
@app.route('/api/media/<int:media_id>', methods=['DELETE'])
|
||||
@app.route('/api/media/deleteById/<int:media_id>', methods=['DELETE'])
|
||||
def delete_media(media_id):
|
||||
try:
|
||||
media = Media.query.get_or_404(media_id)
|
||||
db.session.delete(media)
|
||||
db.session.commit()
|
||||
return '', 204
|
||||
return jsonify({"code": 0, "data": {}, "message": "Deleted successfully"}), 204
|
||||
except Exception as e:
|
||||
db.session.rollback()
|
||||
return jsonify({"code": 2, "data": {}, "message": str(e)}), 500
|
||||
|
||||
@app.route('/api/media/page', methods=['GET'])
|
||||
def get_media_page():
|
||||
try:
|
||||
type = request.args.get('type')
|
||||
current_page = int(request.args.get('currentPage', 1))
|
||||
page_size = int(request.args.get('pageSize', 10))
|
||||
title = request.args.get('title', '')
|
||||
start_date = request.args.get('startDate')
|
||||
end_date = request.args.get('endDate')
|
||||
sort_by = request.args.get('sortBy', 'date')
|
||||
sort_type = request.args.get('sortType', 'desc')
|
||||
|
||||
if not type:
|
||||
return jsonify({"code": 1, "data": {}, "message": "Type is required"}), 400
|
||||
|
||||
query = Media.query.filter(Media.type == type)
|
||||
if title:
|
||||
query = query.filter(Media.title.like(f'%{title}%'))
|
||||
if start_date:
|
||||
query = query.filter(Media.date >= datetime.strptime(start_date, '%Y-%m-%d').date())
|
||||
if end_date:
|
||||
query = query.filter(Media.date <= datetime.strptime(end_date, '%Y-%m-%d').date())
|
||||
|
||||
if sort_by == 'date':
|
||||
if sort_type == 'asc':
|
||||
query = query.order_by(Media.date.asc())
|
||||
else:
|
||||
query = query.order_by(Media.date.desc())
|
||||
elif sort_by == 'score':
|
||||
if sort_type == 'asc':
|
||||
query = query.order_by(Media.rating.asc())
|
||||
else:
|
||||
query = query.order_by(Media.rating.desc())
|
||||
|
||||
pagination = query.paginate(page=current_page, per_page=page_size, error_out=False)
|
||||
media_list = pagination.items
|
||||
|
||||
return jsonify({
|
||||
"code": 0,
|
||||
"data": {
|
||||
"list": [{
|
||||
'id': media.id,
|
||||
'title': media.title,
|
||||
'type': media.type,
|
||||
'status': media.status,
|
||||
'rating': media.rating,
|
||||
'notes': media.notes,
|
||||
'platform': media.platform,
|
||||
'date': media.date.isoformat() if media.date else None,
|
||||
'created_at': media.created_at.isoformat(),
|
||||
'updated_at': media.updated_at.isoformat()
|
||||
} for media in media_list],
|
||||
"total": pagination.total,
|
||||
"currentPage": current_page,
|
||||
"pageSize": page_size
|
||||
},
|
||||
"message": "Success"
|
||||
})
|
||||
except Exception as e:
|
||||
return jsonify({"code": 1, "data": {}, "message": str(e)}), 500
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(debug=True)
|
||||
Reference in New Issue
Block a user