first commit

core/config.py

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+
+"""
+class from config file
+"""
+
+from pydantic import Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
+from rich.console import Console
+
+console = Console()
+
+
+class Settings(BaseSettings):
+
+    model_config = SettingsConfigDict()
+
+    timezone: str = Field(alias="TIMEZONE", default="Europe/Madrid")
+
+    agilent_ip: str = '192.168.2.26'
+    loop_time: int = 5
+
+    # Finot API platform
+    api_finot: bool = Field(alias="API_FINOT", default=False)
+
+    api_finot_loginId: str = "xxxxxxxxxxxxxx-in.eu" # login, please change
+    api_finot_password: str = "XXXXXXXXXXX" # password, please change
+    api_finot_refresh_token: str = "RhWyBEf_xxxxxxxxx-xxxxxxxxxxxxxxxxxxxxxxxxxxx" # refresh token
+
+    # Agilent channels mapping
+    agilent_output: dict = {
+        'TC1':101,
+        'TC2':102,
+        'VDC1':103
+    }
+
+    agilent_requests: dict = {
+        101: 'MEAS:TEMP?',
+        102: 'MEAS:TEMP?',
+        103: 'MEAS:VOLT:DC?'
+    }

core/formatter.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+import logging
+
+
+class CustomFormatter(logging.Formatter):
+    blue = "\x1b[38;5;39m"
+    grey = "\x1b[38;20m"
+    yellow = "\x1b[33;20m"
+    red = "\x1b[31;20m"
+    green = "\x1b[;32m"
+    bold_red = "\x1b[31;1m"
+    reset = "\x1b[0m"
+    format_error = "%(asctime)s - %(name)s - %(levelname)s - %(message)s (%(filename)s:%(lineno)d)"
+    format = "%(asctime)s - %(name)s "
+    FORMATS = {
+        logging.DEBUG: format + blue + "OUTPUT: " + reset + " %(message)s",
+        logging.INFO: format + green + "%(levelname)s: " + reset + " %(message)s",
+        logging.WARNING: format + yellow + "%(levelname)s: " + reset + " %(message)s",
+        logging.ERROR: red + format_error + reset,
+        logging.CRITICAL: bold_red + format_error + reset,
+    }
+
+    def format(self, record):
+        log_fmt = self.FORMATS.get(record.levelno)
+        formatter = logging.Formatter(log_fmt)
+        return formatter.format(record)
+
+
+class Logger:
+    def __init__(self, name):
+        self.logger = logging.getLogger(name)
+        self.logger.setLevel(logging.DEBUG)
+
+        ch = logging.StreamHandler()
+        ch.setLevel(logging.DEBUG)
+        ch.setFormatter(CustomFormatter())
+
+        self.logger.addHandler(ch)
+
+    def info(self, msg):
+        return self.logger.info(msg)
+
+    def warning(self, msg):
+        return self.logger.warning(msg)
+
+    def error(self, msg):
+        return self.logger.error(msg)
+    
+    def critical(self, msg):
+        return self.logger.critical(msg)
+    
+    def debug(self, msg):
+        return self.logger.debug(msg)