我需要将扩展名传递给函数,然后让该函数以小写和大写形式提取所有具有该扩展名的文件。
例如,如果我传递了mov,我需要函数来做:
videos = [file for file in glob.glob(os.path.join(dir, '*.[mM][oO][vV]'))]
如果输入小写字母,我将如何完成上面的小写 + 大写组合?
请您参考如下方法:
是这样的吗?
>>> def foo(extension):
... return '*.' + ''.join('[%s%s]' % (e.lower(), e.upper()) for e in extension)
...
>>> foo('mov')
'*.[mM][oO][vV]'
