mih-project/mih_api_hub/routers/appointments.py

import mysql.connector
from fastapi import APIRouter, HTTPException
from pydantic import BaseModel
#from ..mih_database import dbConnection
import mih_database
from datetime import datetime, timedelta, date
#SuperToken Auth from front end
from supertokens_python.recipe.session.framework.fastapi import verify_session
from supertokens_python.recipe.session import SessionContainer
from fastapi import Depends

router = APIRouter()

class appointmentInsertRequest(BaseModel):
    app_id: str
    business_id: str
    title: str
    description: str
    date: str
    time: str

class appointmentUpdateRequest(BaseModel):
    idappointments: int
    title: str
    description: str
    date: str
    time: str

class appointmentDeleteRequest(BaseModel):
    idappointments: int

# Get List of all files by patient
@router.get("/appointments/business/{business_id}", tags=["Appointments"])
async def read_all_appointments_by_business_id(business_id: str, date: str, session: SessionContainer = Depends(verify_session())): #, session: SessionContainer = Depends(verify_session())
    db = mih_database.dbConnection.dbMzansiCalendarConnect()
    requestDate = datetime.strptime(date, '%Y-%m-%d').date()
    cursor = db.cursor()
    query = "SELECT appointments.idappointments, appointments.app_id, "
    query += "appointments.business_id, appointments.date_time, "
    query += "appointments.title, appointments.description "
    query += "from mzansi_calendar.appointments "
    query = query + "where appointments.business_id = %s and date_time like '" + str(requestDate) + "%' "
    query += "ORDER BY date_time ASC"
    cursor.execute(query, (business_id,))
    items = [
        {
            "idappointments": item[0],
            "app_id": item[1],
            "business_id": item[2],
            "date_time": item[3],
            "title": item[4],
            "description": item[5],
        }
        for item in cursor.fetchall()
    ]
    cursor.close()
    db.close()
    return items

# Get List of all files by patient
@router.get("/appointments/personal/{app_id}", tags=["Appointments"])
async def read_all_appointments_by_business_id(app_id: str, date: str, session: SessionContainer = Depends(verify_session())): #, session: SessionContainer = Depends(verify_session())
    db = mih_database.dbConnection.dbMzansiCalendarConnect()
    requestDate = datetime.strptime(date, '%Y-%m-%d').date()
    cursor = db.cursor()
    query = "SELECT appointments.idappointments, appointments.app_id, "
    query += "appointments.business_id, appointments.date_time, "
    query += "appointments.title, appointments.description "
    query += "from mzansi_calendar.appointments "
    query = query + "where appointments.app_id = %s and date_time like '" + str(requestDate) + "%' "
    query += "ORDER BY date_time ASC"
   
    cursor.execute(query, (app_id,))
    items = [
        {
            "idappointments": item[0],
            "app_id": item[1],
            "business_id": item[2],
            "date_time": item[3],
            "title": item[4],
            "description": item[5],
        }
        for item in cursor.fetchall()
    ]
    cursor.close()
    db.close()
    return items

# Insert Patient note into table
@router.post("/appointment/insert/", tags=["Appointments"], status_code=201)
async def insert_appointment(itemRequest : appointmentInsertRequest, session: SessionContainer = Depends(verify_session())): #, session: SessionContainer = Depends(verify_session())
    date_time = itemRequest.date + " " + itemRequest.time + ":00"
    db = mih_database.dbConnection.dbMzansiCalendarConnect()
    cursor = db.cursor()
    query = "insert into appointments "
    query += "(app_id, business_id, title, description, date_time) "
    query += "values (%s, %s, %s, %s, %s)"
    notetData = (itemRequest.app_id,
                 itemRequest.business_id,
                 itemRequest.title,
                 itemRequest.description,
                 date_time,
                 )
    try:
       cursor.execute(query, notetData) 
    except Exception as error:
        #raise HTTPException(status_code=404, detail="Failed to Create Record")
        return {"message": error}
    db.commit()
    cursor.close()
    db.close()
    return {"message": "Successfully Created file Record"}

# Update Patient on table
@router.put("/appointment/update/", tags=["Appointments"])
async def Update_appointment(itemRequest : appointmentUpdateRequest, session: SessionContainer = Depends(verify_session())): #, session: SessionContainer = Depends(verify_session())
    
    date_time = itemRequest.date + " " + itemRequest.time + ":00"

    db = mih_database.dbConnection.dbMzansiCalendarConnect()
    cursor = db.cursor()
    query = "update appointments "
    query += "set date_time=%s, title=%s, description=%s "
    query += "where idappointments=%s"
    patientData = (date_time, 
                   itemRequest.title,
                   itemRequest.description,
                   itemRequest.idappointments)
    try:
       cursor.execute(query, patientData) 
    except Exception as error:
        print(error)
        raise HTTPException(status_code=404, detail="Failed to Update Record")
        #return {"query": query, "message": error}
    db.commit()
    cursor.close()
    db.close()
    return {"message": "Successfully Updated Record"}

# Update Patient on table
@router.delete("/appointment/delete/", tags=["Appointments"])
async def Delete_appointment(itemRequest : appointmentDeleteRequest, session: SessionContainer = Depends(verify_session())): #, session: SessionContainer = Depends(verify_session())
    db = mih_database.dbConnection.dbMzansiCalendarConnect()
    cursor = db.cursor()
    query = "delete from appointments "
    query += "where idappointments=%s"
    try:
       cursor.execute(query, (str(itemRequest.idappointments),)) 
    except Exception as error:
        print(error)
        raise HTTPException(status_code=404, detail="Failed to Delete Appointment")
        #return {"query": query, "message": error}
    db.commit()
    cursor.close()
    db.close()
    return {"message": "Successfully deleted Appointment"}