如何选择另一个文件不仅仅是.jpg的文件夹

Question

我必须选择一个包含另一个不仅是.jpg的文件的文件夹。 我怎样才能改变它？

import os

def titleDirectory(dir):
  for file in os.listdir(dir):
    print "Processing:",dir+"//"+file
    picture = makePicture(dir+"//"+file)
    addText(picture,10,10,"This is from My CS Class")
    writePictureTo(picture,dir+"//"+"titled-"+file)

Image of code