maskime / media-server Goto Github PK

import os

def get_extension(filename):
    file, ext = os.path.splitext(filename)
    return ext

def is_extwithin(filename, types):
    return get_extension(filename) in types

def get_marker(filename):
    return get_extension(filename).replace('.', '')

files = ['prout.done', 'prout.failed', 'prout.in_progress', 'prout.mp4']

markers_ext = ['.done', '.failed', '.in_progress']

markers = {get_marker(f): f for f in files if is_extwithin(f, markers_ext)}

print os.path.abspath(__file__)