66 lines
1.4 KiB
Python
66 lines
1.4 KiB
Python
from datetime import datetime
|
|
from pathlib import Path
|
|
import json
|
|
import logging
|
|
|
|
from .shared import DEBUG, DEBUG_LOGGING, DEBUG_DUMP, DEBUG_TRACE
|
|
from .config import config, read_config, write_config
|
|
from .enums.colors import BColors
|
|
from .path_manager import LOCATIONS
|
|
|
|
"""
|
|
IO functions
|
|
"""
|
|
|
|
def _apply_color(msg: str, color: BColors) -> str:
|
|
if color is BColors.ENDC:
|
|
return msg
|
|
return color.value + msg + BColors.ENDC.value
|
|
|
|
|
|
def output(msg: str, color: BColors = BColors.ENDC):
|
|
print(_apply_color(msg, color))
|
|
|
|
|
|
def user_input(msg: str, color: BColors = BColors.ENDC):
|
|
return input(_apply_color(msg, color)).strip()
|
|
|
|
|
|
def dump_to_file(file_name: str, payload: str, is_json: bool = False, exit_after_dump: bool = False):
|
|
if not DEBUG_DUMP:
|
|
return
|
|
|
|
path = Path(LOCATIONS.TEMP_DIRECTORY, file_name)
|
|
logging.warning(f"dumping {file_name} to: \"{path}\"")
|
|
|
|
if is_json and isinstance(payload, str):
|
|
payload = json.loads(payload)
|
|
|
|
if isinstance(payload, dict):
|
|
payload = json.dumps(payload, indent=4)
|
|
|
|
with path.open("w") as f:
|
|
f.write(payload)
|
|
|
|
if exit_after_dump:
|
|
exit()
|
|
|
|
|
|
def trace(msg: str):
|
|
if DEBUG_TRACE:
|
|
logging.debug(msg)
|
|
|
|
output("trace: " + msg, BColors.OKBLUE)
|
|
|
|
|
|
"""
|
|
misc functions
|
|
"""
|
|
|
|
def get_current_millis() -> int:
|
|
dt = datetime.now()
|
|
return int(dt.microsecond / 1_000)
|
|
|
|
|
|
def get_unix_time() -> int:
|
|
return int(datetime.now().timestamp()) |