add tio ingestor class to ingestor session

kartoza · Oct 9, 2024 · 1c14665 · 1c14665
1 parent c9f9aff
commit 1c14665
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/django_project/gap/models/ingestor.py b/django_project/gap/models/ingestor.py
@@ -192,6 +192,7 @@ def _run(self, working_dir):
  from gap.ingestor.arable import ArableIngestor
  from gap.ingestor.tahmo_api import TahmoAPIIngestor
  from gap.ingestor.wind_borne_systems import WindBorneSystemsIngestor
+ from gap.ingestor.tio_shortterm import TioShortTermIngestor
 
  ingestor = None
  if self.ingestor_type == IngestorType.TAHMO:
@@ -210,6 +211,8 @@ def _run(self, working_dir):
  ingestor = TahmoAPIIngestor
  elif self.ingestor_type == IngestorType.WIND_BORNE_SYSTEMS_API:
  ingestor = WindBorneSystemsIngestor
+ elif self.ingestor_type == IngestorType.TOMORROWIO:
+ ingestor = TioShortTermIngestor
 
  if ingestor:
  ingestor(self, working_dir).run()