79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
|
import logging
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from datetime import datetime
|
||
|
|
|
||
|
|
import click
|
||
|
|
import coloredlogs
|
||
|
|
|
||
|
|
from vinetrimmer.config import directories, filenames # isort: split
|
||
|
|
from vinetrimmer.commands import dl
|
||
|
|
|
||
|
|
|
||
|
|
@click.command(context_settings=dict(
|
||
|
|
allow_extra_args=True,
|
||
|
|
ignore_unknown_options=True,
|
||
|
|
max_content_width=116, # max PEP8 line-width, -4 to adjust for initial indent
|
||
|
|
))
|
||
|
|
@click.option("--debug", is_flag=True, default=False,
|
||
|
|
help="Enable DEBUG level logs on the console. This is always enabled for log files.")
|
||
|
|
def main(debug):
|
||
|
|
"""
|
||
|
|
vinetrimmer is the most convenient command-line program to
|
||
|
|
download videos from Widevine DRM-protected video platforms.
|
||
|
|
"""
|
||
|
|
LOG_FORMAT = "{asctime} [{levelname[0]}] {name} : {message}"
|
||
|
|
LOG_DATE_FORMAT = "%Y-%m-%d %H:%M:%S"
|
||
|
|
LOG_STYLE = "{"
|
||
|
|
|
||
|
|
def log_exit(self, msg, *args, **kwargs):
|
||
|
|
self.critical(msg, *args, **kwargs)
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
logging.Logger.exit = log_exit
|
||
|
|
|
||
|
|
os.makedirs(directories.logs, exist_ok=True)
|
||
|
|
logging.basicConfig(
|
||
|
|
level=logging.DEBUG,
|
||
|
|
format=LOG_FORMAT,
|
||
|
|
datefmt=LOG_DATE_FORMAT,
|
||
|
|
style=LOG_STYLE,
|
||
|
|
handlers=[logging.FileHandler(
|
||
|
|
os.path.join(directories.logs, filenames.log.format(time=datetime.now().strftime("%Y%m%d-%H%M%S"))),
|
||
|
|
encoding='utf-8'
|
||
|
|
)]
|
||
|
|
)
|
||
|
|
|
||
|
|
coloredlogs.install(
|
||
|
|
level=logging.DEBUG if debug else logging.INFO,
|
||
|
|
fmt=LOG_FORMAT,
|
||
|
|
datefmt=LOG_DATE_FORMAT,
|
||
|
|
style=LOG_STYLE,
|
||
|
|
handlers=[logging.StreamHandler()],
|
||
|
|
)
|
||
|
|
|
||
|
|
log = logging.getLogger("vt")
|
||
|
|
|
||
|
|
log.info("vinetrimmer - Widevine DRM downloader and decrypter")
|
||
|
|
log.info(f"[Root Config] : {filenames.user_root_config}")
|
||
|
|
log.info(f"[Service Configs] : {directories.service_configs}")
|
||
|
|
log.info(f"[Cookies] : {directories.cookies}")
|
||
|
|
log.info(f"[CDM Devices] : {directories.devices}")
|
||
|
|
log.info(f"[Cache] : {directories.cache}")
|
||
|
|
log.info(f"[Logs] : {directories.logs}")
|
||
|
|
log.info(f"[Temp Files] : {directories.temp}")
|
||
|
|
log.info(f"[Downloads] : {directories.downloads}")
|
||
|
|
|
||
|
|
os.environ['PATH'] = os.path.abspath('./binaries')
|
||
|
|
|
||
|
|
if len(sys.argv) > 1 and sys.argv[1].lower() == "dl":
|
||
|
|
sys.argv.pop(1)
|
||
|
|
|
||
|
|
dl()
|
||
|
|
|
||
|
|
# D:\PlayReady-Amazon-Tool-main\.venv\Scripts\python.exe -X pycache_prefix=C:\Users\Aswin\AppData\Local\JetBrains\PyCharm2024.3\cpython-cache "C:/Program Files (x86)/JetBrains/PyCharm 2024.2.4/plugins/python-ce/helpers/pydev/pydevd.py" --port 42000 --module --multiprocess --save-signatures --qt-support=auto --file poetry run vt dl --no-cache --keys AMZN 0H7LY5ZKKBM1MIW0244WE9O2C4
|
||
|
|
# Above seems to work
|
||
|
|
if __name__ == "__main__":
|
||
|
|
#sys.argv = ["vinetrimmer", "dl", "--no-cache", "--keys", "AMZN", "0H7LY5ZKKBM1MIW0244WE9O2C4"]
|
||
|
|
main()
|