139 lines
3.6 KiB
Python
139 lines
3.6 KiB
Python
import logging
|
|
import os
|
|
import typer
|
|
|
|
from pathlib import Path
|
|
from typing import Optional, List
|
|
from dotenv import load_dotenv
|
|
from slugify import slugify
|
|
from pprint import pprint
|
|
|
|
from groove import webserver
|
|
from groove.playlist import Playlist
|
|
from groove.db.manager import database_manager
|
|
from groove.db.scanner import media_scanner
|
|
|
|
|
|
playlist_app = typer.Typer()
|
|
app = typer.Typer()
|
|
app.add_typer(playlist_app, name='playlist', help='Manage playlists.')
|
|
|
|
|
|
def initialize():
|
|
load_dotenv()
|
|
debug = os.getenv('DEBUG', None)
|
|
logging.basicConfig(format='%(asctime)s - %(message)s',
|
|
level=logging.DEBUG if debug else logging.INFO)
|
|
|
|
|
|
@playlist_app.command()
|
|
def delete(
|
|
name: str = typer.Argument(
|
|
...,
|
|
help="The name of the playlist to create."
|
|
),
|
|
no_dry_run: bool = typer.Option(
|
|
False,
|
|
help="If True, actually delete the playlist, Otherwise, show what would be deleted."
|
|
)
|
|
):
|
|
"""
|
|
Delete a playlist
|
|
"""
|
|
initialize()
|
|
with database_manager() as manager:
|
|
pl = Playlist(slug=slugify(name), connection=manager.session, create_if_not_exists=False)
|
|
if not pl.exists:
|
|
logging.info(f"No playlist named '{name}' could be found.")
|
|
return
|
|
|
|
if no_dry_run is False:
|
|
print(f"Would delete playlist {pl.record.id}, which contains {len(pl.entries)} tracks.")
|
|
return
|
|
deleted_playlist = pl.delete()
|
|
print(f"Playlist {deleted_playlist} deleted.")
|
|
|
|
|
|
@playlist_app.command()
|
|
def add(
|
|
name: str = typer.Argument(
|
|
...,
|
|
help="The name of the playlist to create."
|
|
),
|
|
tracks: List[str] = typer.Option(
|
|
None,
|
|
help="A list of tracks to add to the playlist."
|
|
),
|
|
exists_ok: bool = typer.Option(
|
|
True,
|
|
help="If True, it is okay if the playlist already exists."
|
|
),
|
|
multiples_ok: bool = typer.Option(
|
|
False,
|
|
help="If True, the same track can be added to the playlist multiple times."
|
|
)
|
|
):
|
|
"""
|
|
Create a new playlist with the specified name, unless it already exists.
|
|
"""
|
|
initialize()
|
|
with database_manager() as manager:
|
|
pl = Playlist(slug=slugify(name), connection=manager.session, create_if_not_exists=True)
|
|
if pl.exists:
|
|
if not exists_ok:
|
|
raise RuntimeError(f"Playlist with slug {pl.slug} already exists!")
|
|
logging.debug(pl.as_dict)
|
|
if tracks:
|
|
pl.add(tracks)
|
|
pprint(pl.as_dict)
|
|
|
|
|
|
@app.command()
|
|
def scan(
|
|
root: Optional[Path] = typer.Option(
|
|
None,
|
|
help="The path to the root of your media."
|
|
),
|
|
debug: bool = typer.Option(
|
|
False,
|
|
help='Enable debugging output'
|
|
),
|
|
):
|
|
"""
|
|
Scan the filesystem and create track entries in the database.
|
|
"""
|
|
initialize()
|
|
with database_manager() as manager:
|
|
scanner = media_scanner(root=root, db=manager.session)
|
|
count = scanner.scan()
|
|
logging.info(f"Imported {count} new tracks from {root}.")
|
|
|
|
|
|
@app.command()
|
|
def server(
|
|
host: str = typer.Argument(
|
|
'0.0.0.0',
|
|
help="bind address",
|
|
),
|
|
port: int = typer.Argument(
|
|
2323,
|
|
help="bind port",
|
|
),
|
|
debug: bool = typer.Option(
|
|
False,
|
|
help='Enable debugging output'
|
|
),
|
|
):
|
|
"""
|
|
Start the Groove on Demand playlsit server.
|
|
"""
|
|
initialize()
|
|
print("Starting Groove On Demand...")
|
|
with database_manager() as manager:
|
|
manager.import_from_filesystem()
|
|
webserver.start(host=host, port=port, debug=debug)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app()
|