10 logging

.gitignore

@@ -1,5 +1,6 @@
 # Dev environements
 .idea
+test_pyaml_logging.conf
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

pyaml/__init__.py

@@ -12,7 +12,19 @@
 __author__ = "pyAML collaboration"
 __author_email__ = ""
 
+import logging.config
+import os
 from pyaml.exception import PyAMLException
 
-__all__ = [__version__, "YamlLoader", PyAMLException]
+__all__ = [__version__, PyAMLException]
 
+
+config_file = os.getenv("PYAML_LOG_CONFIG", "pyaml_logging.conf")
+
+if os.path.exists(config_file):
+    logging.config.fileConfig(config_file, disable_existing_loggers=False)
+
+logger = logging.getLogger("pyaml")
+level = os.getenv("PYAML_LOG_LEVEL", "").upper()
+if len(level)>0:
+    logger.setLevel(getattr(logging, level, logging.WARNING))

pyaml/pyaml.py

@@ -1,6 +1,7 @@
 """
 PyAML main class
 """
+import logging
 
 from pydantic import BaseModel,ConfigDict
 from .instrument import Instrument
@@ -11,6 +12,8 @@
 # Define the main class name for this module
 PYAMLCLASS = "PyAML"
 
+logger = logging.getLogger(__name__)
+
 class ConfigModel(BaseModel):
 
     model_config = ConfigDict(arbitrary_types_allowed=True)
@@ -32,14 +35,15 @@ def get(self,name:str) -> Instrument:
         raise Exception(f"Instrument {name} not defined")
       return self.INSTRUMENTS[name]
 
-def pyaml(fileName:str) -> PyAML:
+def pyaml(filename:str) -> PyAML:
     """Load an accelerator middle layer"""
+    logger.log(logging.INFO, f"Loading PyAML from file '{filename}'")
 
     # Asume that all files are referenced from folder where main AML file is stored
-    if not os.path.exists(fileName):
-       raise Exception(f"{fileName} file nnot found")
-    rootfolder = os.path.abspath(os.path.dirname(fileName))
+    if not os.path.exists(filename):
+       raise Exception(f"{filename} file not found")
+    rootfolder = os.path.abspath(os.path.dirname(filename))
     set_root_folder(rootfolder)
-    aml_cfg = load(os.path.basename(fileName))
+    aml_cfg = load(os.path.basename(filename))
     aml:PyAML = depthFirstBuild(aml_cfg)
     return aml