Skip to content

Commit 1556654

Browse files
committed
fix: moved functions
1 parent 78c8bd4 commit 1556654

File tree

1 file changed

+22
-22
lines changed

1 file changed

+22
-22
lines changed

functions-python/tasks_executor/src/tasks/data_import/trasportdatagouv/import_tdg_feeds.py

Lines changed: 22 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -216,6 +216,28 @@ def _get_tdg_locations(db_session: Session, dataset: dict) -> List[Any]:
216216
# ---------------------------------------------------------------------------
217217

218218

219+
def _deprecate_stale_feeds(db_session, processed_stable_ids):
220+
"""
221+
Deprecate TDG feeds not seen in this import run.
222+
"""
223+
logger.info("Deprecating stale TDG feeds not in processed_stable_ids")
224+
tdg_feeds = (
225+
db_session.query(Feed)
226+
.filter(Feed.stable_id.like("tdg-%"))
227+
.filter(~Feed.stable_id.in_(processed_stable_ids))
228+
.all()
229+
)
230+
logger.info("Found %d tdg_feeds stale stable_ids", len(tdg_feeds))
231+
deprecated_count = 0
232+
for feed in tdg_feeds:
233+
if feed.status != "deprecated":
234+
feed.status = "deprecated"
235+
deprecated_count += 1
236+
logger.info("Deprecated stale TDG feed stable_id=%s", feed.stable_id)
237+
238+
logger.info("Total deprecated stale TDG feeds: %d", deprecated_count)
239+
240+
219241
def _ensure_tdg_external_id(feed: Feed, resource_id: str) -> None:
220242
"""
221243
Ensure that an Externalid(source='tdg', associated_id=<resource_id>) exists.
@@ -558,28 +580,6 @@ def _process_tdg_dataset(
558580
return deltas, feeds_to_publish
559581

560582

561-
def _deprecate_stale_feeds(db_session, processed_stable_ids):
562-
"""
563-
Deprecate TDG feeds not seen in this import run.
564-
"""
565-
logger.info("Deprecating stale TDG feeds not in processed_stable_ids")
566-
tdg_feeds = (
567-
db_session.query(Feed)
568-
.filter(Feed.stable_id.like("tdg-%"))
569-
.filter(~Feed.stable_id.in_(processed_stable_ids))
570-
.all()
571-
)
572-
logger.info("Found %d tdg_feeds stale stable_ids", len(tdg_feeds))
573-
deprecated_count = 0
574-
for feed in tdg_feeds:
575-
if feed.status != "deprecated":
576-
feed.status = "deprecated"
577-
deprecated_count += 1
578-
logger.info("Deprecated stale TDG feed stable_id=%s", feed.stable_id)
579-
580-
logger.info("Total deprecated stale TDG feeds: %d", deprecated_count)
581-
582-
583583
# ---------------------------------------------------------------------------
584584
# Orchestrator & handler
585585
# ---------------------------------------------------------------------------

0 commit comments

Comments
 (0)