fork download
  1. import fnmatch
  2. import os
  3.  
  4. def filterfiles(dirs, pattern):
  5. for dir in [dirs] if isinstance(dirs, str) else dirs:
  6. for _, _, files in os.walk(dir):
  7. for fname in fnmatch.filter(files, pattern):
  8. yield fname
  9.  
  10. print([fname for fname in filterfiles(".", "*")])
Success #stdin #stdout 0.1s 10096KB
stdin
Standard input is empty
stdout
['prog']