diff --git a/main.py b/main.py index 652cfce..e1c1442 100644 --- a/main.py +++ b/main.py @@ -5,15 +5,13 @@ Entry point for running Pelican or Glory device simulators. """ import argparse -import logging import sys -from glory import GlorySimulator +from loguru import logger # Import simulators -from pelican import PelicanSimulator - -LOG_LEVEL = logging.DEBUG +from source.glory import GlorySimulator +from source.pelican import PelicanSimulator def main(): @@ -69,24 +67,26 @@ Examples: args = parser.parse_args() - # Setup centralized logger - logging.basicConfig( - level=LOG_LEVEL, - format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", + # Configure loguru logger + logger.remove() # Remove default handler + logger.add( + sys.stderr, + format=( + "{time:YYYY-MM-DD HH:mm:ss.SSS} | " + "{level: <8} | " + "{message} " + "({name}:{function}:{line})" + ), + level="DEBUG", ) - logger = logging.getLogger(f"{args.simulator}_simulator") # Run the appropriate simulator try: if args.simulator == "pelican": - simulator = PelicanSimulator( - logger=logger, port=args.port, baudrate=args.baudrate - ) + simulator = PelicanSimulator(port=args.port, baudrate=args.baudrate) simulator.run() elif args.simulator == "glory": - simulator = GlorySimulator( - logger=logger, port=args.port, baudrate=args.baudrate - ) + simulator = GlorySimulator(port=args.port, baudrate=args.baudrate) simulator.run() except KeyboardInterrupt: logger.info("Simulator stopped by user") diff --git a/glory.py b/source/glory.py similarity index 100% rename from glory.py rename to source/glory.py diff --git a/pelican.py b/source/pelican.py similarity index 100% rename from pelican.py rename to source/pelican.py