Moves Logger to common

apps/cli/__init__.py

@@ -1,5 +1,4 @@
 import logging
-from apps.cli.logger import Logger
 
 # PISA-SERVER
 DEFAULT_PISA_API_SERVER = "btc.pisa.watch"
@@ -13,9 +12,22 @@ CLI_PUBLIC_KEY = "cli_pk.der"
 CLI_PRIVATE_KEY = "cli_sk.der"
 PISA_PUBLIC_KEY = "pisa_pk.der"
 
-# Configure logging
-logging.basicConfig(
-    format="%(message)s", level=logging.INFO, handlers=[logging.FileHandler(CLIENT_LOG_FILE), logging.StreamHandler()]
-)
+# Create the file logger
+f_logger = logging.getLogger("cli_file_log")
+f_logger.setLevel(logging.INFO)
 
-logger = Logger("Client")
+fh = logging.FileHandler(CLIENT_LOG_FILE)
+fh.setLevel(logging.INFO)
+fh_formatter = logging.Formatter("%(message)s")
+fh.setFormatter(fh_formatter)
+f_logger.addHandler(fh)
+
+# Create the console logger
+c_logger = logging.getLogger("cli_console_log")
+c_logger.setLevel(logging.INFO)
+
+ch = logging.StreamHandler()
+ch.setLevel(logging.INFO)
+ch_formatter = logging.Formatter("%(asctime)s %(message)s", "%Y-%m-%d %H:%M:%S")
+ch.setFormatter(ch_formatter)
+c_logger.addHandler(ch)

apps/cli/blob.py

@@ -1,9 +1,4 @@
 import re
-from hashlib import sha256
-from binascii import hexlify, unhexlify
-from cryptography.hazmat.primitives.ciphers.aead import ChaCha20Poly1305
-
-from apps.cli import logger
 
 
 class Blob:

apps/cli/logger.py

@@ -1,33 +0,0 @@
-import logging
-import json
-import time
-
-
-class StructuredMessage(object):
-    def __init__(self, message, **kwargs):
-        self.message = message
-        self.time = time.asctime()
-        self.kwargs = kwargs
-
-    def __str__(self):
-        return json.dumps({**self.kwargs, "message": self.message, "time": self.time})
-
-
-class Logger(object):
-    def __init__(self, actor=None):
-        self.actor = actor
-
-    def _add_prefix(self, msg):
-        return msg if self.actor is None else "[{}] {}".format(self.actor, msg)
-
-    def info(self, msg, **kwargs):
-        logging.info(StructuredMessage(self._add_prefix(msg), actor=self.actor, **kwargs))
-
-    def debug(self, msg, **kwargs):
-        logging.debug(StructuredMessage(self._add_prefix(msg), actor=self.actor, **kwargs))
-
-    def error(self, msg, **kwargs):
-        logging.error(StructuredMessage(self._add_prefix(msg), actor=self.actor, **kwargs))
-
-    def warning(self, msg, **kwargs):
-        logging.warning(StructuredMessage(self._add_prefix(msg), actor=self.actor, **kwargs))

apps/cli/pisa_cli.py

@@ -17,15 +17,16 @@ from apps.cli import (
     CLI_PRIVATE_KEY,
     PISA_PUBLIC_KEY,
     APPOINTMENTS_FOLDER_NAME,
-    logger,
 )
 
+from common.logger import Logger
 from common.constants import LOCATOR_LEN_HEX
 from common.cryptographer import Cryptographer
 from common.tools import check_sha256_hex_format
 
 
 HTTP_OK = 200
+logger = Logger("Client")
 
 
 # FIXME: TESTING ENDPOINT, WON'T BE THERE IN PRODUCTION