mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-08 12:15:39 +00:00
89 lines
2.3 KiB
Python
89 lines
2.3 KiB
Python
"""
|
|
This module contains larger functions for the server
|
|
"""
|
|
|
|
from progress.bar import Bar
|
|
import requests
|
|
import os
|
|
from mutagen.flac import MutagenError
|
|
from app import helpers
|
|
from app import instances
|
|
|
|
|
|
def populate():
|
|
'''
|
|
Populate 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.
|
|
'''
|
|
files = helpers.run_fast_scandir(helpers.home_dir, [".flac", ".mp3"])[1]
|
|
|
|
for file in files:
|
|
file_in_db_obj = instances.songs_instance.find_song_by_path(file)
|
|
|
|
try:
|
|
image = file_in_db_obj['image']
|
|
|
|
if not os.path.exists(os.path.join(helpers.app_dir, 'images', 'thumbnails', image)):
|
|
helpers.extract_thumb(file)
|
|
except:
|
|
image = None
|
|
|
|
if image is None:
|
|
try:
|
|
helpers.getTags(file)
|
|
except MutagenError:
|
|
pass
|
|
|
|
return {'msg': 'updated everything'}
|
|
|
|
|
|
def populate_images():
|
|
all_songs = instances.songs_instance.get_all_songs()
|
|
|
|
artists = []
|
|
|
|
for song in all_songs:
|
|
this_artists = song['artists'].split(', ')
|
|
|
|
for artist in this_artists:
|
|
if artist not in artists:
|
|
artists.append(artist)
|
|
|
|
bar = Bar('Processing images', max=len(artists))
|
|
for artist in artists:
|
|
file_path = helpers.app_dir + '/images/artists/' + artist + '.jpg'
|
|
|
|
if not os.path.exists(file_path):
|
|
url = 'https://api.deezer.com/search/artist?q={}'.format(artist)
|
|
response = requests.get(url)
|
|
data = response.json()
|
|
|
|
try:
|
|
image_path = data['data'][0]['picture_xl']
|
|
except:
|
|
image_path = None
|
|
|
|
if image_path is not None:
|
|
try:
|
|
helpers.save_image(image_path, file_path)
|
|
artist_obj = {
|
|
'name': artist
|
|
}
|
|
|
|
instances.artist_instance.insert_artist(artist_obj)
|
|
except:
|
|
pass
|
|
else:
|
|
pass
|
|
|
|
bar.next()
|
|
|
|
bar.finish()
|
|
|
|
artists_in_db = instances.artist_instance.get_all_artists()
|
|
|
|
return {'sample': artists_in_db[:25]}
|