[add] Further logging on API. Using PULL_REQUEST_TRIGGER_LABEL env var to decide uppon which PR label to use for trigger automation

Author: filipecosta90

redis_benchmarks_specification/__api__/app.py

@@ -1,5 +1,4 @@
 import json
-import logging
 
 from flask import jsonify
 import redis
@@ -10,6 +9,7 @@
 from redis_benchmarks_specification.__common__.builder_schema import (
     commit_schema_to_stream,
 )
+from redis_benchmarks_specification.__common__.env import PULL_REQUEST_TRIGGER_LABEL
 
 SIG_HEADER = "X-Hub-Signature"
 
@@ -60,7 +60,7 @@ def base():
             event_type = "Ignored event from webhook"
             use_event = False
             # Pull request labeled
-            trigger_label = "trigger-benchmark"
+            trigger_label = PULL_REQUEST_TRIGGER_LABEL
             if "pull_request" in request_data:
                 action = request_data["action"]
                 if "labeled" == action:
@@ -98,22 +98,24 @@ def base():
 
             if use_event is True:
                 fields = {"git_hash": sha, "ref_label": ref_label, "ref": ref}
-                logging.info(
+                app.logger.info(
                     "Using event {} to trigger benchmark. final fields: {}".format(
                         event_type, fields
                     )
                 )
                 result, response_data, err_message = commit_schema_to_stream(
                     fields, conn, gh_org, gh_repo
                 )
-                logging.info(
+                app.logger.info(
                     "Using event {} to trigger benchmark. final fields: {}".format(
                         event_type, response_data
                     )
                 )
 
             else:
-                logging.info("{}. input json was: {}".format(event_type, request_data))
+                app.logger.info(
+                    "{}. input json was: {}".format(event_type, request_data)
+                )
                 response_data = {"message": event_type}
 
             # Send data back as JSON

redis_benchmarks_specification/__common__/env.py

@@ -50,6 +50,9 @@
 REDIS_SOCKET_TIMEOUT = int(os.getenv("REDIS_SOCKET_TIMEOUT", "300"))
 
 # environment variables
+PULL_REQUEST_TRIGGER_LABEL = os.getenv(
+    "PULL_REQUEST_TRIGGER_LABEL", "trigger-benchmark"
+)
 DATASINK_RTS_PUSH = bool(os.getenv("DATASINK_PUSH_RTS", False))
 DATASINK_RTS_AUTH = os.getenv("DATASINK_RTS_AUTH", None)
 DATASINK_RTS_USER = os.getenv("DATASINK_RTS_USER", None)

utils/tests/test_builder.py

@@ -56,7 +56,7 @@ def test_commit_schema_to_stream_then_build():
                 },
                 conn,
                 "redis",
-                "redis"
+                "redis",
             )
             assert result == True
             assert error_msg == None