Initial Commit

unshackle/core/__main__.py

@@ -0,0 +1,92 @@
+import atexit
+import logging
+from datetime import datetime
+from pathlib import Path
+
+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 __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.utilities import rotate_log_file
+
+LOGGING_PATH = None
+
+
+@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.")
+@click.option(
+    "--log",
+    "log_path",
+    type=Path,
+    default=config.directories.logs / config.filenames.log,
+    help="Log path (or filename). Path can contain the following f-string args: {name} {time}.",
+)
+def main(version: bool, debug: bool, log_path: Path) -> None:
+    """unshackle—Modular Movie, TV, and Music Archival Software."""
+    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 log_path:
+        global LOGGING_PATH
+        console.record = True
+        new_log_path = rotate_log_file(log_path)
+        LOGGING_PATH = new_log_path
+
+    urllib3.disable_warnings(InsecureRequestWarning)
+
+    traceback.install(console=console, width=80, suppress=[click])
+
+    console.print(
+        Padding(
+            Group(
+                Text(
+                    r"▄• ▄▌ ▐ ▄ .▄▄ ·  ▄ .▄ ▄▄▄·  ▄▄· ▄ •▄ ▄▄▌  ▄▄▄ ." + "\n"
+                    r"█▪██▌•█▌▐█▐█ ▀. ██▪▐█▐█ ▀█ ▐█ ▌▪█▌▄▌▪██•  ▀▄.▀·" + "\n"
+                    r"█▌▐█▌▐█▐▐▌▄▀▀▀█▄██▀▐█▄█▀▀█ ██ ▄▄▐▀▀▄·██▪  ▐▀▀▪▄" + "\n"
+                    r"▐█▄█▌██▐█▌▐█▄▪▐███▌▐▀▐█ ▪▐▌▐███▌▐█.█▌▐█▌▐▌▐█▄▄▌" + "\n"
+                    r" ▀▀▀ ▀▀ █▪ ▀▀▀▀ ▀▀▀ · ▀  ▀ ·▀▀▀ ·▀  ▀.▀▀▀  ▀▀▀ " + "\n",
+                    style="ascii.art",
+                ),
+                f"v[repr.number]{__version__}[/] - {datetime.now().year} - sp4rk.y",
+            ),
+            (1, 11, 1, 10),
+            expand=True,
+        ),
+        justify="center",
+    )
+
+    if version:
+        return
+
+
+@atexit.register
+def save_log():
+    if console.record and LOGGING_PATH:
+        # TODO: Currently semi-bust. Everything that refreshes gets duplicated.
+        console.save_text(LOGGING_PATH)
+
+
+if __name__ == "__main__":
+    main()