feat: Added ability to change travel

2024-03-23 23:53:54 +03:00
parent 40ea9689ab
commit 02508ea9fd
9 changed files with 206 additions and 24 deletions
@@ -74,13 +74,19 @@ class Travel(Base):
    def get_travel_text(self):
        return messages.TRAVEL_DETAIL.format(
            title=self.title,
-            description=self.description,
+            description=(
+                self.description if self.description else messages.NOT_SET
+            ),
        )

    @classmethod
    def get_travel_by_id(cls, travel_id):
        return session.query(Travel).filter(Travel.id == travel_id).first()

+    @classmethod
+    def get_travel_queryset_by_id(cls, travel_id):
+        return session.query(Travel).filter(Travel.id == travel_id)
+

 class Location(Base):
    __tablename__ = "locations"