polyline table (#62) (#63)

* polyline table (#62)

* polyline and route_stops models

---------

Co-authored-by: Tq <tqmeng123@gmail.com>

* revert unintended changes

* revert unintended changes

* unintended changes

* updated according to feedback (#64)

* adding polyline fields to route table

* lint

* lint

* unintended changes

* simplier way

* Add migration

---------

Co-authored-by: Yujie <yujiemengca@gmail.com>
Co-authored-by: Colin Toft <47548563+ColinToft@users.noreply.github.com>
Co-authored-by: Colin Toft <colintoft@uwblueprint.org>

Author: 3 people

backend/python/app/models/route.py

@@ -1,6 +1,8 @@
+from datetime import datetime
 from typing import TYPE_CHECKING
 from uuid import UUID, uuid4
 
+from sqlalchemy import Column, DateTime
 from sqlmodel import Field, Relationship, SQLModel
 
 from .base import BaseModel
@@ -16,6 +18,11 @@ class RouteBase(SQLModel):
     name: str = Field(default="", min_length=1, max_length=255)  # can change this later
     notes: str = Field(default="", max_length=1000)  # can change this later
     length: float = Field(ge=0.0)  # in km, must be non-negative
+    encoded_polyline: str | None = Field(default=None, max_length=10000)
+    polyline_updated_at: datetime | None = Field(
+        default=None, sa_column=Column(DateTime(timezone=True), nullable=True)
+    )
+    expires_at: datetime | None = Field(default=None)
 
 
 class Route(RouteBase, BaseModel, table=True):
@@ -32,7 +39,6 @@ class Route(RouteBase, BaseModel, table=True):
     route_stops: list["RouteStop"] = Relationship(
         back_populates="route", sa_relationship_kwargs={"cascade": "all, delete-orphan"}
     )
-
     # Relationship to route group memberships
     route_group_memberships: list["RouteGroupMembership"] = Relationship(
         back_populates="route", sa_relationship_kwargs={"cascade": "all, delete-orphan"}
@@ -60,3 +66,6 @@ class RouteUpdate(SQLModel):
     name: str | None = None
     notes: str | None = None
     length: float | None = None
+    encoded_polyline: str | None = None
+    polyline_updated_at: datetime | None = None
+    expires_at: datetime | None = None

backend/python/migrations/versions/add_polyline_fields_to_routes_table.py

@@ -0,0 +1,34 @@
+"""Add polyline fields to routes table
+
+Revision ID: 8574dd773368
+Revises: afbf8fe1a1a3
+Create Date: 2025-01-27 12:00:00.000000
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlmodel
+
+
+# revision identifiers, used by Alembic.
+revision = '8574dd773368'
+down_revision = 'afbf8fe1a1a3'
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('routes', sa.Column(
+        'encoded_polyline', sqlmodel.sql.sqltypes.AutoString(length=10000), nullable=True))
+    op.add_column('routes', sa.Column(
+        'polyline_updated_at', sa.DateTime(timezone=True), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('routes', 'polyline_updated_at')
+    op.drop_column('routes', 'encoded_polyline')
+    # ### end Alembic commands ###
+