28 lines
799 B
Python
28 lines
799 B
Python
#!/usr/bin/python3
|
|
from datetime import datetime as dt
|
|
from sys import stdout, stderr
|
|
from typing import Union
|
|
|
|
|
|
def log(
|
|
message: str,
|
|
origin: str = "Unknown",
|
|
level: str = "LOG",
|
|
time: Union[dt, str] = "now",
|
|
) -> None:
|
|
if level in ["EXIT", "CRASH"]:
|
|
stream = stderr
|
|
else:
|
|
stream = stdout
|
|
if time == "now":
|
|
dtime = dt.now()
|
|
elif type(time) == str:
|
|
raise ValueError("Only \"now\" is an accepted string argument for time")
|
|
else:
|
|
dtime = time # type: dt #type: ignore
|
|
time = dtime.strftime("%H:%M:%S")
|
|
if not "\n" in message:
|
|
print(f"[{level}][{origin}][{time}] {message}", file=stream)
|
|
else:
|
|
for line in message.split("\n"):
|
|
print(f"[{level}][{origin}][{time}] {line}", file=stream)
|