blob: e94b2880f58c5db87eb851fe4b363349f48c9884 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
import os
def collect_files(path):
'''Given a path to a file, yield that path. Given a path to a directory,
yield the path of all files within that directory recursively, omitting any
that begin with a period (.) character.'''
if os.path.isfile(path):
yield path
return
if not os.path.isdir(path):
raise ValueError('Not found: "%s"' % path)
for root, dirs, file_names in os.walk(path):
for file_name in file_names:
if file_name.startswith('.'):
continue
yield os.path.join(root, file_name)
|
