mirror of https://github.com/afkcodes/zspotify.git
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import sys
|
|
from enum import Enum
|
|
from tqdm import tqdm
|
|
|
|
from config import *
|
|
from zspotify import ZSpotify
|
|
|
|
|
|
class PrintChannel(Enum):
|
|
SPLASH = PRINT_SPLASH
|
|
SKIPS = PRINT_SKIPS
|
|
DOWNLOAD_PROGRESS = PRINT_DOWNLOAD_PROGRESS
|
|
ERRORS = PRINT_ERRORS
|
|
WARNINGS = PRINT_WARNINGS
|
|
DOWNLOADS = PRINT_DOWNLOADS
|
|
API_ERRORS = PRINT_API_ERRORS
|
|
PROGRESS_INFO = PRINT_PROGRESS_INFO
|
|
|
|
|
|
ERROR_CHANNEL = [PrintChannel.ERRORS, PrintChannel.API_ERRORS]
|
|
|
|
|
|
class Printer:
|
|
@staticmethod
|
|
def print(channel: PrintChannel, msg: str) -> None:
|
|
if ZSpotify.CONFIG.get(channel.value):
|
|
if channel in ERROR_CHANNEL:
|
|
print(msg, file=sys.stderr)
|
|
else:
|
|
print(msg)
|
|
|
|
@staticmethod
|
|
def print_loader(channel: PrintChannel, msg: str) -> None:
|
|
if ZSpotify.CONFIG.get(channel.value):
|
|
print(msg, flush=True, end="")
|
|
|
|
@staticmethod
|
|
def progress(iterable=None, desc=None, total=None, unit='it', disable=False, unit_scale=False, unit_divisor=1000):
|
|
if not ZSpotify.CONFIG.get(PrintChannel.DOWNLOAD_PROGRESS.value):
|
|
disable = True
|
|
return tqdm(iterable=iterable, desc=desc, total=total, disable=disable, unit=unit, unit_scale=unit_scale, unit_divisor=unit_divisor)
|