Bugfix add timezone to compare 2 dates

app/models/notion.py

@@ -1,5 +1,6 @@
 from config.database import Base
-from sqlalchemy import Column, Integer, String, DateTime
+from sqlalchemy import Column, String, DateTime
+from datetime import timezone
 
 
 class Notion(Base):
@@ -8,4 +9,4 @@ class Notion(Base):
     page_id = Column(String, primary_key=True)
     content = Column(String, nullable=True)
     embedded_at = Column(DateTime, nullable=True, default=None)
-    updated_at = Column(DateTime, nullable=True, default=None)
+    updated_at = Column(DateTime(timezone=True))

app/routers/v1/media.py

@@ -11,6 +11,7 @@
 from services import notion as ns
 from sqlalchemy.orm import Session
 from schemas.notion import NotionCreateSchema, NotionUpdateSchema
+from datetime import datetime, timezone
 
 
 router = APIRouter(
@@ -64,8 +65,9 @@ def run_notion(db: Session = Depends(get_db)):
             if page:
                 if ns.notion_object_exist(db, page_id=page.page_id):
                     existing_object = ns.get_notion_object_by_page_id(db, page_id=page.page_id)
+                    existing_object_updated_at = existing_object.updated_at.replace(tzinfo=timezone.utc)
 
-                    if existing_object.updated_at < page.updated_at:
+                    if existing_object_updated_at < page.updated_at:
                         ns.update_notion_content(
                             session=db,
                             page_id=page.page_id,

app/schemas/notion.py

@@ -7,7 +7,7 @@ class NotionSchema(BaseModel):
     page_id: str
     content: Optional[str] = None
     embedded_at:  Optional[datetime] = None
-    updated_at: Optional[datetime] = None
+    updated_at: datetime
 
 
 class NotionCreateSchema(BaseModel):