Skip to content

Commit

Permalink
GeoJSON
Browse files Browse the repository at this point in the history
  • Loading branch information
ZoutigeWolf committed Nov 7, 2024
1 parent 3faef6f commit 843bfc3
Show file tree
Hide file tree
Showing 7 changed files with 134 additions and 0 deletions.
32 changes: 32 additions & 0 deletions alembic/versions/6f4e2a27d3c1_transfer_fk_delete.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
"""transfer_fk_delete
Revision ID: 6f4e2a27d3c1
Revises: a4ad5b2eeb59
Create Date: 2024-11-05 01:39:04.203610
"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = '6f4e2a27d3c1'
down_revision: Union[str, None] = 'a4ad5b2eeb59'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_constraint('gtfs_transfers_to_stop_id_fkey', 'gtfs_transfers', type_='foreignkey')
op.drop_constraint('gtfs_transfers_from_stop_id_fkey', 'gtfs_transfers', type_='foreignkey')
# ### end Alembic commands ###


def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.create_foreign_key('gtfs_transfers_from_stop_id_fkey', 'gtfs_transfers', 'gtfs_stops', ['from_stop_id'], ['id'])
op.create_foreign_key('gtfs_transfers_to_stop_id_fkey', 'gtfs_transfers', 'gtfs_stops', ['to_stop_id'], ['id'])
# ### end Alembic commands ###
30 changes: 30 additions & 0 deletions alembic/versions/a4ad5b2eeb59_stop_fk_delete.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,30 @@
"""stop_fk_delete
Revision ID: a4ad5b2eeb59
Revises: bc2968bcbb5a
Create Date: 2024-11-05 01:32:58.021286
"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = 'a4ad5b2eeb59'
down_revision: Union[str, None] = 'bc2968bcbb5a'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.drop_constraint('gtfs_stop_times_stop_id_fkey', 'gtfs_stop_times', type_='foreignkey')
# ### end Alembic commands ###


def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.create_foreign_key('gtfs_stop_times_stop_id_fkey', 'gtfs_stop_times', 'gtfs_stops', ['stop_id'], ['id'])
# ### end Alembic commands ###
45 changes: 45 additions & 0 deletions alembic/versions/bc2968bcbb5a_srid_4326.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
"""srid_4326
Revision ID: bc2968bcbb5a
Revises: a8d52d2f6519
Create Date: 2024-11-05 01:24:14.377941
"""
from typing import Sequence, Union
import geoalchemy2

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = 'bc2968bcbb5a'
down_revision: Union[str, None] = 'a8d52d2f6519'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.alter_column('gtfs_shapes', 'line',
existing_type=geoalchemy2.types.Geometry(geometry_type='LINESTRING', srid=28992, from_text='ST_GeomFromEWKT', name='geometry', _spatial_index_reflected=True),
type_=geoalchemy2.types.Geometry(geometry_type='LINESTRING', srid=4326, from_text='ST_GeomFromEWKT', name='geometry'),
existing_nullable=True)
op.alter_column('gtfs_stops', 'location',
existing_type=geoalchemy2.types.Geometry(geometry_type='POINT', srid=28992, from_text='ST_GeomFromEWKT', name='geometry', _spatial_index_reflected=True),
type_=geoalchemy2.types.Geometry(geometry_type='POINT', srid=4326, from_text='ST_GeomFromEWKT', name='geometry'),
existing_nullable=True)
# ### end Alembic commands ###


def downgrade() -> None:
# ### commands auto generated by Alembic - please adjust! ###
op.alter_column('gtfs_stops', 'location',
existing_type=geoalchemy2.types.Geometry(geometry_type='POINT', srid=4326, from_text='ST_GeomFromEWKT', name='geometry'),
type_=geoalchemy2.types.Geometry(geometry_type='POINT', srid=28992, from_text='ST_GeomFromEWKT', name='geometry', _spatial_index_reflected=True),
existing_nullable=True)
op.alter_column('gtfs_shapes', 'line',
existing_type=geoalchemy2.types.Geometry(geometry_type='LINESTRING', srid=4326, from_text='ST_GeomFromEWKT', name='geometry'),
type_=geoalchemy2.types.Geometry(geometry_type='LINESTRING', srid=28992, from_text='ST_GeomFromEWKT', name='geometry', _spatial_index_reflected=True),
existing_nullable=True)
# ### end Alembic commands ###
3 changes: 3 additions & 0 deletions models/views/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
from models.views.stop_type import StopType
from models.views.shape_type import ShapeType
from models.views.stop_route import StopRoute
8 changes: 8 additions & 0 deletions models/views/shape_type.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
from sqlmodel import SQLModel, Field
from enums import RouteType

class ShapeType(SQLModel, table=True):
__tablename__ = "shape_types" # type: ignore

shape_id: str = Field(primary_key=True, index=True)
type: RouteType = Field(primary_key=True)
8 changes: 8 additions & 0 deletions models/views/stop_route.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
from sqlmodel import SQLModel, Field
from enums import RouteType

class StopRoute(SQLModel, table=True):
__tablename__ = "stop_routes" # type: ignore

stop_id: str = Field(primary_key=True, index=True)
route_id: str = Field(primary_key=True)
8 changes: 8 additions & 0 deletions models/views/stop_type.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
from sqlmodel import SQLModel, Field
from enums import RouteType

class StopType(SQLModel, table=True):
__tablename__ = "stop_types" # type: ignore

stop_id: str = Field(primary_key=True, index=True)
type: RouteType = Field(primary_key=True)

0 comments on commit 843bfc3

Please sign in to comment.