diff --git a/backend/moonstreamapi/routes/queries.py b/backend/moonstreamapi/routes/queries.py index 39e8a491..f5e57d75 100644 --- a/backend/moonstreamapi/routes/queries.py +++ b/backend/moonstreamapi/routes/queries.py @@ -21,7 +21,9 @@ from ..settings import bugout_client as bc logger = logging.getLogger(__name__) -router = APIRouter(prefix="/queries",) +router = APIRouter( + prefix="/queries", +) @router.post("/{query_id}/update", tags=["queries"]) diff --git a/crawlers/mooncrawl/mooncrawl/api.py b/crawlers/mooncrawl/mooncrawl/api.py index abdd8cd0..a52e973c 100644 --- a/crawlers/mooncrawl/mooncrawl/api.py +++ b/crawlers/mooncrawl/mooncrawl/api.py @@ -81,14 +81,17 @@ async def now_handler() -> data.NowResponse: @app.post("/jobs/stats_update", tags=["jobs"]) async def status_handler( - stats_update: data.StatsUpdateRequest, background_tasks: BackgroundTasks, + stats_update: data.StatsUpdateRequest, + background_tasks: BackgroundTasks, ): """ Update dashboard endpoint create are tasks for update. """ dashboard_resource: BugoutResource = bc.get_resource( - token=stats_update.token, resource_id=stats_update.dashboard_id, timeout=10, + token=stats_update.token, + resource_id=stats_update.dashboard_id, + timeout=10, ) # get all user subscriptions @@ -168,7 +171,9 @@ async def status_handler( @app.post("/jobs/{query_id}/query_update", tags=["jobs"]) async def queries_data_update_handler( - query_id: str, request: data.QueryDataUpdate, background_tasks: BackgroundTasks, + query_id: str, + request: data.QueryDataUpdate, + background_tasks: BackgroundTasks, ) -> Dict[str, Any]: s3_client = boto3.client("s3") diff --git a/crawlers/mooncrawl/mooncrawl/settings.py b/crawlers/mooncrawl/mooncrawl/settings.py index f85575a2..66b01bdc 100644 --- a/crawlers/mooncrawl/mooncrawl/settings.py +++ b/crawlers/mooncrawl/mooncrawl/settings.py @@ -89,4 +89,3 @@ if MOONSTREAM_MOONWORM_TASKS_JOURNAL == "": MOONSTREAM_QUERIES_BUCKET = os.environ.get("MOONSTREAM_QUERIES_BUCKET", "") if MOONSTREAM_QUERIES_BUCKET == "": raise ValueError("MOONSTREAM_QUERIES_BUCKET environment variable must be set") - diff --git a/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py b/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py index 782bd293..1b6d89a6 100644 --- a/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py +++ b/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py @@ -5,7 +5,7 @@ from io import StringIO import csv -import boto3 +import boto3 # type: ignore from moonstreamdb.db import yield_db_session_ctx