115 lines
4.8 KiB
Python
115 lines
4.8 KiB
Python
from dotenv import load_dotenv, find_dotenv
|
|
load_dotenv(find_dotenv())
|
|
import os
|
|
import modules.logger_tool as logger
|
|
log_name = 'api_routers_database_init_schools'
|
|
log_dir = os.getenv("LOG_PATH", "/logs") # Default path as fallback
|
|
logging = logger.get_logger(
|
|
name=log_name,
|
|
log_level=os.getenv("LOG_LEVEL", "DEBUG"),
|
|
log_path=log_dir,
|
|
log_file=log_name,
|
|
runtime=True,
|
|
log_format='default'
|
|
)
|
|
from fastapi import APIRouter, File, UploadFile, Form, HTTPException, BackgroundTasks
|
|
import pandas as pd
|
|
import modules.database.tools.neo4j_driver_tools as driver
|
|
from modules.database.tools.neo4j_session_tools import get_node_by_unique_id
|
|
import modules.database.init.init_school_timetable as init_school_timetable
|
|
import modules.database.init.init_worker_timetable as init_worker_timetable
|
|
from modules.database.schemas.nodes.schools.schools import SchoolNode
|
|
import modules.database.init.xl_tools as xl
|
|
import json
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/upload-school-timetable")
|
|
async def upload_school_timetable(
|
|
file: UploadFile = File(...),
|
|
db_name: str = Form(...),
|
|
unique_id: str = Form(...),
|
|
school_uuid: str = Form(...),
|
|
school_name: str = Form(...),
|
|
school_website: str = Form(...),
|
|
path: str = Form(...)
|
|
):
|
|
school_node = SchoolNode(
|
|
unique_id=unique_id,
|
|
school_uuid=school_uuid,
|
|
school_name=school_name,
|
|
school_website=school_website,
|
|
path=path
|
|
)
|
|
if file.content_type != 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet':
|
|
return {"status": "Error", "message": "Invalid file format"}
|
|
logging.info(f"Uploading timetable for {db_name} from {file.filename}")
|
|
dataframes = xl.create_dataframes_from_fastapiuploadfile(file)
|
|
return init_school_timetable.create_school_timetable(dataframes, db_name, school_node)
|
|
|
|
@router.post("/upload-worker-timetable")
|
|
async def upload_worker_timetable(
|
|
background_tasks: BackgroundTasks,
|
|
file: UploadFile = File(...),
|
|
worker_node: str = Form(...)
|
|
):
|
|
if file.content_type != 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet':
|
|
raise HTTPException(status_code=422, detail="Invalid file format")
|
|
|
|
try:
|
|
worker_node_data = json.loads(worker_node)
|
|
logging.info(f"Uploading worker timetable for {worker_node_data['teacher_code']} from {file.filename} for {worker_node_data['worker_db_name']}")
|
|
logging.debug(f"Worker node data: {worker_node_data}")
|
|
|
|
# Read file content into memory
|
|
file_content = await file.read()
|
|
|
|
# Schedule the processing of the timetable in the background
|
|
background_tasks.add_task(
|
|
process_worker_timetable,
|
|
file_content,
|
|
worker_node_data
|
|
)
|
|
|
|
return {
|
|
"status": "Accepted",
|
|
"message": "Processing of teacher timetable started"
|
|
}
|
|
except Exception as e:
|
|
logging.error(f"Error handling timetable upload: {str(e)}")
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
async def process_worker_timetable(file_content, worker_node_data):
|
|
neo_driver = driver.get_driver(db_name=worker_node_data['worker_db_name'])
|
|
if neo_driver is None:
|
|
logging.error(f"Failed to connect to the database {worker_node_data['worker_db_name']}")
|
|
return
|
|
|
|
try:
|
|
# Create a DataFrame from the file content
|
|
from io import BytesIO
|
|
timetable_df = pd.read_excel(BytesIO(file_content))
|
|
|
|
# Get the school version of the worker node
|
|
logging.info(f"Getting school worker node for {worker_node_data['unique_id']} from {worker_node_data['worker_db_name']}")
|
|
|
|
with neo_driver.session(database=worker_node_data['worker_db_name']) as neo_session:
|
|
school_worker_node = get_node_by_unique_id(session=neo_session, unique_id=worker_node_data['unique_id'])
|
|
|
|
if school_worker_node is None:
|
|
error_msg = f"School worker node not found for unique_id: {worker_node_data['unique_id']}"
|
|
logging.error(error_msg)
|
|
raise Exception(error_msg)
|
|
|
|
logging.debug(f"School worker node found: {school_worker_node}")
|
|
|
|
logging.info(f"Initializing worker timetable for school worker: {school_worker_node['teacher_code']}")
|
|
init_worker_timetable.init_worker_timetable(timetable_df, school_worker_node)
|
|
logging.info(f"Worker timetable initialized for school worker: {school_worker_node['teacher_code']}")
|
|
|
|
except Exception as e:
|
|
logging.error(f"Error processing worker timetable: {str(e)}")
|
|
raise
|
|
finally:
|
|
logging.info(f"Closing driver for {worker_node_data['worker_db_name']}")
|
|
driver.close_driver(neo_driver) |