From bb06c8a958d4fe90d368bbf868e937db4cc4b9f1 Mon Sep 17 00:00:00 2001 From: Gui Heise Date: Fri, 11 Feb 2022 16:47:24 -0500 Subject: [PATCH] Add export task --- mev_inspect/queue/tasks.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/mev_inspect/queue/tasks.py b/mev_inspect/queue/tasks.py index 78a4c3e..a55efd5 100644 --- a/mev_inspect/queue/tasks.py +++ b/mev_inspect/queue/tasks.py @@ -2,6 +2,8 @@ import asyncio import logging from contextlib import contextmanager +from mev_inspect.s3_export import export_block + from .middleware import DbMiddleware, InspectorMiddleware logger = logging.getLogger(__name__) @@ -23,6 +25,11 @@ def inspect_many_blocks_task( ) +def export_block_task(block_number: int): + with _session_scope(DbMiddleware.get_inspect_sessionmaker()) as inspect_db_session: + export_block(inspect_db_session, block_number) + + @contextmanager def _session_scope(Session=None): if Session is None: