# A very simple script to read a presite note and load to a postgres database
import os
from etl.surveyedData.surveryedData import surveyedDataProcessor
from etl.db.db import get_db_session, init_db
from etl.transform.preSiteNoteTypes import AssessorInfo


pre_site_note_path = os.path.join(os.getcwd(), "..", "example_data", "pre_site_note.pdf")

survey_one = surveyedDataProcessor("122 Fake Street", [pre_site_note_path])

assert survey_one.pre_site_note.assessor_information.accreditation_number == "QUID211435"

init_db()


assessor0 = AssessorInfo(
    **survey_one.pre_site_note.assessor_information.model_dump()
)

with get_db_session() as session:
    session.add(assessor0)
    session.commit()