cinnamon-cli 0.2.19__py3-none-any.whl

cinnamon/config.py

@@ -0,0 +1,121 @@
+import json
+import os
+import sys
+from pathlib import Path
+
+from .errors import InvalidConfig
+
+
+def _config_dir():
+    if sys.platform == "win32":
+        base = Path(os.environ.get("APPDATA", Path.home() / "AppData" / "Roaming"))
+    elif os.environ.get("TERMUX_VERSION"):
+        base = Path.home() / ".config"
+    elif xdg := os.environ.get("XDG_CONFIG_HOME"):
+        base = Path(xdg)
+    else:
+        base = Path.home() / ".config"
+    return base / "cinnamon"
+
+
+CONFIG_DIR = _config_dir()
+CONFIG_FILE = CONFIG_DIR / "config.json"
+SCRAPERS_DIR = CONFIG_DIR / "scrapers"
+
+DEFAULTS = {
+    "tmdb_api_key": "",
+    "default_scraper": "webstream",
+    "default_player": "auto",
+    "theme": "cinnamon",
+    "scrapers": {},
+}
+
+
+THEMES = {
+    "cinnamon": {
+        "accent": "orange1",
+        "accent2": "gold1",
+        "success": "green",
+        "error": "red",
+        "info": "cyan",
+        "dim": "grey50",
+        "panel": "bright_yellow",
+        "border": "orange1",
+    },
+    "ocean": {
+        "accent": "deep_sky_blue1",
+        "accent2": "cyan",
+        "success": "spring_green2",
+        "error": "red",
+        "info": "light_cyan1",
+        "dim": "grey50",
+        "panel": "deep_sky_blue1",
+        "border": "blue",
+    },
+    "mono": {
+        "accent": "white",
+        "accent2": "bright_white",
+        "success": "green",
+        "error": "red",
+        "info": "bright_white",
+        "dim": "grey35",
+        "panel": "bright_white",
+        "border": "grey50",
+    },
+}
+
+
+def get_theme():
+    cfg = load_config()
+    name = cfg.get("theme", "cinnamon")
+    return THEMES.get(name, THEMES["cinnamon"])
+
+
+def load_config():
+    if CONFIG_FILE.exists():
+        try:
+            data = json.loads(CONFIG_FILE.read_text())
+            return {**DEFAULTS, **data}
+        except (json.JSONDecodeError, OSError) as e:
+            raise InvalidConfig(str(e))
+    return dict(DEFAULTS)
+
+
+def save_config(config):
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    merged = {**DEFAULTS, **config}
+    CONFIG_FILE.write_text(json.dumps(merged, indent=2))
+
+
+def get_tmdb_api_key():
+    api_key = os.getenv("TMDB_API_KEY")
+    if api_key:
+        return api_key
+    return load_config().get("tmdb_api_key", "")
+
+
+def set_tmdb_api_key(api_key):
+    config = load_config()
+    config["tmdb_api_key"] = api_key
+    save_config(config)
+
+
+def get_scraper_config(scraper_name):
+    config = load_config()
+    return config.get("scrapers", {}).get(scraper_name, {})
+
+
+def set_scraper_config(scraper_name, key, value):
+    config = load_config()
+    if "scrapers" not in config:
+        config["scrapers"] = {}
+    if scraper_name not in config["scrapers"]:
+        config["scrapers"][scraper_name] = {}
+    config["scrapers"][scraper_name][key] = value
+    save_config(config)
+
+
+def get_ensured_dirs():
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    SCRAPERS_DIR.mkdir(parents=True, exist_ok=True)
+    return CONFIG_DIR, SCRAPERS_DIR

cinnamon/downloads.py

@@ -0,0 +1,83 @@
+import json
+import os
+import sys
+import time
+import uuid
+
+_DATA_DIR = None
+
+
+def _data_dir():
+    global _DATA_DIR
+    if _DATA_DIR is None:
+        if sys.platform == "win32":
+            base = os.environ.get("APPDATA", os.path.expanduser("~"))
+        else:
+            base = os.path.join(os.path.expanduser("~"), ".local", "share")
+        _DATA_DIR = os.path.join(base, "cinnamon")
+        os.makedirs(_DATA_DIR, exist_ok=True)
+    return _DATA_DIR
+
+
+def _db():
+    return os.path.join(_data_dir(), "downloads.json")
+
+
+def _load():
+    try:
+        with open(_db(), encoding="utf-8") as f:
+            return json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError):
+        return []
+
+
+def _save(dls):
+    with open(_db(), "w", encoding="utf-8") as f:
+        json.dump(dls, f, indent=2, ensure_ascii=False)
+
+
+def create(info):
+    dls = _load()
+    entry = {
+        "id": str(uuid.uuid4())[:8],
+        "title": info.get("title", "Untitled"),
+        "url": info.get("url"),
+        "tv_id": info.get("tv_id"),
+        "season": info.get("season"),
+        "episode": info.get("episode"),
+        "quality": info.get("quality"),
+        "referer": info.get("referer"),
+        "timestamp": time.time(),
+        "status": "queued",
+    }
+    dls.append(entry)
+    _save(dls)
+    return entry["id"]
+
+
+def update(download_id, **kwargs):
+    dls = _load()
+    for d in dls:
+        if d["id"] == download_id:
+            d.update(kwargs)
+            break
+    _save(dls)
+
+
+def list_all(status=None):
+    dls = _load()
+    if status:
+        return [d for d in dls if d.get("status") == status]
+    return dls
+
+
+def get(download_id):
+    for d in _load():
+        if d["id"] == download_id:
+            return d
+    return None
+
+
+def remove(download_id):
+    dls = _load()
+    _save([d for d in dls if d["id"] != download_id])

cinnamon/errors.py

@@ -0,0 +1,145 @@
+from typing import Optional
+
+
+class CinnamonError(Exception):
+    """Base for all cinnamon errors."""
+
+
+class ConfigError(CinnamonError):
+    """Configuration issues."""
+
+
+class MissingAPIKey(ConfigError):
+    def __init__(self):
+        super().__init__("TMDB API key not configured.")
+
+
+class InvalidConfig(ConfigError):
+    def __init__(self, msg=""):
+        super().__init__(msg or "Configuration file is invalid.")
+
+
+class TMDBError(CinnamonError):
+    """Base for TMDB API errors."""
+
+    def __init__(self, message, status_code=None, details=None):
+        self.status_code = status_code
+        self.details = details
+        super().__init__(message)
+
+
+class TMDBConnectionError(TMDBError):
+    def __init__(self, cause=None):
+        msg = "Could not connect to TMDB. Check your internet connection."
+        if cause:
+            msg = f"{msg}\n  Details: {cause}"
+        super().__init__(msg)
+
+
+class TMDBAuthError(TMDBError):
+    def __init__(self):
+        super().__init__(
+            "Your TMDB API key is invalid or has been revoked.\n"
+            "  Run [bold]cinnamon setup[/bold] to update it.",
+            status_code=401,
+        )
+
+
+class TMDBRateLimitError(TMDBError):
+    def __init__(self, retry_after=2):
+        self.retry_after = retry_after
+        super().__init__(
+            f"TMDB rate limit hit. Auto-retrying in {retry_after}s...",
+            status_code=429,
+        )
+
+
+class TMDBNotFoundError(TMDBError):
+    def __init__(self, resource):
+        super().__init__(
+            f"{resource} not found on TMDB. The ID may be wrong or removed.",
+            status_code=404,
+        )
+
+
+class TMDBAPIError(TMDBError):
+    def __init__(self, status_code, message=""):
+        super().__init__(
+            message or f"TMDB API returned error {status_code}.",
+            status_code=status_code,
+        )
+
+
+class ScraperError(CinnamonError):
+    """Base for scraper errors."""
+
+
+class ScraperNotFoundError(ScraperError):
+    def __init__(self, name, available=None):
+        msg = f"Scraper [bold]{name}[/bold] not found."
+        if available:
+            msg += f"\n  Available: {', '.join(available)}"
+        super().__init__(msg)
+
+
+class ScraperNetworkError(ScraperError):
+    def __init__(self, scraper, cause=None):
+        msg = f"Network error in scraper [bold]{scraper}[/bold]."
+        if cause:
+            msg += f"\n  Details: {cause}"
+        super().__init__(msg)
+
+
+class ScraperAuthError(ScraperError):
+    def __init__(self, scraper):
+        super().__init__(
+            f"Scraper [bold]{scraper}[/bold] requires authentication.\n"
+            f"  Configure it via [bold]cinnamon config scraper {scraper} --key ...[/bold]"
+        )
+
+
+class ScraperRateLimitError(ScraperError):
+    def __init__(self, scraper, retry_after=None):
+        msg = f"Scraper [bold]{scraper}[/bold] is rate-limited."
+        if retry_after:
+            msg += f" Retry in {retry_after}s."
+        super().__init__(msg)
+
+
+class ScraperParseError(ScraperError):
+    def __init__(self, scraper, cause=None):
+        msg = f"Failed to parse response from scraper [bold]{scraper}[/bold]."
+        if cause:
+            msg += f"\n  Details: {cause}"
+        super().__init__(msg)
+
+
+class ScraperNoStreamError(ScraperError):
+    def __init__(self, scraper, message=None):
+        super().__init__(
+            message or f"Scraper [bold]{scraper}[/bold] found no streams for this episode."
+        )
+
+
+class PlayerError(CinnamonError):
+    """Base for player errors."""
+
+
+class PlayerNotFoundError(PlayerError):
+    def __init__(self, player="auto"):
+        msg = (
+            f"No media player found."
+        )
+        if player in ("mpv", "auto"):
+            msg += "\n  Install mpv: https://mpv.io"
+        if player in ("vlc", "auto"):
+            msg += "\n  Install VLC: https://videolan.org"
+        super().__init__(msg)
+
+
+class PlayerLaunchError(PlayerError):
+    def __init__(self, player, cause=None):
+        msg = f"Failed to launch [bold]{player}[/bold]."
+        if cause:
+            msg += f"\n  Details: {cause}"
+        super().__init__(msg)

cinnamon/history.py

@@ -0,0 +1,66 @@
+import json
+import os
+import sys
+import time
+
+_DATA_DIR = None
+
+
+def _data_dir():
+    global _DATA_DIR
+    if _DATA_DIR is None:
+        if sys.platform == "win32":
+            base = os.environ.get("APPDATA", os.path.expanduser("~"))
+        else:
+            base = os.path.join(os.path.expanduser("~"), ".local", "share")
+        _DATA_DIR = os.path.join(base, "cinnamon")
+        os.makedirs(_DATA_DIR, exist_ok=True)
+    return _DATA_DIR
+
+
+def _db():
+    return os.path.join(_data_dir(), "history.json")
+
+
+def _load():
+    try:
+        with open(_db(), encoding="utf-8") as f:
+            return json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError):
+        return {}
+
+
+def _save(h):
+    with open(_db(), "w", encoding="utf-8") as f:
+        json.dump(h, f, indent=2, ensure_ascii=False)
+
+
+def set_history(title, season, episode, scraper=None, translation=None, quality=None):
+    h = _load()
+    h[title] = {
+        "season": season,
+        "episode": episode,
+        "timestamp": time.time(),
+        "scraper": scraper,
+        "translation": translation,
+        "quality": quality,
+    }
+    _save(h)
+
+
+def get_history(title):
+    return _load().get(title)
+
+
+def list_history():
+    h = _load()
+    return sorted(h.items(), key=lambda kv: kv[1].get("timestamp", 0), reverse=True)
+
+
+def clear_history(title=None):
+    h = _load()
+    if title:
+        h.pop(title, None)
+    else:
+        h.clear()
+    _save(h)