From 27ceff60c15c8199eb87b609041b2403f10e6e85 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Sun, 19 Jun 2022 00:43:30 +0200 Subject: [PATCH] info log for callback --- pyinfra/default_objects.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/pyinfra/default_objects.py b/pyinfra/default_objects.py index 4ade2b6..5f593f3 100644 --- a/pyinfra/default_objects.py +++ b/pyinfra/default_objects.py @@ -1,7 +1,8 @@ import logging from functools import lru_cache +from operator import itemgetter -from funcy import rcompose, omit, merge, lmap +from funcy import rcompose, omit, merge, lmap, project from pyinfra.config import CONFIG from pyinfra.exceptions import AnalysisFailure @@ -17,7 +18,6 @@ from pyinfra.visitor import QueueVisitor from pyinfra.visitor.strategies.response.aggregation import AggregationStorageStrategy logger = logging.getLogger(__name__) -logger.setLevel(logging.ERROR) @lru_cache(maxsize=None) @@ -44,7 +44,15 @@ def get_storage(): @lru_cache(maxsize=None) def get_callback(analysis_base_url=None): analysis_base_url = analysis_base_url or CONFIG.rabbitmq.callback.analysis_endpoint - return Callback(analysis_base_url) + + callback = Callback(analysis_base_url) + + def wrapped(body): + body_repr = project(body, ["dossierId", "fileId", "pages", "images", "operation"]) + logger.info(f"Processing {body_repr}...") + return callback(body) + + return wrapped @lru_cache(maxsize=None)