mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-06 03:05:35 +00:00
101 lines
3.0 KiB
Python
101 lines
3.0 KiB
Python
from concurrent.futures import ThreadPoolExecutor
|
|
from tqdm import tqdm
|
|
|
|
from app import settings
|
|
from app.db.sqlite.tracks import SQLiteTrackMethods
|
|
from app.db.store import Store
|
|
|
|
from app.lib.taglib import extract_thumb, get_tags
|
|
from app.logger import log
|
|
from app.models import Album, Artist, Track
|
|
from app.utils import run_fast_scandir
|
|
|
|
get_all_tracks = SQLiteTrackMethods.get_all_tracks
|
|
insert_many_tracks = SQLiteTrackMethods.insert_many_tracks
|
|
|
|
|
|
class Populate:
|
|
"""
|
|
Populates the database with all songs in the music directory
|
|
|
|
checks if the song is in the database, if not, it adds it
|
|
also checks if the album art exists in the image path, if not tries to extract it.
|
|
"""
|
|
|
|
def __init__(self) -> None:
|
|
|
|
tracks = get_all_tracks()
|
|
tracks = list(tracks)
|
|
|
|
files = run_fast_scandir(settings.HOME_DIR, full=True)[1]
|
|
|
|
untagged = self.filter_untagged(tracks, files)
|
|
|
|
if len(untagged) == 0:
|
|
log.info("All clear, no unread files.")
|
|
return
|
|
|
|
self.tag_untagged(untagged)
|
|
|
|
@staticmethod
|
|
def filter_untagged(tracks: list[Track], files: list[str]):
|
|
tagged_files = [t.filepath for t in tracks]
|
|
return set(files) - set(tagged_files)
|
|
|
|
@staticmethod
|
|
def tag_untagged(untagged: set[str]):
|
|
log.info("Found %s new tracks", len(untagged))
|
|
tagged_tracks: list[dict] = []
|
|
tagged_count = 0
|
|
|
|
for file in tqdm(untagged, desc="Reading files"):
|
|
tags = get_tags(file)
|
|
|
|
if tags is not None:
|
|
tagged_tracks.append(tags)
|
|
track = Track(**tags)
|
|
|
|
Store.add_track(track)
|
|
Store.add_folder(track.folder)
|
|
|
|
if not Store.album_exists(track.albumhash):
|
|
Store.add_album(Store.create_album(track))
|
|
|
|
for artist in track.artist:
|
|
if not Store.artist_exists(artist.artisthash):
|
|
Store.add_artist(Artist(artist.name))
|
|
|
|
for artist in track.albumartist:
|
|
if not Store.artist_exists(artist.artisthash):
|
|
Store.add_artist(Artist(artist.name))
|
|
|
|
tagged_count += 1
|
|
else:
|
|
log.warning("Could not read file: %s", file)
|
|
|
|
if len(tagged_tracks) > 0:
|
|
insert_many_tracks(tagged_tracks)
|
|
|
|
log.info("Added %s/%s tracks", tagged_count, len(untagged))
|
|
|
|
|
|
def get_image(album: Album):
|
|
for track in Store.tracks:
|
|
if track.albumhash == album.albumhash:
|
|
extract_thumb(track.filepath, track.image)
|
|
break
|
|
|
|
|
|
class ProcessTrackThumbnails:
|
|
def __init__(self) -> None:
|
|
with ThreadPoolExecutor(max_workers=4) as pool:
|
|
results = list(
|
|
tqdm(
|
|
pool.map(get_image, Store.albums),
|
|
total=len(Store.albums),
|
|
desc="Extracting track images",
|
|
)
|
|
)
|
|
|
|
results = [r for r in results]
|