first commit
This commit is contained in:
@@ -0,0 +1,82 @@
|
||||
from flask import jsonify, request
|
||||
from flask_jwt_extended import jwt_required, get_jwt_identity
|
||||
from . import api_bp
|
||||
from ..models import Workspace
|
||||
from ..schemas import WorkspaceSchema, WorkspaceCreateSchema, WorkspaceUpdateSchema
|
||||
from ..services import WorkspaceService
|
||||
|
||||
|
||||
workspace_schema = WorkspaceSchema()
|
||||
workspace_create_schema = WorkspaceCreateSchema()
|
||||
workspace_update_schema = WorkspaceUpdateSchema()
|
||||
|
||||
|
||||
@api_bp.route("/workspaces", methods=["GET"])
|
||||
@jwt_required()
|
||||
def get_workspaces():
|
||||
user_id = get_jwt_identity()
|
||||
workspaces = Workspace.query.filter_by(user_id=user_id).all()
|
||||
return jsonify(workspace_schema.dump(workspaces, many=True))
|
||||
|
||||
|
||||
@api_bp.route("/workspaces", methods=["POST"])
|
||||
@jwt_required()
|
||||
def create_workspace():
|
||||
user_id = get_jwt_identity()
|
||||
data = request.get_json()
|
||||
errors = workspace_create_schema.validate(data)
|
||||
if errors:
|
||||
return jsonify({"error": errors}), 400
|
||||
|
||||
try:
|
||||
workspace = WorkspaceService.create_workspace(
|
||||
user_id, data["name"], data.get("description")
|
||||
)
|
||||
return jsonify(workspace_schema.dump(workspace)), 201
|
||||
except ValueError as e:
|
||||
return jsonify({"error": str(e)}), 400
|
||||
|
||||
|
||||
@api_bp.route("/workspaces/<int:workspace_id>", methods=["GET"])
|
||||
@jwt_required()
|
||||
def get_workspace(workspace_id):
|
||||
user_id = get_jwt_identity()
|
||||
workspace = Workspace.query.filter_by(id=workspace_id, user_id=user_id).first()
|
||||
if not workspace:
|
||||
return jsonify({"error": "工作空间不存在"}), 404
|
||||
return jsonify(workspace_schema.dump(workspace))
|
||||
|
||||
|
||||
@api_bp.route("/workspaces/<int:workspace_id>", methods=["PUT"])
|
||||
@jwt_required()
|
||||
def update_workspace(workspace_id):
|
||||
user_id = get_jwt_identity()
|
||||
workspace = Workspace.query.filter_by(id=workspace_id, user_id=user_id).first()
|
||||
if not workspace:
|
||||
return jsonify({"error": "工作空间不存在"}), 404
|
||||
|
||||
data = request.get_json()
|
||||
errors = workspace_update_schema.validate(data)
|
||||
if errors:
|
||||
return jsonify({"error": errors}), 400
|
||||
|
||||
try:
|
||||
updated_workspace = WorkspaceService.update_workspace(workspace, data)
|
||||
return jsonify(workspace_schema.dump(updated_workspace))
|
||||
except ValueError as e:
|
||||
return jsonify({"error": str(e)}), 400
|
||||
|
||||
|
||||
@api_bp.route("/workspaces/<int:workspace_id>", methods=["DELETE"])
|
||||
@jwt_required()
|
||||
def delete_workspace(workspace_id):
|
||||
user_id = get_jwt_identity()
|
||||
workspace = Workspace.query.filter_by(id=workspace_id, user_id=user_id).first()
|
||||
if not workspace:
|
||||
return jsonify({"error": "工作空间不存在"}), 404
|
||||
|
||||
try:
|
||||
WorkspaceService.delete_workspace(workspace)
|
||||
return jsonify({"message": "工作空间已删除"})
|
||||
except ValueError as e:
|
||||
return jsonify({"error": str(e)}), 400
|
||||
Reference in New Issue
Block a user