diff --git a/vxlogging.py b/vxlogging.py index cc26c91..852ac4c 100644 --- a/vxlogging.py +++ b/vxlogging.py @@ -1,22 +1,36 @@ IS_DEBUG = False +from flask import request +def generic(message): + invocation_id = None + try: + if 'serverless.context' in request.environ: + invocation_id = request.environ['serverless.context'].aws_request_id + except: + pass + if invocation_id is None: + invocation_id = "" + else: + invocation_id = str(invocation_id)+" " + message = str(f"{invocation_id}{message}") + print(message) def info(message): message = str(message) - print(f" > [ I ] {message}") + generic(f" > [ I ] {message}") def success(message): message = str(message) - print(f" > [ ✔ ] {message}") + generic(f" > [ ✔ ] {message}") def error(message): message = str(message) - print(f" > [ ✘ ] {message}") + generic(f" > [ ✘ ] {message}") def warn(message): message = str(message) - print(f" > [ ! ] {message}") + generic(f" > [ ! ] {message}") def debug(message): if IS_DEBUG: message = str(message) - print(f" > [ D ] {message}") \ No newline at end of file + generic(f" > [ D ] {message}") \ No newline at end of file