feat: enhance media API with structured responses and error handling; add pagination support

This commit is contained in:
ethan.chen
2025-05-19 17:36:33 +08:00
parent c8e8becffc
commit ca22da355d

102
app.py
View File

@@ -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)