djsundog/ffplayout
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update config and file extensions

Browse Source
This commit is contained in:
Jonathan Baecker 2020-04-16 18:25:51 +02:00
parent 3d994ae121
commit 07cae246a0
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
ffplayout/api/utils.py
View File

@ -15,7 +15,7 @@ from natsort import natsorted
def read_yaml(): def read_yaml():
config = GuiSettings.objects.filter(id=1).values()[0] config = GuiSettings.objects.filter(id=1).values()[0]
if os.path.isfile(config['playout_config']): if config and os.path.isfile(config['playout_config']):
with open(config['playout_config'], 'r') as config_file: with open(config['playout_config'], 'r') as config_file:
return yaml.safe_load(config_file) return yaml.safe_load(config_file)
@ -142,8 +142,11 @@ def set_root(path):
return path.replace(dir, '').strip('/') return path.replace(dir, '').strip('/')
def get_media_path(dir=None): def get_media_path(extensions, dir=None):
config = read_yaml() config = read_yaml()
extensions = extensions.split(' ')
playout_extensions = config['storage']['extensions']
gui_extensions = [x for x in extensions if x not in playout_extensions]
media_dir = config['storage']['path'].replace('\\', '/').rstrip('/') media_dir = config['storage']['path'].replace('\\', '/').rstrip('/')
if not dir: if not dir:
if not os.path.isdir(media_dir): if not os.path.isdir(media_dir):
@ -159,7 +162,8 @@ def get_media_path(dir=None):
media_files = [] media_files = []
for file in files: for file in files:
if os.path.splitext(file)[1] in config['storage']['extensions']: ext = os.path.splitext(file)[1]
if ext in playout_extensions:
media_info = MediaInfo.parse(os.path.join(root, file)) media_info = MediaInfo.parse(os.path.join(root, file))
duration = 0 duration = 0
for track in media_info.tracks: for track in media_info.tracks:
@ -171,6 +175,8 @@ def get_media_path(dir=None):
except KeyError: except KeyError:
pass pass
media_files.append({'file': file, 'duration': duration}) media_files.append({'file': file, 'duration': duration})
elif ext in gui_extensions:
media_files.append({'file': file, 'duration': ''})
dirs = natsorted(dirs) dirs = natsorted(dirs)