Walk and detect CMYK Jpeg, and print it's path name

detect_cmyk_jpeg.py

#!/usr/bin/env python
"""
Detect CMYK Jpeg under current directory
use PIL (or Pillow)
"""
import os
from PIL import Image

START_PATH='.'

for (path, dirs, files) in os.walk(START_PATH):
    for f in files:
        _, ext = os.path.splitext(f)
        if ext == '.jpg' or ext == '.jpeg':
            p = os.path.join(path, f)
            try:
                image = Image.open(p)
                if image.mode == 'CMYK':
                    print('CMYK', p)
            except IOError as e:
                print(e, p)