swingmusic/server/app/helpers.py

from genericpath import exists
import os
import json
import requests
import urllib

from mutagen.mp3 import MP3
from mutagen.id3 import ID3
from mutagen.flac import FLAC

from bson import json_util

from io import BytesIO
from PIL import Image

from app.models import AllSongs
from app.configs import default_configs

all_songs_instance = AllSongs()
music_dir = os.environ.get("music_dir")
music_dirs = os.environ.get("music_dirs")

home_dir = os.path.expanduser('~') + "/"
app_dir = home_dir + '/.musicx'

PORT = os.environ.get("PORT")


def run_fast_scandir(dir, ext):
    subfolders = []
    files = []

    for f in os.scandir(dir):
        if f.is_dir() and not f.name.startswith('.'):
            subfolders.append(f.path)
        if f.is_file():
            if os.path.splitext(f.name)[1].lower() in ext:
                files.append(f.path)

    for dir in list(subfolders):
        sf, f = run_fast_scandir(dir, ext)
        subfolders.extend(sf)
        files.extend(f)

    return subfolders, files


def extract_thumb(path):
    webp_path = path.split('/')[-1] + '.webp'
    img_path = app_dir + "/images/thumbnails/" + webp_path

    if os.path.exists(img_path):
        return webp_path

    if path.endswith('.flac'):
        try:
            audio = FLAC(path)
            album_art = audio.pictures[0].data
        except:
            album_art = None
    elif path.endswith('.mp3'):
        try:
            audio = ID3(path)
            album_art = audio.getall('APIC')[0].data
        except:
            album_art = None

    if album_art is None:
        return "null.webp"
    else:
        img = Image.open(BytesIO(album_art))

        try:
            small_img = img.resize((150, 150), Image.ANTIALIAS)
            small_img.save(img_path, format="webp")
        except OSError:
            try:
                png = img.convert('RGB')
                small_img = png.resize((150, 150), Image.ANTIALIAS)
                small_img.save(img_path, format="webp")
                print('{} :: was png'.format(
                    img_path
                ))

            except:
                img_path = None

    return webp_path


def getTags(full_path):
    if full_path.endswith('.flac'):
        try:
            audio = FLAC(full_path)
        except:
            return
    elif full_path.endswith('.mp3'):
        try:
            audio = MP3(full_path)
        except:
            return

    try:
        artists = audio['artist'][0]
    except KeyError:
        try:
            artists = audio['TPE1'][0]
        except:
            artists = 'Unknown'
    except IndexError:
        artists = 'Unknown'

    try:
        album_artist = audio['albumartist'][0]
    except KeyError:
        try:
            album_artist = audio['TPE2'][0]
        except:
            album_artist = 'Unknown'
    except IndexError:
        album_artist = 'Unknown'

    try:
        title = audio['title'][0]
    except KeyError:
        try:
            title = audio['TIT2'][0]
        except:
            title = full_path.split('/')[-1]
    except:
        title = full_path.split('/')[-1]

    try:
        album = audio['album'][0]
    except KeyError:
        try:
            album = audio['TALB'][0]
        except:
            album = "Unknown"
    except IndexError:
        album = "Unknown"

    try:
        genre = audio['genre'][0]
    except KeyError:
        try:
            genre = audio['TCON'][0]
        except:
            genre = "Unknown"
    except IndexError:
        genre = "Unknown"

    img_path = extract_thumb(full_path)

    tags = {
        "filepath": full_path.replace(home_dir, ''),
        "folder": os.path.dirname(full_path).replace(home_dir, ""),
        "title": title,
        "artists": artists,
        "album_artist": album_artist,
        "album": album,
        "genre": genre,
        "length": round(audio.info.length),
        "bitrate": audio.info.bitrate,
        "image": img_path,
        "type": {
            "name": None,
            "id": None
        }
    }

    all_songs_instance.insert_song(tags)
    return tags


def convert_one_to_json(song):
    json_song = json.dumps(song, default=json_util.default)
    loaded_song = json.loads(json_song)

    return loaded_song


def convert_to_json(array):
    songs = []

    for song in array:
        json_song = json.dumps(song, default=json_util.default)
        loaded_song = json.loads(json_song)

        songs.append(loaded_song)

    return songs


def get_folders():
    folders = []

    for dir in default_configs['dirs']:
        entry = os.scandir(dir)
        folders.append(entry)


def remove_duplicates(array):
    song_num = 0

    while song_num < len(array) -1:
        for index, song in enumerate(array):
            try:

                if array[song_num]["title"] == song["title"] and array[song_num]["album"] == song["album"] and array[song_num]["artists"] == song["artists"] and index != song_num:
                    array.remove(song)
            except:
                print('whe')    
        song_num += 1

    return array


def save_image(url, path):
    response = requests.get(url)
    img = Image.open(BytesIO(response.content))
    img.save(path, 'JPEG')


def isValidFile(filename):
    if filename.endswith('.flac') or filename.endswith('.mp3'):
        return True
    else:
        return False


def isValidAudioFrom(folder):
    folder_content = os.scandir(folder)
    files = []

    for entry in folder_content:
        if isValidFile(entry.name) == True:
            file = {
                "path": entry.path,
                "name": entry.name
            }

            files.append(file)

    return files


def getFolderContents(filepath, folder):

    folder_name = urllib.parse.unquote(folder)

    path = filepath
    name = filepath.split('/')[-1]
    tags = {}

    if name.endswith('.flac'):
        image_path = folder_name + '/.thumbnails/' + \
            name.replace('.flac', '.jpg')
        audio = FLAC(path)

    if name.endswith('.mp3'):
        image_path = folder_name + '/.thumbnails/' + \
            name.replace('.mp3', '.jpg')
        audio = MP3(path)

    abslt_path = urllib.parse.quote(path.replace(music_dir, ''))

    if os.path.exists(image_path):
        img_url = 'http://localhost:{}/{}'.format(
            PORT,
            urllib.parse.quote(image_path.replace(music_dir, ''))
        )

    try:
        audio_url = 'http://localhost:{}/{}'.format(
            PORT, abslt_path
        )
        tags = getTags(audio_url, audio, img_url, folder_name)
    except:
        pass

    return tags


def create_config_dir():
    home_dir = os.path.expanduser('~')
    config_folder = home_dir + app_dir

    dirs = ["", "/images", "/images/artists", "/images/thumbnails"]

    for dir in dirs:
        if not os.path.exists(config_folder + dir):
            os.makedirs(config_folder + dir)