mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
from flask import (
|
|
Blueprint,
|
|
jsonify,
|
|
request
|
|
)
|
|
|
|
from sqlalchemy.exc import DataError
|
|
from sqlalchemy.orm.exc import NoResultFound
|
|
|
|
from app.dao.jobs_dao import (
|
|
save_job,
|
|
get_job,
|
|
get_jobs_by_service
|
|
)
|
|
|
|
from app.schemas import (
|
|
job_schema,
|
|
jobs_schema
|
|
)
|
|
|
|
job = Blueprint('job', __name__, url_prefix='/service/<int:service_id>/job')
|
|
|
|
|
|
@job.route('/<job_id>', methods=['GET'])
|
|
@job.route('', methods=['GET'])
|
|
def get_job_for_service(service_id, job_id=None):
|
|
if job_id:
|
|
try:
|
|
job = get_job(service_id, job_id)
|
|
data, errors = job_schema.dump(job)
|
|
return jsonify(data=data)
|
|
except DataError:
|
|
return jsonify(result="error", message="Invalid job id"), 400
|
|
except NoResultFound:
|
|
return jsonify(result="error", message="Job not found"), 404
|
|
else:
|
|
jobs = get_jobs_by_service(service_id)
|
|
data, errors = jobs_schema.dump(jobs)
|
|
return jsonify(data=data)
|
|
|
|
|
|
@job.route('', methods=['POST'])
|
|
def create_job(service_id):
|
|
job, errors = job_schema.load(request.get_json())
|
|
if errors:
|
|
return jsonify(result="error", message=errors), 400
|
|
try:
|
|
save_job(job)
|
|
except Exception as e:
|
|
return jsonify(result="error", message=str(e)), 500
|
|
return jsonify(data=job_schema.dump(job).data), 201
|