获取当前目录下所有的图片文件名，并且写在list.txt中

**#!/usr/bin/python
# -*- coding:utf-8 -*-
#__author__='Kakarotto'
#获取当前目录下所有的图片文件名，并且写在list.txt中

import os

#read filename from current path
def ListFilesToTxt(dir, file, wildcard, recursion):
    exts = wildcard.split(" ")
    files = os.listdir(dir)
    for name in files:
        fullname = os.path.join(dir, name)
        if (os.path.isdir(fullname) & recursion):
            ListFilesToTxt(fullname, file, wildcard, recursion)
        else:
            for ext in exts:
                if (name.endswith(ext)):
                    file.write(name + "\n")
                    break


def Test():
    dir = "."
    outfile = "list.txt"
    #wildcard = ".txt .exe .dll .py"
    wildcard = ".jpg"
    file = open(outfile, "w")
    if not file:
        print ("cannot open the file %s for writing" % outfile)

    ListFilesToTxt(dir, file, wildcard, 1)

    file.close()
Test()**