Spaces:
Sleeping
Sleeping
| """Dispatcher: routes to the correct extractor and saves records.""" | |
| from extraction.tonnage_extractor import extract_tonnage | |
| from extraction.cargo_vc_extractor import extract_cargo_vc | |
| from extraction.cargo_tc_extractor import extract_cargo_tc | |
| from database.models import Tonnage, CargoVC, CargoTC | |
| def extract(category: str, text: str, email_id: int, session) -> list[dict]: | |
| """Extract structured data, persist to DB, return list of extracted dicts.""" | |
| records = [] | |
| if category == 'tonnage': | |
| vessels = extract_tonnage(text, email_id) | |
| for v in vessels: | |
| obj = Tonnage(**v) | |
| session.add(obj) | |
| records.append(v) | |
| elif category == 'cargo_vc': | |
| cargos = extract_cargo_vc(text, email_id) | |
| for c in cargos: | |
| obj = CargoVC(**c) | |
| session.add(obj) | |
| records.append(c) | |
| elif category == 'cargo_tc': | |
| cargos = extract_cargo_tc(text, email_id) | |
| for c in cargos: | |
| obj = CargoTC(**c) | |
| session.add(obj) | |
| records.append(c) | |
| return records | |