forked from DuarteMRAlves/AASMA-2022
-
Notifications
You must be signed in to change notification settings - Fork 0
/
log.py
28 lines (21 loc) · 1.08 KB
/
log.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import env
import entity
import logging
logging.basicConfig(format="t = %(timestep)s \t %(levelname)s \t %(name)s \t %(message)s")
def new(name: str, lvl: str = "info") -> logging.Logger:
logger = logging.getLogger(name=name)
lvl = getattr(logging, lvl.upper(), None)
if not isinstance(lvl, int):
raise ValueError(f"Invalid log level: {lvl}")
logger.setLevel(lvl)
return logger
def create_taxi(logger: logging.Logger, t: int, taxi: entity.Taxi):
logger.info("Created %r", taxi, extra={"timestep": t})
def create_passenger(logger: logging.Logger, t: int, passenger: entity.Passenger):
logger.info("Created %r", passenger, extra={"timestep": t})
def choosen_action(logger: logging.Logger, t: int, agent: int, action: "env.Action"):
logger.info("Agent %d wants to %r", agent, action, extra={"timestep": t})
def taxi(logger: logging.Logger, t: int, taxi: entity.Taxi):
logger.info("Taxi %r", taxi, extra={"timestep": t})
def passenger(logger: logging.Logger, t: int, passenger: entity.Passenger):
logger.info("Passenger %r", passenger, extra={"timestep": t})