Add agency model

mbta_gtfs_sqlite/ingest.py

@@ -7,6 +7,7 @@
 from .utils.time import date_from_string, seconds_from_string
 from .utils.decorators import listify
 from .utils.indexes import bucket_by
+from .models.agency import Agency
 from .models.base import Base
 from .models.calendar_attributes import CalendarAttribute
 from .models.calendar_dates import CalendarServiceException
@@ -130,6 +131,13 @@ def ingest_gtfs_csv_into_db(
     batch_size: Union[None, int] = None,
 ):
     feed_info = ingest_feed_info(session, download, reader)
+    ingest_rows(
+        session=session,
+        model=Agency,
+        feed_info=feed_info,
+        rows=reader.read_agency(),
+        batch_size=batch_size,
+    )
     ingest_rows(
         session=session,
         model=CalendarService,

mbta_gtfs_sqlite/models/__init__.py

@@ -1,3 +1,4 @@
+from .agency import *
 from .base import *
 from .calendar import *
 from .calendar_attributes import *

mbta_gtfs_sqlite/models/agency.py

@@ -0,0 +1,15 @@
+from sqlalchemy.types import String
+from sqlalchemy.orm import mapped_column, Mapped
+
+from .base import Base
+
+
+class Agency(Base):
+    agency_id: Mapped[str] = mapped_column(String)
+    agency_name: Mapped[str] = mapped_column(String)
+    agency_url: Mapped[str] = mapped_column(String)
+    agency_timezone: Mapped[str] = mapped_column(String)
+    agency_lang: Mapped[str] = mapped_column(String, nullable=True)
+    agency_phone: Mapped[str] = mapped_column(String, nullable=True)
+    agency_fare_url: Mapped[str] = mapped_column(String, nullable=True)
+    agency_email: Mapped[str] = mapped_column(String, nullable=True)

mbta_gtfs_sqlite/reader.py

@@ -19,6 +19,7 @@ def load():
 
     def __init__(self, root: str):
         self.root = root
+        self.read_agency = self._reader_by_file_name("agency")
         self.read_calendar = self._reader_by_file_name("calendar")
         self.read_calendar_attributes = self._reader_by_file_name("calendar_attributes")
         self.read_calendar_dates = self._reader_by_file_name("calendar_dates")

pyproject.toml

@@ -1,9 +1,9 @@
 [tool.poetry]
 name = "mbta-gtfs-sqlite"
-version = "1.1.1"
+version = "1.1.2"
 description = "Query the MBTA's static GTFS feeds using sqlite"
 authors = ["Ian Reynolds <[email protected]>"]
-packages = [{include = "mbta_gtfs_sqlite"}]
+packages = [{ include = "mbta_gtfs_sqlite" }]
 
 [tool.poetry.dependencies]
 python = "^3.8"