72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
from flask import jsonify, request
|
|
from flask_jwt_extended import jwt_required
|
|
from . import api_bp
|
|
from ..models import CronJob
|
|
from ..schemas import CronJobSchema, CronJobCreateSchema, CronJobUpdateSchema
|
|
from ..services import CronJobService
|
|
|
|
|
|
cron_job_schema = CronJobSchema()
|
|
cron_job_create_schema = CronJobCreateSchema()
|
|
cron_job_update_schema = CronJobUpdateSchema()
|
|
|
|
|
|
@api_bp.route("/agents/<int:agent_id>/cron-jobs", methods=["GET"])
|
|
@jwt_required()
|
|
def get_cron_jobs(agent_id):
|
|
cron_jobs = CronJob.query.filter_by(agent_id=agent_id).all()
|
|
return jsonify(cron_job_schema.dump(cron_jobs, many=True))
|
|
|
|
|
|
@api_bp.route("/agents/<int:agent_id>/cron-jobs", methods=["POST"])
|
|
@jwt_required()
|
|
def create_cron_job(agent_id):
|
|
data = request.get_json()
|
|
errors = cron_job_create_schema.validate(data)
|
|
if errors:
|
|
return jsonify({"error": errors}), 400
|
|
|
|
try:
|
|
cron_job = CronJobService.create_cron_job(agent_id, data)
|
|
return jsonify(cron_job_schema.dump(cron_job)), 201
|
|
except ValueError as e:
|
|
return jsonify({"error": str(e)}), 400
|
|
|
|
|
|
@api_bp.route("/cron-jobs/<int:cron_job_id>", methods=["GET"])
|
|
@jwt_required()
|
|
def get_cron_job(cron_job_id):
|
|
cron_job = CronJob.query.get(cron_job_id)
|
|
if not cron_job:
|
|
return jsonify({"error": "定时任务不存在"}), 404
|
|
return jsonify(cron_job_schema.dump(cron_job))
|
|
|
|
|
|
@api_bp.route("/cron-jobs/<int:cron_job_id>", methods=["PUT"])
|
|
@jwt_required()
|
|
def update_cron_job(cron_job_id):
|
|
cron_job = CronJob.query.get(cron_job_id)
|
|
if not cron_job:
|
|
return jsonify({"error": "定时任务不存在"}), 404
|
|
|
|
data = request.get_json()
|
|
errors = cron_job_update_schema.validate(data)
|
|
if errors:
|
|
return jsonify({"error": errors}), 400
|
|
|
|
try:
|
|
updated_cron_job = CronJobService.update_cron_job(cron_job, data)
|
|
return jsonify(cron_job_schema.dump(updated_cron_job))
|
|
except ValueError as e:
|
|
return jsonify({"error": str(e)}), 400
|
|
|
|
|
|
@api_bp.route("/cron-jobs/<int:cron_job_id>", methods=["DELETE"])
|
|
@jwt_required()
|
|
def delete_cron_job(cron_job_id):
|
|
try:
|
|
CronJobService.delete_cron_job(cron_job_id)
|
|
return jsonify({"message": "定时任务已删除"})
|
|
except ValueError as e:
|
|
return jsonify({"error": str(e)}), 400
|