Restyle Move populate to new file (#48)

This commit is contained in:
restyled-io[bot]
2022-05-04 01:42:26 +03:00
committed by GitHub
parent 4c09350b41
commit 559b36bd7b
7 changed files with 34 additions and 35 deletions
+5 -6
View File
@@ -10,12 +10,11 @@ from app import api
from app import functions
from app import instances
from app import models
from app import settings
from app.lib import taglib
from app.lib import trackslib
from progress.bar import Bar
from app.lib import taglib
from app import settings
def get_all_albums() -> List[models.Album]:
"""
@@ -66,7 +65,8 @@ def find_album(albumtitle: str, artist: str) -> int or None:
iter += 1
mid = (left + right) // 2
if api.ALBUMS[mid].title == albumtitle and api.ALBUMS[mid].artist == artist:
if api.ALBUMS[mid].title == albumtitle and api.ALBUMS[
mid].artist == artist:
return mid
if api.ALBUMS[mid].title < albumtitle:
@@ -155,8 +155,7 @@ def create_album(track) -> models.Album:
album["date"] = album_tracks[0]["date"]
album["artistimage"] = urllib.parse.quote_plus(
album_tracks[0]["albumartist"] + ".webp"
)
album_tracks[0]["albumartist"] + ".webp")
album["image"] = get_album_image(album_tracks)
+16 -16
View File
@@ -1,15 +1,18 @@
from dataclasses import asdict
from app.helpers import run_fast_scandir
from app import settings
from app.instances import tracks_instance, album_instance
from progress.bar import Bar
from app.logger import Log
from app.lib.taglib import get_tags
from os import path
from app.lib.albumslib import find_album, create_album
from app import api
from app.models import Track
from app import settings
from app.helpers import run_fast_scandir
from app.instances import album_instance
from app.instances import tracks_instance
from app.lib import folderslib
from app.lib.albumslib import create_album
from app.lib.albumslib import find_album
from app.lib.taglib import get_tags
from app.logger import Log
from app.models import Track
from progress.bar import Bar
class Populate:
@@ -105,8 +108,7 @@ class Populate:
if index is None:
try:
track = [
track
for track in self.tagged_tracks
track for track in self.tagged_tracks
if track["album"] == album["title"]
and track["albumartist"] == album["artist"]
][0]
@@ -126,9 +128,8 @@ class Populate:
bar.next()
bar.finish()
Log(
f"{exist_count} of {len(self.pre_albums)} albums were already in the database"
)
Log(f"{exist_count} of {len(self.pre_albums)} albums were already in the database"
)
def create_tracks(self):
"""
@@ -150,9 +151,8 @@ class Populate:
bar.next()
bar.finish()
Log(
f"Added {len(self.tagged_tracks) - failed_count} of {len(self.tagged_tracks)} new tracks and {len(self.albums)} new albums"
)
Log(f"Added {len(self.tagged_tracks) - failed_count} of {len(self.tagged_tracks)} new tracks and {len(self.albums)} new albums"
)
def create_folders(self):
"""