diff --git a/src/pms/__init__.py b/src/pms/__init__.py index b374b40..4549e10 100644 --- a/src/pms/__init__.py +++ b/src/pms/__init__.py @@ -1,7 +1,5 @@ import logging -import os -logging.basicConfig(level=os.getenv("LEVEL", "WARNING")) logger = logging.getLogger(__name__) diff --git a/src/pms/cli.py b/src/pms/cli.py index a5cfc8d..d857644 100644 --- a/src/pms/cli.py +++ b/src/pms/cli.py @@ -1,3 +1,5 @@ +import logging +import os import sys from datetime import datetime from enum import Enum @@ -25,6 +27,9 @@ main.command(name=ep.name)(ep.load()) +logging.basicConfig(level=os.getenv("LEVEL", "WARNING")) + + def version_callback(value: bool): # pragma: no cover if not value: return @@ -45,8 +50,7 @@ def callback( version: Optional[bool] = Option(None, "--version", "-V", callback=version_callback), ): """Read serial sensor""" - if debug: # pragma: no cover - logger.setLevel("DEBUG") + logger.setLevel("DEBUG" if debug else os.getenv("LEVEL", "WARNING")) ctx.obj = {"reader": SensorReader(model, port, seconds, samples)} @@ -127,3 +131,7 @@ def csv( csv.write("time,sensor,hex\n") for raw in reader(raw=True): csv.write(f"{raw.time},{sensor_name},{raw.hex}\n") + + +if __name__ == "__main__": # pragma: no cover + main()