feat: able to specify db path
This commit is contained in:
+16
-3
@@ -13,7 +13,7 @@ from urllib.parse import quote
|
||||
import cyclopts
|
||||
from loguru import logger
|
||||
|
||||
from .config import enable_debug
|
||||
from .config import DB_PATH, enable_debug
|
||||
from .models import TrackMeta, CacheStatus
|
||||
from .mpris import get_current_track
|
||||
from .core import LrcManager
|
||||
@@ -29,10 +29,13 @@ app.register_install_completion_command()
|
||||
cache_app = cyclopts.App(name="cache", help="Manage the local SQLite cache.")
|
||||
app.command(cache_app)
|
||||
|
||||
manager = LrcManager()
|
||||
|
||||
# Global state set by the meta launcher
|
||||
_player: str | None = None
|
||||
_db_path: str | None = None
|
||||
|
||||
# Will be initialized before any command runs, safe to set to None here
|
||||
manager: LrcManager = None # type: ignore
|
||||
|
||||
|
||||
@app.meta.default
|
||||
@@ -51,11 +54,21 @@ def launcher(
|
||||
help="Target a specific MPRIS player using its DBus name or a portion thereof.",
|
||||
),
|
||||
] = None,
|
||||
db_path: Annotated[
|
||||
str | None,
|
||||
cyclopts.Parameter(
|
||||
name=["--db-path", "-c"],
|
||||
help=f"Custom path for the cache database file (default: {DB_PATH}).",
|
||||
),
|
||||
] = None,
|
||||
):
|
||||
global _player
|
||||
global _player, _db_path
|
||||
if debug:
|
||||
enable_debug()
|
||||
_player = player
|
||||
_db_path = str(Path(db_path).resolve()) if db_path else None
|
||||
global manager
|
||||
manager = LrcManager(db_path=_db_path)
|
||||
app(tokens)
|
||||
|
||||
|
||||
|
||||
+3
-3
@@ -19,7 +19,7 @@ from .fetchers import FetcherMethodType, create_fetchers
|
||||
from .fetchers.base import BaseFetcher
|
||||
from .cache import CacheEngine
|
||||
from .lrc import normalize_tags, normalize_unsynced, detect_sync_status
|
||||
from .config import TTL_SYNCED, TTL_UNSYNCED, TTL_NOT_FOUND, TTL_NETWORK_ERROR
|
||||
from .config import DB_PATH, TTL_SYNCED, TTL_UNSYNCED, TTL_NOT_FOUND, TTL_NETWORK_ERROR
|
||||
from .models import TrackMeta, LyricResult, CacheStatus
|
||||
from .enrichers import enrich_track
|
||||
|
||||
@@ -36,8 +36,8 @@ _STATUS_TTL: dict[CacheStatus, Optional[int]] = {
|
||||
class LrcManager:
|
||||
"""Main entry point for fetching lyrics with caching."""
|
||||
|
||||
def __init__(self) -> None:
|
||||
self.cache = CacheEngine()
|
||||
def __init__(self, db_path: Optional[str] = None) -> None:
|
||||
self.cache = CacheEngine(db_path=db_path if db_path else DB_PATH)
|
||||
self.fetchers = create_fetchers(self.cache)
|
||||
|
||||
def _build_sequence(
|
||||
|
||||
Reference in New Issue
Block a user