mirror of
https://github.com/tcsenpai/swingmusic.git
synced 2025-06-08 12:15:39 +00:00
85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
"""
|
|
This module contains larger functions for the server
|
|
"""
|
|
|
|
import time
|
|
from progress.bar import Bar
|
|
import requests
|
|
import os
|
|
from mutagen.flac import MutagenError
|
|
from app import helpers
|
|
from app import instances
|
|
from app import api
|
|
|
|
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.
|
|
'''
|
|
print('\nchecking for new tracks')
|
|
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
|
|
api.all_the_f_music = helpers.getAllSongs()
|
|
print('\ncheck done')
|
|
|
|
|
|
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.replace('/', '::') + '.jpg'
|
|
|
|
if not os.path.exists(file_path):
|
|
url = 'https://api.deezer.com/search/artist?q={}'.format(artist)
|
|
|
|
try:
|
|
response = requests.get(url)
|
|
except:
|
|
print('\n sleeping for 5 seconds')
|
|
time.sleep(5)
|
|
response = requests.get(url)
|
|
|
|
data = response.json()
|
|
|
|
try:
|
|
img_data = data['data'][0]['picture_xl']
|
|
except:
|
|
img_data = None
|
|
|
|
if img_data is not None:
|
|
helpers.save_image(img_data, file_path)
|
|
|
|
bar.next()
|
|
|
|
bar.finish()
|