mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-14 21:15:06 +00:00
imSp4rky
e3f4e52444
Headless serve with plain stderr logging; /api/dashboard/* (status, sessions, jobs, logs, SSE/poll events) behind a separate serve.dashboard.key.
107 lines
3.9 KiB
Python
107 lines
3.9 KiB
Python
import atexit
|
|
import logging
|
|
import sys
|
|
from datetime import datetime
|
|
|
|
import click
|
|
import urllib3
|
|
from rich import traceback
|
|
from rich.console import Group
|
|
from rich.padding import Padding
|
|
from rich.text import Text
|
|
from urllib3.exceptions import InsecureRequestWarning
|
|
|
|
from unshackle.core import __code_hash__, __version__
|
|
from unshackle.core.commands import Commands
|
|
from unshackle.core.config import config
|
|
from unshackle.core.console import ComfyRichHandler, console
|
|
from unshackle.core.constants import context_settings
|
|
from unshackle.core.update_checker import UpdateChecker
|
|
from unshackle.core.utilities import close_debug_logger, init_debug_logger
|
|
|
|
|
|
@click.command(cls=Commands, invoke_without_command=True, context_settings=context_settings)
|
|
@click.option("-v", "--version", is_flag=True, default=False, help="Print version information.")
|
|
@click.option("-d", "--debug", is_flag=True, default=False, help="Enable DEBUG level logs and JSON debug logging.")
|
|
def main(version: bool, debug: bool) -> None:
|
|
"""unshackle: Modular Movie, TV, and Music Archival Software."""
|
|
debug_logging_enabled = debug or config.debug
|
|
|
|
logging.basicConfig(
|
|
level=logging.DEBUG if debug else logging.INFO,
|
|
format="%(message)s",
|
|
handlers=[
|
|
ComfyRichHandler(
|
|
show_time=False,
|
|
show_path=debug,
|
|
console=console,
|
|
rich_tracebacks=True,
|
|
tracebacks_suppress=[click],
|
|
log_renderer=console._log_render, # noqa
|
|
)
|
|
],
|
|
)
|
|
|
|
if debug_logging_enabled:
|
|
init_debug_logger(enabled=True)
|
|
|
|
if debug and not config.debug_requests:
|
|
for noisy in ("urllib3", "urllib3.connectionpool", "requests", "rnet", "httpx", "httpcore", "hpack", "h2"):
|
|
logging.getLogger(noisy).setLevel(logging.WARNING)
|
|
|
|
urllib3.disable_warnings(InsecureRequestWarning)
|
|
|
|
traceback.install(console=console, width=80, suppress=[click])
|
|
|
|
if "--quiet" in sys.argv[1:] or "-q" in sys.argv[1:]:
|
|
return
|
|
|
|
console.print(
|
|
Padding(
|
|
Group(
|
|
Text(
|
|
r"▄• ▄▌ ▐ ▄ .▄▄ · ▄ .▄ ▄▄▄· ▄▄· ▄ •▄ ▄▄▌ ▄▄▄ ." + "\n"
|
|
r"█▪██▌•█▌▐█▐█ ▀. ██▪▐█▐█ ▀█ ▐█ ▌▪█▌▄▌▪██• ▀▄.▀·" + "\n"
|
|
r"█▌▐█▌▐█▐▐▌▄▀▀▀█▄██▀▐█▄█▀▀█ ██ ▄▄▐▀▀▄·██▪ ▐▀▀▪▄" + "\n"
|
|
r"▐█▄█▌██▐█▌▐█▄▪▐███▌▐▀▐█ ▪▐▌▐███▌▐█.█▌▐█▌▐▌▐█▄▄▌" + "\n"
|
|
r" ▀▀▀ ▀▀ █▪ ▀▀▀▀ ▀▀▀ · ▀ ▀ ·▀▀▀ ·▀ ▀.▀▀▀ ▀▀▀ ",
|
|
style="ascii.art",
|
|
),
|
|
f"v [repr.number]{__version__}[/]{f' ({__code_hash__})' if __code_hash__ else ''}"
|
|
f" - © 2025-{datetime.now().year} - github.com/unshackle-dl/unshackle",
|
|
),
|
|
(1, 11, 1, 10),
|
|
expand=True,
|
|
),
|
|
justify="center",
|
|
)
|
|
|
|
if version:
|
|
return
|
|
|
|
if config.update_checks:
|
|
try:
|
|
latest_version = UpdateChecker.check_for_updates_sync(__version__)
|
|
if latest_version:
|
|
console.print(
|
|
f"\n[yellow]Update available![/yellow] "
|
|
f"Current: {__version__} → Latest: [green]{latest_version}[/green]",
|
|
justify="center",
|
|
)
|
|
console.print(
|
|
"Visit: https://github.com/unshackle-dl/unshackle/releases/latest\n",
|
|
justify="center",
|
|
)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
@atexit.register
|
|
def cleanup():
|
|
"""Clean up resources on exit."""
|
|
close_debug_logger()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|