os

Check file extension:

a = 'a.jpg'
assert a.endswith('jpg') is True

# Note we can also pass a tuple to endswith
assert a.endswith(('jpg', 'jpeg')) is True

Split extension:

os.path.splitext('a.txt.exe') == ('a.txt', '.exe')