Check if string ends with one of the strings from a list
Though not widely known, str.endswith also accepts a tuple. You don't need to loop.
Take an extension from the file and see if it is in the set of extensions:
import os extensions = set(['.mp3','.avi']) file_name = 'test.mp3'extension = os.path.splitext(file_name) extension in extensionsTrue
Using a set because time complexity for lookups in sets is O(1) (docs).