meta_getter.py

"""
Will keep running in background even if main.py closes
until meta data for all files have been extracted saved
"""

import os
import sys
import json
import subprocess as sp

from url_validate import url_is_valid

ARGS = sys.argv[1:]

cur_dir = os.path.dirname(os.path.realpath(__file__))
os.chdir(cur_dir)

'''
<       (less than)
>       (greater than)
:       (colon - sometimes works, but is actually NTFS Alternate Data Streams)
"       (double quote)
/       (forward slash)
\       (backslash)
|       (vertical bar or pipe)
?       (question mark)
*       (asterisk)
'''

illegal_path_chars = list('<>:"/\\|?*')

def get_meta(supported_file_types):

    """
    media_list: a list of absolute file paths to all media
    supported_file_types: a list of supported file types (usually taken from "main.py")
    """

    os.chdir(cur_dir)
    valid_medias_list = []

    with open('data/snd_files.json', 'r', encoding='utf-8') as fp:
        media_list = json.load(fp)

    if type(media_list) == str:
        try:
            if type(eval(media_list)) == list:
                media_list = eval(media_list)
        except Exception:
            pass

    for media in media_list:
        if os.path.isfile(media):
            if media.endswith(tuple(supported_file_types)):
                valid_medias_list.append(media)
        elif url_is_valid(media):
            valid_medias_list.append(media)

    for _, media in enumerate(valid_medias_list):
        meta_info = sp.run(['ffprobe',
                            '-v',
                            'quiet',
                            '-print_format',
                            'json',
                            '-show_format',
                            '-show_streams',
                            media],
                            capture_output=True,
                            text=True).stdout

        meta_info = json.loads(meta_info)
        meta_info['format'].update({'bpm': 120})

        media = ''.join(['~' if chr in illegal_path_chars else chr for chr in media])
        with open(f'data/mediameta_{media}.json', 'w', encoding='utf-8') as fp:
            json.dump(meta_info, fp, indent=2)

    return valid_medias_list

if __name__ == "__main__":
    ARGS = sys.argv[1:]
    if len(ARGS) == 1:
        try:
            get_meta(ARGS[0])
            sys.exit(0)
        except Exception:
            sys.exit(1)
    else:
        sys.exit(1)