-
Notifications
You must be signed in to change notification settings - Fork 326
/
TokenLogger.py
24 lines (20 loc) · 810 Bytes
/
TokenLogger.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import logging
from opencensus.ext.azure.log_exporter import AzureLogHandler
from ..helpers.EnvHelper import EnvHelper
class TokenLogger:
def __init__(self, name: str = __name__):
env_helper: EnvHelper = EnvHelper()
self.logger = logging.getLogger(name)
if env_helper.APPINSIGHTS_ENABLED:
self.logger.addHandler(
AzureLogHandler(
connection_string=env_helper.APPINSIGHTS_CONNECTION_STRING
)
)
self.logger.setLevel(logging.INFO)
def get_logger(self):
return self.logger
def log(self, message: str, custom_dimensions: dict):
# Setting log properties
log_properties = {"custom_dimensions": custom_dimensions}
self.logger.info(message, extra=log_properties)