/ Published in: Python
Expand |
Embed | Plain Text
Copy this code and paste it in your HTML
# http://www.diveintopython.org/file_handling/os_module.html # Use a one-line list comprehension to get all the files in a given directory with a given extension. import os dir = '.' ext = '.txt' txt_files = [f for f in os.listdir(dir) if os.path.isfile(os.path.join(dir,f)) and f.endswith(ext)] # os.path.join joins a directory and a filename into a path. You can also split a path name into directory and file with # os.path.split(), and you can split a filename with extension into filename and extension with os.path.splitext() # os.path.expandUser() will expand '~' to the absolute path of the current user's home directory on Windows, Linux or Mac # The rest of the os.path module: # http://docs.python.org/lib/module-os.path.html