112 lines
3.3 KiB
Python
112 lines
3.3 KiB
Python
from flask import request, jsonify
|
|
from pony.orm import ObjectNotFound
|
|
from datetime import datetime, timedelta
|
|
from ITPlanning.db import Appointment, Planning, Service, Ticket, Customer
|
|
from ITPlanning.api.exception import NotFound, GenericError
|
|
from . import api
|
|
|
|
|
|
def datefiltering(query):
|
|
before, after = map(request.values.get, ("before", "after"))
|
|
if before:
|
|
before = datetime.strptime(before, "%Y-%m-%dT%H%M")
|
|
query = query.filter(lambda a: a.start_datetime < before)
|
|
if after:
|
|
after = datetime.strptime(after, "%Y-%m-%dT%H%M")
|
|
query = query.filter(lambda a: a.start_datetime > after)
|
|
return query
|
|
|
|
|
|
def search_appointment(by=None, id=None):
|
|
rv = []
|
|
if by == None and id is not None:
|
|
rv.append(Appointment[id].to_dict())
|
|
else:
|
|
query = Appointment.select()
|
|
if by is not None:
|
|
query.filter(lambda a: getattr(a, by).id == id)
|
|
query = datefiltering(query)
|
|
for f in query:
|
|
rv.append(f.to_dict())
|
|
if len(rv) == 0:
|
|
raise NotFound("Appointment")
|
|
return jsonify(rv)
|
|
|
|
|
|
@api.route("/appointment")
|
|
@api.route("/appointment/")
|
|
@api.route("/appointment/<id>")
|
|
def get_appointment(id=None):
|
|
return search_appointment(id=id)
|
|
|
|
|
|
@api.route("/appointment/planning/<id>")
|
|
def get_appointment_by_planning(id=None):
|
|
return search_appointment("planning", id)
|
|
|
|
|
|
@api.route("/appointment/service/<id>")
|
|
def get_appointment_by_service(id):
|
|
return search_appointment("service", id)
|
|
|
|
|
|
@api.route("/appointment/customer/<id>")
|
|
def get_appointment_by_customer(id):
|
|
return search_appointment("customer", id)
|
|
|
|
|
|
@api.route("/appointment/ticket/<id>")
|
|
def get_appointment_by_ticket(id):
|
|
return search_appointment("ticket", id)
|
|
|
|
|
|
@api.route("/appointment", methods=["POST"])
|
|
def new_appointment():
|
|
book_datetime = datetime.now()
|
|
start_datetime = request.values.get("start_datetime")
|
|
service = request.values.get("service")
|
|
planning = request.values.get("planning")
|
|
ticket = request.values.get("ticket")
|
|
customer = request.values.get("customer")
|
|
is_unavaillable = request.values.get("is_unavaillable")
|
|
notes = request.values.get("notes")
|
|
if start_datetime is None or service is None or planning is None:
|
|
return GenericError("missing parameter")
|
|
if is_unavaillable is None:
|
|
is_unavaillable = False
|
|
service = Service[service]
|
|
planning = Planning[planning]
|
|
start_datetime = datetime.strptime(start_datetime, "%Y-%m-%dT%H%M")
|
|
end_datetime = start_datetime + timedelta(minutes=service.service_category.duration)
|
|
if customer:
|
|
customer = Customer[customer]
|
|
|
|
if ticket:
|
|
ticket = Ticket[ticket]
|
|
if notes is None:
|
|
notes = ""
|
|
newappointment = Appointment(
|
|
book_datetime=book_datetime,
|
|
start_datetime=start_datetime,
|
|
end_datetime=end_datetime,
|
|
is_unavaillable=is_unavaillable,
|
|
service=service,
|
|
planning=planning,
|
|
ticket=ticket,
|
|
customer=customer,
|
|
notes="",
|
|
)
|
|
rv = jsonify(newappointment.to_dict())
|
|
rv.status_code = 201
|
|
return rv
|
|
|
|
|
|
@api.route("/appointment", methods=["PUT"])
|
|
def update_appointment():
|
|
raise NotImplementedError
|
|
|
|
|
|
@api.route("/appointment", methods=["DELETE"])
|
|
def delete_appointment():
|
|
raise NotImplementedError
|