feat: cleaned up
This commit is contained in:
parent
fe81fee82b
commit
2e5a46c3f6
@ -26,7 +26,7 @@ from ..utils.config import main_settings, logging_settings
|
|||||||
from ..utils.shared import DEBUG
|
from ..utils.shared import DEBUG
|
||||||
|
|
||||||
if DEBUG:
|
if DEBUG:
|
||||||
from ..utils.debug_utils import dump_to_file
|
from ..utils import dump_to_file
|
||||||
|
|
||||||
|
|
||||||
def _parse_artist_url(url: str) -> str:
|
def _parse_artist_url(url: str) -> str:
|
||||||
|
@ -23,8 +23,7 @@ from ..objects import (
|
|||||||
DatabaseObject
|
DatabaseObject
|
||||||
)
|
)
|
||||||
from ..utils.shared import DEBUG
|
from ..utils.shared import DEBUG
|
||||||
if DEBUG:
|
from ..utils import dump_to_file
|
||||||
from ..utils.debug_utils import dump_to_file
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -17,8 +17,7 @@ from ...utils.config import main_settings, youtube_settings, logging_settings
|
|||||||
from ...utils.shared import DEBUG, DEBUG_YOUTUBE_INITIALIZING
|
from ...utils.shared import DEBUG, DEBUG_YOUTUBE_INITIALIZING
|
||||||
from ...utils import get_current_millis
|
from ...utils import get_current_millis
|
||||||
|
|
||||||
if DEBUG:
|
from ...utils import dump_to_file
|
||||||
from ...utils.debug_utils import dump_to_file
|
|
||||||
|
|
||||||
from ...objects import Source, DatabaseObject
|
from ...objects import Source, DatabaseObject
|
||||||
from ..abstract import Page
|
from ..abstract import Page
|
||||||
|
@ -1,13 +1,19 @@
|
|||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
from pathlib import Path
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from .shared import DEBUG, DEBUG_LOGGING, DEBUG_PAGES
|
||||||
from .config import config, read_config, write_config
|
from .config import config, read_config, write_config
|
||||||
from .enums.colors import BColors
|
from .enums.colors import BColors
|
||||||
|
from .path_manager import LOCATIONS
|
||||||
|
|
||||||
|
DEBUG = False
|
||||||
|
|
||||||
"""
|
"""
|
||||||
Here are all global important functions.
|
IO functions
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
|
||||||
def _apply_color(msg: str, color: BColors) -> str:
|
def _apply_color(msg: str, color: BColors) -> str:
|
||||||
if color is BColors.ENDC:
|
if color is BColors.ENDC:
|
||||||
return msg
|
return msg
|
||||||
@ -22,6 +28,31 @@ def user_input(msg: str, color: BColors = BColors.ENDC):
|
|||||||
return input(_apply_color(msg, color)).strip()
|
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_PAGES:
|
||||||
|
return
|
||||||
|
|
||||||
|
path = Path(LOCATIONS.TEMP_DIRECTORY, file_name)
|
||||||
|
logging.warning(f"dumping {file_name} to: \"{path}\"")
|
||||||
|
|
||||||
|
if is_json:
|
||||||
|
payload = json.dumps(json.loads(payload), indent=4)
|
||||||
|
|
||||||
|
if isinstance(payload, dict):
|
||||||
|
payload = json.dumps(payload, indent=4)
|
||||||
|
|
||||||
|
with path.open("w") as f:
|
||||||
|
f.write(payload)
|
||||||
|
|
||||||
|
if exit_after_dump:
|
||||||
|
exit()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
"""
|
||||||
|
misc functions
|
||||||
|
"""
|
||||||
|
|
||||||
def get_current_millis() -> int:
|
def get_current_millis() -> int:
|
||||||
dt = datetime.now()
|
dt = datetime.now()
|
||||||
return int(dt.microsecond / 1_000)
|
return int(dt.microsecond / 1_000)
|
||||||
|
@ -1,21 +0,0 @@
|
|||||||
from pathlib import Path
|
|
||||||
import json
|
|
||||||
|
|
||||||
from .path_manager import LOCATIONS
|
|
||||||
|
|
||||||
|
|
||||||
def dump_to_file(file_name: str, payload: str, is_json: bool = False, exit_after_dump: bool = True):
|
|
||||||
path = Path(LOCATIONS.TEMP_DIRECTORY, file_name)
|
|
||||||
print(f"Dumping payload to: \"{path}\"")
|
|
||||||
|
|
||||||
if is_json:
|
|
||||||
payload = json.dumps(json.loads(payload), indent=4)
|
|
||||||
|
|
||||||
if isinstance(payload, dict):
|
|
||||||
payload = json.dumps(payload, indent=4)
|
|
||||||
|
|
||||||
with path.open("w") as f:
|
|
||||||
f.write(payload)
|
|
||||||
|
|
||||||
if exit_after_dump:
|
|
||||||
exit()
|
|
Loading…
Reference in New Issue
Block a user