78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
import asyncio
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
import music_tag
|
|
|
|
from pathlib import Path
|
|
from typing import Callable, Union, Iterable
|
|
from sqlalchemy import func
|
|
|
|
import groove.db
|
|
|
|
|
|
class MediaScanner:
|
|
"""
|
|
Scan a directory structure containing audio files and import them into the database.
|
|
"""
|
|
def __init__(self, root: Path, db: Callable, glob: Union[str, None] = None) -> None:
|
|
self._db = db
|
|
self._glob = tuple((glob or os.environ.get('MEDIA_GLOB')).split(','))
|
|
try:
|
|
self._root = root or Path(os.environ.get('MEDIA_ROOT'))
|
|
except TypeError:
|
|
logging.error("Could not find media root. Do you need to define MEDIA_ROOT in your environment?")
|
|
sys.exit(1)
|
|
logging.debug(f"Configured media scanner for root: {self._root}")
|
|
|
|
@property
|
|
def db(self) -> Callable:
|
|
return self._db
|
|
|
|
@property
|
|
def root(self) -> Path:
|
|
return self._root
|
|
|
|
@property
|
|
def glob(self) -> tuple:
|
|
return self._glob
|
|
|
|
def find_sources(self, pattern):
|
|
return self.root.rglob(pattern) # pragma: no cover
|
|
|
|
def import_tracks(self, sources: Iterable) -> None:
|
|
async def _do_import():
|
|
logging.debug("Scanning filesystem (this may take a minute)...")
|
|
for path in sources:
|
|
asyncio.create_task(self._import_one_track(path))
|
|
asyncio.run(_do_import())
|
|
self.db.commit()
|
|
|
|
async def _import_one_track(self, path):
|
|
tags = music_tag.load_file(path)
|
|
relpath = str(path.relative_to(self.root))
|
|
stmt = groove.db.track.insert({
|
|
'relpath': relpath,
|
|
'artist': str(tags.resolve('album_artist')),
|
|
'title': str(tags['title']),
|
|
}).prefix_with('OR IGNORE')
|
|
logging.debug(f"{tags['artist']} - {tags['title']}")
|
|
self.db.execute(stmt)
|
|
|
|
def scan(self) -> int:
|
|
"""
|
|
Walk the media root and insert Track table entries for each media file
|
|
found. Existing entries will be ignored.
|
|
"""
|
|
count = self.db.query(func.count(groove.db.track.c.relpath)).scalar()
|
|
logging.debug(f"Track table currently contains {count} entries.")
|
|
for pattern in self.glob:
|
|
self.import_tracks(self.find_sources(pattern))
|
|
newcount = self.db.query(func.count(groove.db.track.c.relpath)).scalar() - count
|
|
logging.debug(f"Inserted {newcount} new tracks so far this run...")
|
|
return newcount
|
|
|
|
|
|
media_scanner = MediaScanner
|