danvics/docling-studio
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Fix naive datetime fallbacks in repository layer

Browse source 
Use UTC-aware datetimes consistently to prevent TypeError when
comparing aware and naive datetime objects.
This commit is contained in:
Pier-Jean Malandrino 2026-04-03 13:04:44 +02:00
parent 87269b9393
commit dbcd9dfa4c
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
document-parser/persistence/analysis_repo.py
View file

@ -2,7 +2,7 @@
from __future__ import annotations from __future__ import annotations
from datetime import datetime from datetime import UTC, datetime
from domain.models import AnalysisJob, AnalysisStatus from domain.models import AnalysisJob, AnalysisStatus
from persistence.database import get_connection from persistence.database import get_connection
@ -29,7 +29,7 @@ def _row_to_job(row) -> AnalysisJob:
error_message=row["error_message"], error_message=row["error_message"],
started_at=_parse_dt(row["started_at"]), started_at=_parse_dt(row["started_at"]),
completed_at=_parse_dt(row["completed_at"]), completed_at=_parse_dt(row["completed_at"]),
created_at=_parse_dt(row["created_at"]) or datetime.now(), created_at=_parse_dt(row["created_at"]) or datetime.now(UTC),
document_filename=row["filename"] if "filename" in keys else None, document_filename=row["filename"] if "filename" in keys else None,
) )

4
document-parser/persistence/document_repo.py
View file

@ -2,7 +2,7 @@
from __future__ import annotations from __future__ import annotations
from datetime import datetime from datetime import UTC, datetime
from domain.models import Document from domain.models import Document
from persistence.database import get_connection from persistence.database import get_connection
@ -12,6 +12,8 @@ def _row_to_document(row) -> Document:
created = row["created_at"] created = row["created_at"]
if isinstance(created, str): if isinstance(created, str):
created = datetime.fromisoformat(created) created = datetime.fromisoformat(created)
if created.tzinfo is None:
created = created.replace(tzinfo=UTC)
return Document( return Document(
id=row["id"], id=row["id"],
filename=row["filename"], filename=row["filename"],