Permalink
Jan 1, 1991
Jan 1, 1991
Dec 18, 2021
Dec 18, 2021
Dec 18, 2021
Dec 18, 2021
Feb 23, 2013
Sep 11, 2014
Jun 18, 2020
Dec 18, 2021
Jun 18, 2020
Dec 18, 2021
Aug 12, 2014
Dec 18, 2021
Dec 18, 2021
Dec 16, 2012
Dec 18, 2021
Dec 18, 2021
Jan 1, 1991
Dec 18, 2021
Jan 1, 1991
Jun 18, 2020
Sep 6, 2016
Sep 11, 2014
Sep 6, 2016
Jun 18, 2020
Jun 23, 2021
Jun 18, 2020
Jan 1, 1991
Oct 2, 2008
Sep 11, 2014
Nov 18, 2013
Newer
100644
251 lines (219 sloc)
8.48 KB
2
10
13
def glob(pathname, *, root_dir=None, dir_fd=None, recursive=False,
14
include_hidden=False):
22
If `include_hidden` is true, the patterns '*', '?', '**' will match hidden
23
directories.
24
25
If `recursive` is true, the pattern '**' will match any files and
28
return list(iglob(pathname, root_dir=root_dir, dir_fd=dir_fd, recursive=recursive,
29
include_hidden=include_hidden))
31
def iglob(pathname, *, root_dir=None, dir_fd=None, recursive=False,
32
include_hidden=False):
35
The pattern may contain simple shell-style wildcards a la
36
fnmatch. However, unlike fnmatch, filenames starting with a
37
dot are special cases that are not matched by '*' and '?'
38
patterns.
40
If recursive is true, the pattern '**' will match any files and
41
zero or more directories and subdirectories.
45
if root_dir is not None:
46
root_dir = os.fspath(root_dir)
47
else:
48
root_dir = pathname[:0]
49
it = _iglob(pathname, root_dir, dir_fd, recursive, False,
50
include_hidden=include_hidden)
51
if not pathname or recursive and _isrecursive(pathname[:2]):
52
try:
53
s = next(it) # skip empty string
54
if s:
55
it = itertools.chain((s,), it)
56
except StopIteration:
57
pass
60
def _iglob(pathname, root_dir, dir_fd, recursive, dironly,
61
include_hidden=False):
67
yield pathname
68
else:
69
# Patterns ending with a slash should match only directories
75
yield from _glob2(root_dir, basename, dir_fd, dironly,
76
include_hidden=include_hidden)
78
yield from _glob1(root_dir, basename, dir_fd, dironly,
79
include_hidden=include_hidden)
81
# `os.path.split()` returns the argument itself as a dirname if it is a
82
# drive or UNC path. Prevent an infinite recursion if a drive or UNC path
83
# contains magic characters (i.e. r'\\?\C:').
84
if dirname != pathname and has_magic(dirname):
85
dirs = _iglob(dirname, root_dir, dir_fd, recursive, True,
86
include_hidden=include_hidden)
97
for name in glob_in_dir(_join(root_dir, dirname), basename, dir_fd, dironly,
98
include_hidden=include_hidden):
104
107
if include_hidden or not _ishidden(pattern):
108
names = (x for x in names if include_hidden or not _ishidden(x))
110
116
# `os.path.split()` returns an empty basename for paths ending with a
117
# directory separator. 'q*x/' should match only directories.
118
if _isdir(dirname, dir_fd):
122
# Following functions are not public but can be used by third-party code.
123
124
def glob0(dirname, pattern):
130
# This helper function recursively yields relative pathnames inside a literal
131
# directory.
132
139
# If dironly is false, yields all file names inside a directory.
140
# If dironly is true, yields only directory names.
143
fd = None
144
fsencode = None
145
if dir_fd is not None:
146
if dirname:
147
fd = arg = os.open(dirname, _dir_open_flags, dir_fd=dir_fd)
148
else:
149
arg = dir_fd
150
if isinstance(dirname, bytes):
151
fsencode = os.fsencode
152
elif dirname:
153
arg = dirname
154
elif isinstance(dirname, bytes):
155
arg = bytes(os.curdir, 'ASCII')
156
else:
157
arg = os.curdir
158
try:
159
with os.scandir(arg) as it:
160
for entry in it:
161
try:
162
if not dironly or entry.is_dir():
163
if fsencode is not None:
164
yield fsencode(entry.name)
165
else:
166
yield entry.name
167
except OSError:
168
pass
169
finally:
170
if fd is not None:
171
os.close(fd)
175
def _listdir(dirname, dir_fd, dironly):
176
with contextlib.closing(_iterdir(dirname, dir_fd, dironly)) as it:
177
return list(it)
178
191
def _lexists(pathname, dir_fd):
192
# Same as os.path.lexists(), but with dir_fd
193
if dir_fd is None:
194
return os.path.lexists(pathname)
195
try:
196
os.lstat(pathname, dir_fd=dir_fd)
197
except (OSError, ValueError):
198
return False
199
else:
200
return True
201
202
def _isdir(pathname, dir_fd):
203
# Same as os.path.isdir(), but with dir_fd
204
if dir_fd is None:
205
return os.path.isdir(pathname)
206
try:
207
st = os.stat(pathname, dir_fd=dir_fd)
208
except (OSError, ValueError):
209
return False
210
else:
211
return stat.S_ISDIR(st.st_mode)
212
213
def _join(dirname, basename):
214
# It is common if dirname or basename is empty
215
if not dirname or not basename:
216
return dirname or basename
217
return os.path.join(dirname, basename)
218
222
def has_magic(s):
223
if isinstance(s, bytes):
224
match = magic_check_bytes.search(s)
225
else:
226
match = magic_check.search(s)
227
return match is not None
232
def _isrecursive(pattern):
233
if isinstance(pattern, bytes):
234
return pattern == b'**'
235
else:
236
return pattern == '**'
237
238
def escape(pathname):
239
"""Escape all special characters.
240
"""
241
# Escaping is done by wrapping any of "*?[" between square brackets.
242
# Metacharacters do not work in the drive part and shouldn't be escaped.
243
drive, pathname = os.path.splitdrive(pathname)
244
if isinstance(pathname, bytes):
245
pathname = magic_check_bytes.sub(br'[\1]', pathname)
246
else:
247
pathname = magic_check.sub(r'[\1]', pathname)
248
return drive + pathname