From 935ebe11a22ab7cbe80958269229ea7c57e95918 Mon Sep 17 00:00:00 2001 From: Andrey Date: Thu, 25 Aug 2022 15:21:09 +0300 Subject: [PATCH] remove print. black formating. --- .../mooncrawl/mooncrawl/stats_worker/queries.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py b/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py index 1a56abca..7d15aa93 100644 --- a/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py +++ b/crawlers/mooncrawl/mooncrawl/stats_worker/queries.py @@ -10,7 +10,10 @@ from moonstreamdb.db import create_moonstream_engine, MOONSTREAM_DB_URI_READ_ONL from sqlalchemy.orm import sessionmaker from ..reporter import reporter -from ..settings import MOONSTREAM_S3_QUERIES_BUCKET_PREFIX, MOONSTREAM_QUERY_API_DB_STATEMENT_TIMEOUT_MILLIS +from ..settings import ( + MOONSTREAM_S3_QUERIES_BUCKET_PREFIX, + MOONSTREAM_QUERY_API_DB_STATEMENT_TIMEOUT_MILLIS, +) logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) @@ -57,7 +60,6 @@ def to_json_types(value): return str(value) - def data_generate( bucket: str, query_id: str, @@ -71,7 +73,11 @@ def data_generate( s3 = boto3.client("s3") # Create session - engine = create_moonstream_engine(MOONSTREAM_DB_URI_READ_ONLY, pool_size=1, statement_timeout=MOONSTREAM_QUERY_API_DB_STATEMENT_TIMEOUT_MILLIS) + engine = create_moonstream_engine( + MOONSTREAM_DB_URI_READ_ONLY, + pool_size=1, + statement_timeout=MOONSTREAM_QUERY_API_DB_STATEMENT_TIMEOUT_MILLIS, + ) process_session = sessionmaker(bind=engine) db_session = process_session() @@ -129,8 +135,7 @@ def data_generate( [ "queries", "execution", - f"query_id:{query_id}" - f"file_type:{file_type}", + f"query_id:{query_id}" f"file_type:{file_type}", ], ) finally: