blob: 7d1d1150b0da849fc1db5b9a9283cb0f35a25295 [file] [log] [blame]
Antoine Pitrou31119e42013-11-22 17:38:12 +01001import fnmatch
2import functools
3import io
4import ntpath
5import os
6import posixpath
7import re
8import sys
Serhiy Storchaka81108372017-09-26 00:55:55 +03009from _collections_abc import Sequence
Jörg Stucked5c120f2019-05-21 19:44:40 +020010from errno import EINVAL, ENOENT, ENOTDIR, EBADF, ELOOP
Antoine Pitrou31119e42013-11-22 17:38:12 +010011from operator import attrgetter
12from stat import S_ISDIR, S_ISLNK, S_ISREG, S_ISSOCK, S_ISBLK, S_ISCHR, S_ISFIFO
Antoine Pitrou069a5e12013-12-03 09:41:35 +010013from urllib.parse import quote_from_bytes as urlquote_from_bytes
Antoine Pitrou31119e42013-11-22 17:38:12 +010014
15
16supports_symlinks = True
Antoine Pitroudb118f52014-11-19 00:32:08 +010017if os.name == 'nt':
Antoine Pitrou31119e42013-11-22 17:38:12 +010018 import nt
Antoine Pitrou31119e42013-11-22 17:38:12 +010019 if sys.getwindowsversion()[:2] >= (6, 0):
20 from nt import _getfinalpathname
21 else:
22 supports_symlinks = False
23 _getfinalpathname = None
Antoine Pitroudb118f52014-11-19 00:32:08 +010024else:
25 nt = None
Antoine Pitrou31119e42013-11-22 17:38:12 +010026
27
28__all__ = [
29 "PurePath", "PurePosixPath", "PureWindowsPath",
30 "Path", "PosixPath", "WindowsPath",
31 ]
32
33#
34# Internals
35#
36
penguindustin96466302019-05-06 14:57:17 -040037# EBADF - guard against macOS `stat` throwing EBADF
Jörg Stucked5c120f2019-05-21 19:44:40 +020038_IGNORED_ERROS = (ENOENT, ENOTDIR, EBADF, ELOOP)
Przemysław Spodymek216b7452018-08-27 23:33:45 +020039
Steve Dower2f6fae62019-02-03 23:08:18 -080040_IGNORED_WINERRORS = (
41 21, # ERROR_NOT_READY - drive exists but is not accessible
Jörg Stucked5c120f2019-05-21 19:44:40 +020042 1921, # ERROR_CANT_RESOLVE_FILENAME - fix for broken symlink pointing to itself
Steve Dower2f6fae62019-02-03 23:08:18 -080043)
44
45def _ignore_error(exception):
46 return (getattr(exception, 'errno', None) in _IGNORED_ERROS or
47 getattr(exception, 'winerror', None) in _IGNORED_WINERRORS)
48
49
Antoine Pitrou31119e42013-11-22 17:38:12 +010050def _is_wildcard_pattern(pat):
51 # Whether this pattern needs actual matching using fnmatch, or can
52 # be looked up directly as a file.
53 return "*" in pat or "?" in pat or "[" in pat
54
55
56class _Flavour(object):
57 """A flavour implements a particular (platform-specific) set of path
58 semantics."""
59
60 def __init__(self):
61 self.join = self.sep.join
62
63 def parse_parts(self, parts):
64 parsed = []
65 sep = self.sep
66 altsep = self.altsep
67 drv = root = ''
68 it = reversed(parts)
69 for part in it:
70 if not part:
71 continue
72 if altsep:
73 part = part.replace(altsep, sep)
74 drv, root, rel = self.splitroot(part)
75 if sep in rel:
76 for x in reversed(rel.split(sep)):
77 if x and x != '.':
78 parsed.append(sys.intern(x))
79 else:
80 if rel and rel != '.':
81 parsed.append(sys.intern(rel))
82 if drv or root:
83 if not drv:
84 # If no drive is present, try to find one in the previous
85 # parts. This makes the result of parsing e.g.
86 # ("C:", "/", "a") reasonably intuitive.
87 for part in it:
Antoine Pitrou57fffd62015-02-15 18:03:59 +010088 if not part:
89 continue
90 if altsep:
91 part = part.replace(altsep, sep)
Antoine Pitrou31119e42013-11-22 17:38:12 +010092 drv = self.splitroot(part)[0]
93 if drv:
94 break
95 break
96 if drv or root:
97 parsed.append(drv + root)
98 parsed.reverse()
99 return drv, root, parsed
100
101 def join_parsed_parts(self, drv, root, parts, drv2, root2, parts2):
102 """
103 Join the two paths represented by the respective
104 (drive, root, parts) tuples. Return a new (drive, root, parts) tuple.
105 """
106 if root2:
Serhiy Storchakaa9939022013-12-06 17:14:12 +0200107 if not drv2 and drv:
108 return drv, root2, [drv + root2] + parts2[1:]
109 elif drv2:
110 if drv2 == drv or self.casefold(drv2) == self.casefold(drv):
111 # Same drive => second path is relative to the first
112 return drv, root, parts + parts2[1:]
Antoine Pitrou31119e42013-11-22 17:38:12 +0100113 else:
Serhiy Storchakaa9939022013-12-06 17:14:12 +0200114 # Second path is non-anchored (common case)
115 return drv, root, parts + parts2
116 return drv2, root2, parts2
Antoine Pitrou31119e42013-11-22 17:38:12 +0100117
118
119class _WindowsFlavour(_Flavour):
120 # Reference for Windows paths can be found at
121 # http://msdn.microsoft.com/en-us/library/aa365247%28v=vs.85%29.aspx
122
123 sep = '\\'
124 altsep = '/'
125 has_drv = True
126 pathmod = ntpath
127
Antoine Pitroudb118f52014-11-19 00:32:08 +0100128 is_supported = (os.name == 'nt')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100129
Jon Dufresne39726282017-05-18 07:35:54 -0700130 drive_letters = set('abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ')
Antoine Pitrou31119e42013-11-22 17:38:12 +0100131 ext_namespace_prefix = '\\\\?\\'
132
133 reserved_names = (
134 {'CON', 'PRN', 'AUX', 'NUL'} |
135 {'COM%d' % i for i in range(1, 10)} |
136 {'LPT%d' % i for i in range(1, 10)}
137 )
138
139 # Interesting findings about extended paths:
140 # - '\\?\c:\a', '//?/c:\a' and '//?/c:/a' are all supported
141 # but '\\?\c:/a' is not
142 # - extended paths are always absolute; "relative" extended paths will
143 # fail.
144
145 def splitroot(self, part, sep=sep):
146 first = part[0:1]
147 second = part[1:2]
148 if (second == sep and first == sep):
149 # XXX extended paths should also disable the collapsing of "."
150 # components (according to MSDN docs).
151 prefix, part = self._split_extended_path(part)
152 first = part[0:1]
153 second = part[1:2]
154 else:
155 prefix = ''
156 third = part[2:3]
157 if (second == sep and first == sep and third != sep):
158 # is a UNC path:
159 # vvvvvvvvvvvvvvvvvvvvv root
160 # \\machine\mountpoint\directory\etc\...
161 # directory ^^^^^^^^^^^^^^
162 index = part.find(sep, 2)
163 if index != -1:
164 index2 = part.find(sep, index + 1)
165 # a UNC path can't have two slashes in a row
166 # (after the initial two)
167 if index2 != index + 1:
168 if index2 == -1:
169 index2 = len(part)
170 if prefix:
171 return prefix + part[1:index2], sep, part[index2+1:]
172 else:
173 return part[:index2], sep, part[index2+1:]
174 drv = root = ''
175 if second == ':' and first in self.drive_letters:
176 drv = part[:2]
177 part = part[2:]
178 first = third
179 if first == sep:
180 root = first
181 part = part.lstrip(sep)
182 return prefix + drv, root, part
183
184 def casefold(self, s):
185 return s.lower()
186
187 def casefold_parts(self, parts):
188 return [p.lower() for p in parts]
189
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300190 def compile_pattern(self, pattern):
191 return re.compile(fnmatch.translate(pattern), re.IGNORECASE).fullmatch
192
Steve Dower98eb3602016-11-09 12:58:17 -0800193 def resolve(self, path, strict=False):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100194 s = str(path)
195 if not s:
196 return os.getcwd()
Steve Dower98eb3602016-11-09 12:58:17 -0800197 previous_s = None
Antoine Pitrou31119e42013-11-22 17:38:12 +0100198 if _getfinalpathname is not None:
Steve Dower98eb3602016-11-09 12:58:17 -0800199 if strict:
200 return self._ext_to_normal(_getfinalpathname(s))
201 else:
Antoine Pietriadd98eb2017-06-07 17:29:17 +0200202 tail_parts = [] # End of the path after the first one not found
Steve Dower98eb3602016-11-09 12:58:17 -0800203 while True:
204 try:
205 s = self._ext_to_normal(_getfinalpathname(s))
206 except FileNotFoundError:
207 previous_s = s
Antoine Pietriadd98eb2017-06-07 17:29:17 +0200208 s, tail = os.path.split(s)
209 tail_parts.append(tail)
Steve Dower4b1e98b2016-12-28 16:02:59 -0800210 if previous_s == s:
211 return path
Steve Dower98eb3602016-11-09 12:58:17 -0800212 else:
Antoine Pietriadd98eb2017-06-07 17:29:17 +0200213 return os.path.join(s, *reversed(tail_parts))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100214 # Means fallback on absolute
215 return None
216
217 def _split_extended_path(self, s, ext_prefix=ext_namespace_prefix):
218 prefix = ''
219 if s.startswith(ext_prefix):
220 prefix = s[:4]
221 s = s[4:]
222 if s.startswith('UNC\\'):
223 prefix += s[:3]
224 s = '\\' + s[3:]
225 return prefix, s
226
227 def _ext_to_normal(self, s):
228 # Turn back an extended path into a normal DOS-like path
229 return self._split_extended_path(s)[1]
230
231 def is_reserved(self, parts):
232 # NOTE: the rules for reserved names seem somewhat complicated
233 # (e.g. r"..\NUL" is reserved but not r"foo\NUL").
234 # We err on the side of caution and return True for paths which are
235 # not considered reserved by Windows.
236 if not parts:
237 return False
238 if parts[0].startswith('\\\\'):
239 # UNC paths are never reserved
240 return False
241 return parts[-1].partition('.')[0].upper() in self.reserved_names
242
243 def make_uri(self, path):
244 # Under Windows, file URIs use the UTF-8 encoding.
245 drive = path.drive
246 if len(drive) == 2 and drive[1] == ':':
247 # It's a path on a local drive => 'file:///c:/a/b'
248 rest = path.as_posix()[2:].lstrip('/')
249 return 'file:///%s/%s' % (
250 drive, urlquote_from_bytes(rest.encode('utf-8')))
251 else:
252 # It's a path on a network drive => 'file://host/share/a/b'
253 return 'file:' + urlquote_from_bytes(path.as_posix().encode('utf-8'))
254
Antoine Pitrou8477ed62014-12-30 20:54:45 +0100255 def gethomedir(self, username):
256 if 'HOME' in os.environ:
257 userhome = os.environ['HOME']
258 elif 'USERPROFILE' in os.environ:
259 userhome = os.environ['USERPROFILE']
260 elif 'HOMEPATH' in os.environ:
Antoine Pitrou5d4e27e2014-12-30 22:09:42 +0100261 try:
262 drv = os.environ['HOMEDRIVE']
263 except KeyError:
264 drv = ''
265 userhome = drv + os.environ['HOMEPATH']
Antoine Pitrou8477ed62014-12-30 20:54:45 +0100266 else:
267 raise RuntimeError("Can't determine home directory")
268
269 if username:
270 # Try to guess user home directory. By default all users
271 # directories are located in the same place and are named by
272 # corresponding usernames. If current user home directory points
273 # to nonstandard place, this guess is likely wrong.
274 if os.environ['USERNAME'] != username:
275 drv, root, parts = self.parse_parts((userhome,))
276 if parts[-1] != os.environ['USERNAME']:
277 raise RuntimeError("Can't determine home directory "
278 "for %r" % username)
279 parts[-1] = username
280 if drv or root:
281 userhome = drv + root + self.join(parts[1:])
282 else:
283 userhome = self.join(parts)
284 return userhome
Antoine Pitrou31119e42013-11-22 17:38:12 +0100285
286class _PosixFlavour(_Flavour):
287 sep = '/'
288 altsep = ''
289 has_drv = False
290 pathmod = posixpath
291
292 is_supported = (os.name != 'nt')
293
294 def splitroot(self, part, sep=sep):
295 if part and part[0] == sep:
296 stripped_part = part.lstrip(sep)
297 # According to POSIX path resolution:
298 # http://pubs.opengroup.org/onlinepubs/009695399/basedefs/xbd_chap04.html#tag_04_11
299 # "A pathname that begins with two successive slashes may be
300 # interpreted in an implementation-defined manner, although more
301 # than two leading slashes shall be treated as a single slash".
302 if len(part) - len(stripped_part) == 2:
303 return '', sep * 2, stripped_part
304 else:
305 return '', sep, stripped_part
306 else:
307 return '', '', part
308
309 def casefold(self, s):
310 return s
311
312 def casefold_parts(self, parts):
313 return parts
314
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300315 def compile_pattern(self, pattern):
316 return re.compile(fnmatch.translate(pattern)).fullmatch
317
Steve Dower98eb3602016-11-09 12:58:17 -0800318 def resolve(self, path, strict=False):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100319 sep = self.sep
Antoine Pitrou31119e42013-11-22 17:38:12 +0100320 accessor = path._accessor
Antoine Pitrouc274fd22013-12-16 19:57:41 +0100321 seen = {}
322 def _resolve(path, rest):
323 if rest.startswith(sep):
324 path = ''
325
326 for name in rest.split(sep):
327 if not name or name == '.':
328 # current dir
329 continue
330 if name == '..':
331 # parent dir
332 path, _, _ = path.rpartition(sep)
333 continue
334 newpath = path + sep + name
335 if newpath in seen:
336 # Already seen this path
337 path = seen[newpath]
338 if path is not None:
339 # use cached value
340 continue
341 # The symlink is not resolved, so we must have a symlink loop.
342 raise RuntimeError("Symlink loop from %r" % newpath)
343 # Resolve the symbolic link
344 try:
345 target = accessor.readlink(newpath)
346 except OSError as e:
Antoine Pietriadd98eb2017-06-07 17:29:17 +0200347 if e.errno != EINVAL and strict:
348 raise
349 # Not a symlink, or non-strict mode. We just leave the path
350 # untouched.
Antoine Pitrouc274fd22013-12-16 19:57:41 +0100351 path = newpath
352 else:
353 seen[newpath] = None # not resolved symlink
354 path = _resolve(path, target)
355 seen[newpath] = path # resolved symlink
356
357 return path
358 # NOTE: according to POSIX, getcwd() cannot contain path components
359 # which are symlinks.
360 base = '' if path.is_absolute() else os.getcwd()
361 return _resolve(base, str(path)) or sep
Antoine Pitrou31119e42013-11-22 17:38:12 +0100362
363 def is_reserved(self, parts):
364 return False
365
366 def make_uri(self, path):
367 # We represent the path using the local filesystem encoding,
368 # for portability to other applications.
369 bpath = bytes(path)
370 return 'file://' + urlquote_from_bytes(bpath)
371
Antoine Pitrou8477ed62014-12-30 20:54:45 +0100372 def gethomedir(self, username):
373 if not username:
374 try:
375 return os.environ['HOME']
376 except KeyError:
377 import pwd
378 return pwd.getpwuid(os.getuid()).pw_dir
379 else:
380 import pwd
381 try:
382 return pwd.getpwnam(username).pw_dir
383 except KeyError:
384 raise RuntimeError("Can't determine home directory "
385 "for %r" % username)
386
Antoine Pitrou31119e42013-11-22 17:38:12 +0100387
388_windows_flavour = _WindowsFlavour()
389_posix_flavour = _PosixFlavour()
390
391
392class _Accessor:
393 """An accessor implements a particular (system-specific or not) way of
394 accessing paths on the filesystem."""
395
396
397class _NormalAccessor(_Accessor):
398
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200399 stat = os.stat
Antoine Pitrou31119e42013-11-22 17:38:12 +0100400
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200401 lstat = os.lstat
Antoine Pitrou31119e42013-11-22 17:38:12 +0100402
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200403 open = os.open
Antoine Pitrou31119e42013-11-22 17:38:12 +0100404
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200405 listdir = os.listdir
Antoine Pitrou31119e42013-11-22 17:38:12 +0100406
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200407 scandir = os.scandir
Antoine Pitrou31119e42013-11-22 17:38:12 +0100408
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200409 chmod = os.chmod
Antoine Pitrou31119e42013-11-22 17:38:12 +0100410
411 if hasattr(os, "lchmod"):
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200412 lchmod = os.lchmod
Antoine Pitrou31119e42013-11-22 17:38:12 +0100413 else:
414 def lchmod(self, pathobj, mode):
415 raise NotImplementedError("lchmod() not available on this system")
416
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200417 mkdir = os.mkdir
Antoine Pitrou31119e42013-11-22 17:38:12 +0100418
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200419 unlink = os.unlink
Antoine Pitrou31119e42013-11-22 17:38:12 +0100420
Toke Høiland-Jørgensen092435e2019-12-16 13:23:55 +0100421 if hasattr(os, "link"):
422 link_to = os.link
423 else:
424 @staticmethod
425 def link_to(self, target):
426 raise NotImplementedError("os.link() not available on this system")
Joannah Nanjekye6b5b0132019-05-04 11:27:10 -0400427
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200428 rmdir = os.rmdir
Antoine Pitrou31119e42013-11-22 17:38:12 +0100429
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200430 rename = os.rename
Antoine Pitrou31119e42013-11-22 17:38:12 +0100431
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200432 replace = os.replace
Antoine Pitrou31119e42013-11-22 17:38:12 +0100433
434 if nt:
435 if supports_symlinks:
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200436 symlink = os.symlink
Antoine Pitrou31119e42013-11-22 17:38:12 +0100437 else:
438 def symlink(a, b, target_is_directory):
439 raise NotImplementedError("symlink() not available on this system")
440 else:
441 # Under POSIX, os.symlink() takes two args
442 @staticmethod
443 def symlink(a, b, target_is_directory):
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200444 return os.symlink(a, b)
Antoine Pitrou31119e42013-11-22 17:38:12 +0100445
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200446 utime = os.utime
Antoine Pitrou31119e42013-11-22 17:38:12 +0100447
448 # Helper for resolve()
449 def readlink(self, path):
450 return os.readlink(path)
451
452
453_normal_accessor = _NormalAccessor()
454
455
456#
457# Globbing helpers
458#
459
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300460def _make_selector(pattern_parts, flavour):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100461 pat = pattern_parts[0]
462 child_parts = pattern_parts[1:]
463 if pat == '**':
464 cls = _RecursiveWildcardSelector
465 elif '**' in pat:
466 raise ValueError("Invalid pattern: '**' can only be an entire path component")
467 elif _is_wildcard_pattern(pat):
468 cls = _WildcardSelector
469 else:
470 cls = _PreciseSelector
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300471 return cls(pat, child_parts, flavour)
Antoine Pitrou31119e42013-11-22 17:38:12 +0100472
473if hasattr(functools, "lru_cache"):
474 _make_selector = functools.lru_cache()(_make_selector)
475
476
477class _Selector:
478 """A selector matches a specific glob pattern part against the children
479 of a given path."""
480
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300481 def __init__(self, child_parts, flavour):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100482 self.child_parts = child_parts
483 if child_parts:
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300484 self.successor = _make_selector(child_parts, flavour)
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300485 self.dironly = True
Antoine Pitrou31119e42013-11-22 17:38:12 +0100486 else:
487 self.successor = _TerminatingSelector()
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300488 self.dironly = False
Antoine Pitrou31119e42013-11-22 17:38:12 +0100489
490 def select_from(self, parent_path):
491 """Iterate over all child paths of `parent_path` matched by this
492 selector. This can contain parent_path itself."""
493 path_cls = type(parent_path)
494 is_dir = path_cls.is_dir
495 exists = path_cls.exists
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300496 scandir = parent_path._accessor.scandir
497 if not is_dir(parent_path):
498 return iter([])
499 return self._select_from(parent_path, is_dir, exists, scandir)
Antoine Pitrou31119e42013-11-22 17:38:12 +0100500
501
502class _TerminatingSelector:
503
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300504 def _select_from(self, parent_path, is_dir, exists, scandir):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100505 yield parent_path
506
507
508class _PreciseSelector(_Selector):
509
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300510 def __init__(self, name, child_parts, flavour):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100511 self.name = name
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300512 _Selector.__init__(self, child_parts, flavour)
Antoine Pitrou31119e42013-11-22 17:38:12 +0100513
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300514 def _select_from(self, parent_path, is_dir, exists, scandir):
Guido van Rossum6c2d33a2016-01-06 09:42:07 -0800515 try:
Guido van Rossum6c2d33a2016-01-06 09:42:07 -0800516 path = parent_path._make_child_relpath(self.name)
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300517 if (is_dir if self.dironly else exists)(path):
518 for p in self.successor._select_from(path, is_dir, exists, scandir):
Guido van Rossum6c2d33a2016-01-06 09:42:07 -0800519 yield p
520 except PermissionError:
Antoine Pitrou31119e42013-11-22 17:38:12 +0100521 return
Antoine Pitrou31119e42013-11-22 17:38:12 +0100522
523
524class _WildcardSelector(_Selector):
525
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300526 def __init__(self, pat, child_parts, flavour):
527 self.match = flavour.compile_pattern(pat)
528 _Selector.__init__(self, child_parts, flavour)
Antoine Pitrou31119e42013-11-22 17:38:12 +0100529
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300530 def _select_from(self, parent_path, is_dir, exists, scandir):
Guido van Rossum6c2d33a2016-01-06 09:42:07 -0800531 try:
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300532 entries = list(scandir(parent_path))
533 for entry in entries:
Jörg Stucked5c120f2019-05-21 19:44:40 +0200534 entry_is_dir = False
535 try:
536 entry_is_dir = entry.is_dir()
537 except OSError as e:
538 if not _ignore_error(e):
539 raise
540 if not self.dironly or entry_is_dir:
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300541 name = entry.name
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300542 if self.match(name):
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300543 path = parent_path._make_child_relpath(name)
544 for p in self.successor._select_from(path, is_dir, exists, scandir):
545 yield p
Guido van Rossum6c2d33a2016-01-06 09:42:07 -0800546 except PermissionError:
Antoine Pitrou31119e42013-11-22 17:38:12 +0100547 return
Guido van Rossum6c2d33a2016-01-06 09:42:07 -0800548
Antoine Pitrou31119e42013-11-22 17:38:12 +0100549
550
551class _RecursiveWildcardSelector(_Selector):
552
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +0300553 def __init__(self, pat, child_parts, flavour):
554 _Selector.__init__(self, child_parts, flavour)
Antoine Pitrou31119e42013-11-22 17:38:12 +0100555
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300556 def _iterate_directories(self, parent_path, is_dir, scandir):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100557 yield parent_path
Guido van Rossumbc9fdda2016-01-07 10:56:36 -0800558 try:
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300559 entries = list(scandir(parent_path))
560 for entry in entries:
Przemysław Spodymek216b7452018-08-27 23:33:45 +0200561 entry_is_dir = False
562 try:
563 entry_is_dir = entry.is_dir()
564 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -0800565 if not _ignore_error(e):
Przemysław Spodymek216b7452018-08-27 23:33:45 +0200566 raise
567 if entry_is_dir and not entry.is_symlink():
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300568 path = parent_path._make_child_relpath(entry.name)
569 for p in self._iterate_directories(path, is_dir, scandir):
Guido van Rossumbc9fdda2016-01-07 10:56:36 -0800570 yield p
571 except PermissionError:
572 return
Antoine Pitrou31119e42013-11-22 17:38:12 +0100573
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300574 def _select_from(self, parent_path, is_dir, exists, scandir):
Guido van Rossum6c2d33a2016-01-06 09:42:07 -0800575 try:
Serhiy Storchaka680cb152016-09-07 10:58:05 +0300576 yielded = set()
577 try:
578 successor_select = self.successor._select_from
579 for starting_point in self._iterate_directories(parent_path, is_dir, scandir):
580 for p in successor_select(starting_point, is_dir, exists, scandir):
581 if p not in yielded:
582 yield p
583 yielded.add(p)
584 finally:
585 yielded.clear()
Guido van Rossum6c2d33a2016-01-06 09:42:07 -0800586 except PermissionError:
Antoine Pitrou31119e42013-11-22 17:38:12 +0100587 return
Antoine Pitrou31119e42013-11-22 17:38:12 +0100588
589
590#
591# Public API
592#
593
594class _PathParents(Sequence):
595 """This object provides sequence-like access to the logical ancestors
596 of a path. Don't try to construct it yourself."""
597 __slots__ = ('_pathcls', '_drv', '_root', '_parts')
598
599 def __init__(self, path):
600 # We don't store the instance to avoid reference cycles
601 self._pathcls = type(path)
602 self._drv = path._drv
603 self._root = path._root
604 self._parts = path._parts
605
606 def __len__(self):
607 if self._drv or self._root:
608 return len(self._parts) - 1
609 else:
610 return len(self._parts)
611
612 def __getitem__(self, idx):
613 if idx < 0 or idx >= len(self):
614 raise IndexError(idx)
615 return self._pathcls._from_parsed_parts(self._drv, self._root,
616 self._parts[:-idx - 1])
617
618 def __repr__(self):
619 return "<{}.parents>".format(self._pathcls.__name__)
620
621
622class PurePath(object):
chasondfa015c2018-02-19 08:36:32 +0900623 """Base class for manipulating paths without I/O.
624
625 PurePath represents a filesystem path and offers operations which
Antoine Pitrou31119e42013-11-22 17:38:12 +0100626 don't imply any actual filesystem I/O. Depending on your system,
627 instantiating a PurePath will return either a PurePosixPath or a
628 PureWindowsPath object. You can also instantiate either of these classes
629 directly, regardless of your system.
630 """
631 __slots__ = (
632 '_drv', '_root', '_parts',
633 '_str', '_hash', '_pparts', '_cached_cparts',
634 )
635
636 def __new__(cls, *args):
637 """Construct a PurePath from one or several strings and or existing
638 PurePath objects. The strings and path objects are combined so as
639 to yield a canonicalized path, which is incorporated into the
640 new PurePath object.
641 """
642 if cls is PurePath:
643 cls = PureWindowsPath if os.name == 'nt' else PurePosixPath
644 return cls._from_parts(args)
645
646 def __reduce__(self):
647 # Using the parts tuple helps share interned path parts
648 # when pickling related paths.
649 return (self.__class__, tuple(self._parts))
650
651 @classmethod
652 def _parse_args(cls, args):
653 # This is useful when you don't want to create an instance, just
654 # canonicalize some constructor arguments.
655 parts = []
656 for a in args:
657 if isinstance(a, PurePath):
658 parts += a._parts
Antoine Pitrou31119e42013-11-22 17:38:12 +0100659 else:
Brett Cannon568be632016-06-10 12:20:49 -0700660 a = os.fspath(a)
661 if isinstance(a, str):
662 # Force-cast str subclasses to str (issue #21127)
663 parts.append(str(a))
664 else:
665 raise TypeError(
666 "argument should be a str object or an os.PathLike "
667 "object returning str, not %r"
668 % type(a))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100669 return cls._flavour.parse_parts(parts)
670
671 @classmethod
672 def _from_parts(cls, args, init=True):
673 # We need to call _parse_args on the instance, so as to get the
674 # right flavour.
675 self = object.__new__(cls)
676 drv, root, parts = self._parse_args(args)
677 self._drv = drv
678 self._root = root
679 self._parts = parts
680 if init:
681 self._init()
682 return self
683
684 @classmethod
685 def _from_parsed_parts(cls, drv, root, parts, init=True):
686 self = object.__new__(cls)
687 self._drv = drv
688 self._root = root
689 self._parts = parts
690 if init:
691 self._init()
692 return self
693
694 @classmethod
695 def _format_parsed_parts(cls, drv, root, parts):
696 if drv or root:
697 return drv + root + cls._flavour.join(parts[1:])
698 else:
699 return cls._flavour.join(parts)
700
701 def _init(self):
Martin Pantere26da7c2016-06-02 10:07:09 +0000702 # Overridden in concrete Path
Antoine Pitrou31119e42013-11-22 17:38:12 +0100703 pass
704
705 def _make_child(self, args):
706 drv, root, parts = self._parse_args(args)
707 drv, root, parts = self._flavour.join_parsed_parts(
708 self._drv, self._root, self._parts, drv, root, parts)
709 return self._from_parsed_parts(drv, root, parts)
710
711 def __str__(self):
712 """Return the string representation of the path, suitable for
713 passing to system calls."""
714 try:
715 return self._str
716 except AttributeError:
717 self._str = self._format_parsed_parts(self._drv, self._root,
718 self._parts) or '.'
719 return self._str
720
Brett Cannon568be632016-06-10 12:20:49 -0700721 def __fspath__(self):
722 return str(self)
723
Antoine Pitrou31119e42013-11-22 17:38:12 +0100724 def as_posix(self):
725 """Return the string representation of the path with forward (/)
726 slashes."""
727 f = self._flavour
728 return str(self).replace(f.sep, '/')
729
730 def __bytes__(self):
731 """Return the bytes representation of the path. This is only
732 recommended to use under Unix."""
Serhiy Storchaka62a99512017-03-25 13:42:11 +0200733 return os.fsencode(self)
Antoine Pitrou31119e42013-11-22 17:38:12 +0100734
735 def __repr__(self):
736 return "{}({!r})".format(self.__class__.__name__, self.as_posix())
737
738 def as_uri(self):
739 """Return the path as a 'file' URI."""
740 if not self.is_absolute():
741 raise ValueError("relative path can't be expressed as a file URI")
742 return self._flavour.make_uri(self)
743
744 @property
745 def _cparts(self):
746 # Cached casefolded parts, for hashing and comparison
747 try:
748 return self._cached_cparts
749 except AttributeError:
750 self._cached_cparts = self._flavour.casefold_parts(self._parts)
751 return self._cached_cparts
752
753 def __eq__(self, other):
754 if not isinstance(other, PurePath):
755 return NotImplemented
756 return self._cparts == other._cparts and self._flavour is other._flavour
757
Antoine Pitrou31119e42013-11-22 17:38:12 +0100758 def __hash__(self):
759 try:
760 return self._hash
761 except AttributeError:
762 self._hash = hash(tuple(self._cparts))
763 return self._hash
764
765 def __lt__(self, other):
766 if not isinstance(other, PurePath) or self._flavour is not other._flavour:
767 return NotImplemented
768 return self._cparts < other._cparts
769
770 def __le__(self, other):
771 if not isinstance(other, PurePath) or self._flavour is not other._flavour:
772 return NotImplemented
773 return self._cparts <= other._cparts
774
775 def __gt__(self, other):
776 if not isinstance(other, PurePath) or self._flavour is not other._flavour:
777 return NotImplemented
778 return self._cparts > other._cparts
779
780 def __ge__(self, other):
781 if not isinstance(other, PurePath) or self._flavour is not other._flavour:
782 return NotImplemented
783 return self._cparts >= other._cparts
784
Batuhan Taşkaya526606b2019-12-08 23:31:15 +0300785 def __class_getitem__(cls, type):
786 return cls
787
Antoine Pitrou31119e42013-11-22 17:38:12 +0100788 drive = property(attrgetter('_drv'),
789 doc="""The drive prefix (letter or UNC path), if any.""")
790
791 root = property(attrgetter('_root'),
792 doc="""The root of the path, if any.""")
793
794 @property
795 def anchor(self):
796 """The concatenation of the drive and root, or ''."""
797 anchor = self._drv + self._root
798 return anchor
799
800 @property
801 def name(self):
802 """The final path component, if any."""
803 parts = self._parts
804 if len(parts) == (1 if (self._drv or self._root) else 0):
805 return ''
806 return parts[-1]
807
808 @property
809 def suffix(self):
Ram Rachum8d4fef42019-11-02 18:46:24 +0200810 """
811 The final component's last suffix, if any.
812
813 This includes the leading period. For example: '.txt'
814 """
Antoine Pitrou31119e42013-11-22 17:38:12 +0100815 name = self.name
816 i = name.rfind('.')
817 if 0 < i < len(name) - 1:
818 return name[i:]
819 else:
820 return ''
821
822 @property
823 def suffixes(self):
Ram Rachum8d4fef42019-11-02 18:46:24 +0200824 """
825 A list of the final component's suffixes, if any.
826
827 These include the leading periods. For example: ['.tar', '.gz']
828 """
Antoine Pitrou31119e42013-11-22 17:38:12 +0100829 name = self.name
830 if name.endswith('.'):
831 return []
832 name = name.lstrip('.')
833 return ['.' + suffix for suffix in name.split('.')[1:]]
834
835 @property
836 def stem(self):
837 """The final path component, minus its last suffix."""
838 name = self.name
839 i = name.rfind('.')
840 if 0 < i < len(name) - 1:
841 return name[:i]
842 else:
843 return name
844
845 def with_name(self, name):
846 """Return a new path with the file name changed."""
847 if not self.name:
848 raise ValueError("%r has an empty name" % (self,))
Antoine Pitrou7084e732014-07-06 21:31:12 -0400849 drv, root, parts = self._flavour.parse_parts((name,))
850 if (not name or name[-1] in [self._flavour.sep, self._flavour.altsep]
851 or drv or root or len(parts) != 1):
852 raise ValueError("Invalid name %r" % (name))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100853 return self._from_parsed_parts(self._drv, self._root,
854 self._parts[:-1] + [name])
855
856 def with_suffix(self, suffix):
Stefan Otte46dc4e32018-08-03 22:49:42 +0200857 """Return a new path with the file suffix changed. If the path
858 has no suffix, add given suffix. If the given suffix is an empty
859 string, remove the suffix from the path.
860 """
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400861 f = self._flavour
862 if f.sep in suffix or f.altsep and f.altsep in suffix:
Berker Peksag423d05f2018-08-11 08:45:06 +0300863 raise ValueError("Invalid suffix %r" % (suffix,))
Antoine Pitroue50dafc2014-07-06 21:37:15 -0400864 if suffix and not suffix.startswith('.') or suffix == '.':
Antoine Pitrou1b02da92014-01-03 00:07:17 +0100865 raise ValueError("Invalid suffix %r" % (suffix))
Antoine Pitrou31119e42013-11-22 17:38:12 +0100866 name = self.name
867 if not name:
868 raise ValueError("%r has an empty name" % (self,))
869 old_suffix = self.suffix
870 if not old_suffix:
871 name = name + suffix
872 else:
873 name = name[:-len(old_suffix)] + suffix
874 return self._from_parsed_parts(self._drv, self._root,
875 self._parts[:-1] + [name])
876
877 def relative_to(self, *other):
878 """Return the relative path to another path identified by the passed
879 arguments. If the operation is not possible (because this is not
880 a subpath of the other path), raise ValueError.
881 """
882 # For the purpose of this method, drive and root are considered
883 # separate parts, i.e.:
884 # Path('c:/').relative_to('c:') gives Path('/')
885 # Path('c:/').relative_to('/') raise ValueError
886 if not other:
887 raise TypeError("need at least one argument")
888 parts = self._parts
889 drv = self._drv
890 root = self._root
Antoine Pitrou156b3612013-12-28 19:49:04 +0100891 if root:
892 abs_parts = [drv, root] + parts[1:]
Antoine Pitrou31119e42013-11-22 17:38:12 +0100893 else:
894 abs_parts = parts
895 to_drv, to_root, to_parts = self._parse_args(other)
Antoine Pitrou156b3612013-12-28 19:49:04 +0100896 if to_root:
897 to_abs_parts = [to_drv, to_root] + to_parts[1:]
Antoine Pitrou31119e42013-11-22 17:38:12 +0100898 else:
899 to_abs_parts = to_parts
900 n = len(to_abs_parts)
Antoine Pitrou156b3612013-12-28 19:49:04 +0100901 cf = self._flavour.casefold_parts
902 if (root or drv) if n == 0 else cf(abs_parts[:n]) != cf(to_abs_parts):
Antoine Pitrou31119e42013-11-22 17:38:12 +0100903 formatted = self._format_parsed_parts(to_drv, to_root, to_parts)
904 raise ValueError("{!r} does not start with {!r}"
905 .format(str(self), str(formatted)))
Antoine Pitrou156b3612013-12-28 19:49:04 +0100906 return self._from_parsed_parts('', root if n == 1 else '',
907 abs_parts[n:])
Antoine Pitrou31119e42013-11-22 17:38:12 +0100908
Hai Shi82642a02019-08-13 14:54:02 -0500909 def is_relative_to(self, *other):
910 """Return True if the path is relative to another path or False.
911 """
912 try:
913 self.relative_to(*other)
914 return True
915 except ValueError:
916 return False
917
Antoine Pitrou31119e42013-11-22 17:38:12 +0100918 @property
919 def parts(self):
920 """An object providing sequence-like access to the
921 components in the filesystem path."""
922 # We cache the tuple to avoid building a new one each time .parts
923 # is accessed. XXX is this necessary?
924 try:
925 return self._pparts
926 except AttributeError:
927 self._pparts = tuple(self._parts)
928 return self._pparts
929
930 def joinpath(self, *args):
931 """Combine this path with one or several arguments, and return a
932 new path representing either a subpath (if all arguments are relative
933 paths) or a totally different path (if one of the arguments is
934 anchored).
935 """
936 return self._make_child(args)
937
938 def __truediv__(self, key):
aiudirog4c69be22019-08-08 01:41:10 -0400939 try:
940 return self._make_child((key,))
941 except TypeError:
942 return NotImplemented
Antoine Pitrou31119e42013-11-22 17:38:12 +0100943
944 def __rtruediv__(self, key):
aiudirog4c69be22019-08-08 01:41:10 -0400945 try:
946 return self._from_parts([key] + self._parts)
947 except TypeError:
948 return NotImplemented
Antoine Pitrou31119e42013-11-22 17:38:12 +0100949
950 @property
951 def parent(self):
952 """The logical parent of the path."""
953 drv = self._drv
954 root = self._root
955 parts = self._parts
956 if len(parts) == 1 and (drv or root):
957 return self
958 return self._from_parsed_parts(drv, root, parts[:-1])
959
960 @property
961 def parents(self):
962 """A sequence of this path's logical parents."""
963 return _PathParents(self)
964
965 def is_absolute(self):
966 """True if the path is absolute (has both a root and, if applicable,
967 a drive)."""
968 if not self._root:
969 return False
970 return not self._flavour.has_drv or bool(self._drv)
971
972 def is_reserved(self):
973 """Return True if the path contains one of the special names reserved
974 by the system, if any."""
975 return self._flavour.is_reserved(self._parts)
976
977 def match(self, path_pattern):
978 """
979 Return True if this path matches the given pattern.
980 """
981 cf = self._flavour.casefold
982 path_pattern = cf(path_pattern)
983 drv, root, pat_parts = self._flavour.parse_parts((path_pattern,))
984 if not pat_parts:
985 raise ValueError("empty pattern")
986 if drv and drv != cf(self._drv):
987 return False
988 if root and root != cf(self._root):
989 return False
990 parts = self._cparts
991 if drv or root:
992 if len(pat_parts) != len(parts):
993 return False
994 pat_parts = pat_parts[1:]
995 elif len(pat_parts) > len(parts):
996 return False
997 for part, pat in zip(reversed(parts), reversed(pat_parts)):
998 if not fnmatch.fnmatchcase(part, pat):
999 return False
1000 return True
1001
Brett Cannon568be632016-06-10 12:20:49 -07001002# Can't subclass os.PathLike from PurePath and keep the constructor
1003# optimizations in PurePath._parse_args().
1004os.PathLike.register(PurePath)
1005
Antoine Pitrou31119e42013-11-22 17:38:12 +01001006
1007class PurePosixPath(PurePath):
chasondfa015c2018-02-19 08:36:32 +09001008 """PurePath subclass for non-Windows systems.
1009
1010 On a POSIX system, instantiating a PurePath should return this object.
1011 However, you can also instantiate it directly on any system.
1012 """
Antoine Pitrou31119e42013-11-22 17:38:12 +01001013 _flavour = _posix_flavour
1014 __slots__ = ()
1015
1016
1017class PureWindowsPath(PurePath):
chasondfa015c2018-02-19 08:36:32 +09001018 """PurePath subclass for Windows systems.
1019
1020 On a Windows system, instantiating a PurePath should return this object.
1021 However, you can also instantiate it directly on any system.
1022 """
Antoine Pitrou31119e42013-11-22 17:38:12 +01001023 _flavour = _windows_flavour
1024 __slots__ = ()
1025
1026
1027# Filesystem-accessing classes
1028
1029
1030class Path(PurePath):
chasondfa015c2018-02-19 08:36:32 +09001031 """PurePath subclass that can make system calls.
1032
1033 Path represents a filesystem path but unlike PurePath, also offers
1034 methods to do system calls on path objects. Depending on your system,
1035 instantiating a Path will return either a PosixPath or a WindowsPath
1036 object. You can also instantiate a PosixPath or WindowsPath directly,
1037 but cannot instantiate a WindowsPath on a POSIX system or vice versa.
1038 """
Antoine Pitrou31119e42013-11-22 17:38:12 +01001039 __slots__ = (
1040 '_accessor',
1041 '_closed',
1042 )
1043
1044 def __new__(cls, *args, **kwargs):
1045 if cls is Path:
1046 cls = WindowsPath if os.name == 'nt' else PosixPath
1047 self = cls._from_parts(args, init=False)
1048 if not self._flavour.is_supported:
1049 raise NotImplementedError("cannot instantiate %r on your system"
1050 % (cls.__name__,))
1051 self._init()
1052 return self
1053
1054 def _init(self,
1055 # Private non-constructor arguments
1056 template=None,
1057 ):
1058 self._closed = False
1059 if template is not None:
1060 self._accessor = template._accessor
1061 else:
1062 self._accessor = _normal_accessor
1063
1064 def _make_child_relpath(self, part):
1065 # This is an optimization used for dir walking. `part` must be
1066 # a single part relative to this path.
1067 parts = self._parts + [part]
1068 return self._from_parsed_parts(self._drv, self._root, parts)
1069
1070 def __enter__(self):
1071 if self._closed:
1072 self._raise_closed()
1073 return self
1074
1075 def __exit__(self, t, v, tb):
1076 self._closed = True
1077
1078 def _raise_closed(self):
1079 raise ValueError("I/O operation on closed path")
1080
1081 def _opener(self, name, flags, mode=0o666):
1082 # A stub for the opener argument to built-in open()
1083 return self._accessor.open(self, flags, mode)
1084
Antoine Pitrou4a60d422013-12-02 21:25:18 +01001085 def _raw_open(self, flags, mode=0o777):
1086 """
1087 Open the file pointed by this path and return a file descriptor,
1088 as os.open() does.
1089 """
1090 if self._closed:
1091 self._raise_closed()
1092 return self._accessor.open(self, flags, mode)
1093
Antoine Pitrou31119e42013-11-22 17:38:12 +01001094 # Public API
1095
1096 @classmethod
1097 def cwd(cls):
1098 """Return a new path pointing to the current working directory
1099 (as returned by os.getcwd()).
1100 """
1101 return cls(os.getcwd())
1102
Antoine Pitrou17cba7d2015-01-12 21:03:41 +01001103 @classmethod
1104 def home(cls):
1105 """Return a new path pointing to the user's home directory (as
1106 returned by os.path.expanduser('~')).
1107 """
1108 return cls(cls()._flavour.gethomedir(None))
1109
Antoine Pitrou43e3d942014-05-13 10:50:15 +02001110 def samefile(self, other_path):
Berker Peksag05492b82015-10-22 03:34:16 +03001111 """Return whether other_path is the same or not as this file
Berker Peksag267597f2015-10-21 20:10:24 +03001112 (as returned by os.path.samefile()).
Antoine Pitrou43e3d942014-05-13 10:50:15 +02001113 """
1114 st = self.stat()
1115 try:
1116 other_st = other_path.stat()
1117 except AttributeError:
1118 other_st = os.stat(other_path)
1119 return os.path.samestat(st, other_st)
1120
Antoine Pitrou31119e42013-11-22 17:38:12 +01001121 def iterdir(self):
1122 """Iterate over the files in this directory. Does not yield any
1123 result for the special paths '.' and '..'.
1124 """
1125 if self._closed:
1126 self._raise_closed()
1127 for name in self._accessor.listdir(self):
1128 if name in {'.', '..'}:
1129 # Yielding a path object for these makes little sense
1130 continue
1131 yield self._make_child_relpath(name)
1132 if self._closed:
1133 self._raise_closed()
1134
1135 def glob(self, pattern):
1136 """Iterate over this subtree and yield all existing files (of any
Eivind Teig537b6ca2019-02-11 11:47:09 +01001137 kind, including directories) matching the given relative pattern.
Antoine Pitrou31119e42013-11-22 17:38:12 +01001138 """
Berker Peksag4a208e42016-01-30 17:50:48 +02001139 if not pattern:
1140 raise ValueError("Unacceptable pattern: {!r}".format(pattern))
Antoine Pitrou31119e42013-11-22 17:38:12 +01001141 drv, root, pattern_parts = self._flavour.parse_parts((pattern,))
1142 if drv or root:
1143 raise NotImplementedError("Non-relative patterns are unsupported")
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +03001144 selector = _make_selector(tuple(pattern_parts), self._flavour)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001145 for p in selector.select_from(self):
1146 yield p
1147
1148 def rglob(self, pattern):
1149 """Recursively yield all existing files (of any kind, including
Eivind Teig537b6ca2019-02-11 11:47:09 +01001150 directories) matching the given relative pattern, anywhere in
1151 this subtree.
Antoine Pitrou31119e42013-11-22 17:38:12 +01001152 """
Antoine Pitrou31119e42013-11-22 17:38:12 +01001153 drv, root, pattern_parts = self._flavour.parse_parts((pattern,))
1154 if drv or root:
1155 raise NotImplementedError("Non-relative patterns are unsupported")
Serhiy Storchaka10ecbad2019-10-21 20:37:15 +03001156 selector = _make_selector(("**",) + tuple(pattern_parts), self._flavour)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001157 for p in selector.select_from(self):
1158 yield p
1159
1160 def absolute(self):
1161 """Return an absolute version of this path. This function works
1162 even if the path doesn't point to anything.
1163
1164 No normalization is done, i.e. all '.' and '..' will be kept along.
1165 Use resolve() to get the canonical path to a file.
1166 """
1167 # XXX untested yet!
1168 if self._closed:
1169 self._raise_closed()
1170 if self.is_absolute():
1171 return self
1172 # FIXME this must defer to the specific flavour (and, under Windows,
1173 # use nt._getfullpathname())
1174 obj = self._from_parts([os.getcwd()] + self._parts, init=False)
1175 obj._init(template=self)
1176 return obj
1177
Steve Dower98eb3602016-11-09 12:58:17 -08001178 def resolve(self, strict=False):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001179 """
1180 Make the path absolute, resolving all symlinks on the way and also
1181 normalizing it (for example turning slashes into backslashes under
1182 Windows).
1183 """
1184 if self._closed:
1185 self._raise_closed()
Steve Dower98eb3602016-11-09 12:58:17 -08001186 s = self._flavour.resolve(self, strict=strict)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001187 if s is None:
1188 # No symlink resolution => for consistency, raise an error if
1189 # the path doesn't exist or is forbidden
1190 self.stat()
1191 s = str(self.absolute())
1192 # Now we have no symlinks in the path, it's safe to normalize it.
1193 normed = self._flavour.pathmod.normpath(s)
1194 obj = self._from_parts((normed,), init=False)
1195 obj._init(template=self)
1196 return obj
1197
1198 def stat(self):
1199 """
1200 Return the result of the stat() system call on this path, like
1201 os.stat() does.
1202 """
1203 return self._accessor.stat(self)
1204
1205 def owner(self):
1206 """
1207 Return the login name of the file owner.
1208 """
1209 import pwd
1210 return pwd.getpwuid(self.stat().st_uid).pw_name
1211
1212 def group(self):
1213 """
1214 Return the group name of the file gid.
1215 """
1216 import grp
1217 return grp.getgrgid(self.stat().st_gid).gr_name
1218
Antoine Pitrou31119e42013-11-22 17:38:12 +01001219 def open(self, mode='r', buffering=-1, encoding=None,
1220 errors=None, newline=None):
1221 """
1222 Open the file pointed by this path and return a file object, as
1223 the built-in open() function does.
1224 """
1225 if self._closed:
1226 self._raise_closed()
Serhiy Storchaka62a99512017-03-25 13:42:11 +02001227 return io.open(self, mode, buffering, encoding, errors, newline,
Antoine Pitrou31119e42013-11-22 17:38:12 +01001228 opener=self._opener)
1229
Georg Brandlea683982014-10-01 19:12:33 +02001230 def read_bytes(self):
1231 """
1232 Open the file in bytes mode, read it, and close the file.
1233 """
1234 with self.open(mode='rb') as f:
1235 return f.read()
1236
1237 def read_text(self, encoding=None, errors=None):
1238 """
1239 Open the file in text mode, read it, and close the file.
1240 """
1241 with self.open(mode='r', encoding=encoding, errors=errors) as f:
1242 return f.read()
1243
1244 def write_bytes(self, data):
1245 """
1246 Open the file in bytes mode, write to it, and close the file.
1247 """
1248 # type-check for the buffer interface before truncating the file
1249 view = memoryview(data)
1250 with self.open(mode='wb') as f:
1251 return f.write(view)
1252
1253 def write_text(self, data, encoding=None, errors=None):
1254 """
1255 Open the file in text mode, write to it, and close the file.
1256 """
1257 if not isinstance(data, str):
1258 raise TypeError('data must be str, not %s' %
1259 data.__class__.__name__)
1260 with self.open(mode='w', encoding=encoding, errors=errors) as f:
1261 return f.write(data)
1262
Girtsa01ba332019-10-23 14:18:40 -07001263 def readlink(self):
1264 """
1265 Return the path to which the symbolic link points.
1266 """
1267 path = self._accessor.readlink(self)
1268 obj = self._from_parts((path,), init=False)
1269 obj._init(template=self)
1270 return obj
1271
Antoine Pitrou31119e42013-11-22 17:38:12 +01001272 def touch(self, mode=0o666, exist_ok=True):
1273 """
1274 Create this file with the given access mode, if it doesn't exist.
1275 """
1276 if self._closed:
1277 self._raise_closed()
1278 if exist_ok:
1279 # First try to bump modification time
1280 # Implementation note: GNU touch uses the UTIME_NOW option of
1281 # the utimensat() / futimens() functions.
Antoine Pitrou31119e42013-11-22 17:38:12 +01001282 try:
Antoine Pitrou2cf39172013-11-23 15:25:59 +01001283 self._accessor.utime(self, None)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001284 except OSError:
1285 # Avoid exception chaining
1286 pass
1287 else:
1288 return
1289 flags = os.O_CREAT | os.O_WRONLY
1290 if not exist_ok:
1291 flags |= os.O_EXCL
1292 fd = self._raw_open(flags, mode)
1293 os.close(fd)
1294
Barry Warsaw7c549c42014-08-05 11:28:12 -04001295 def mkdir(self, mode=0o777, parents=False, exist_ok=False):
Serhiy Storchakaaf7b9ec2017-03-24 20:51:53 +02001296 """
1297 Create a new directory at this given path.
1298 """
Antoine Pitrou31119e42013-11-22 17:38:12 +01001299 if self._closed:
1300 self._raise_closed()
Serhiy Storchakaaf7b9ec2017-03-24 20:51:53 +02001301 try:
1302 self._accessor.mkdir(self, mode)
1303 except FileNotFoundError:
1304 if not parents or self.parent == self:
1305 raise
Armin Rigo22a594a2017-04-13 20:08:15 +02001306 self.parent.mkdir(parents=True, exist_ok=True)
1307 self.mkdir(mode, parents=False, exist_ok=exist_ok)
Serhiy Storchakaaf7b9ec2017-03-24 20:51:53 +02001308 except OSError:
1309 # Cannot rely on checking for EEXIST, since the operating system
1310 # could give priority to other errors like EACCES or EROFS
1311 if not exist_ok or not self.is_dir():
1312 raise
Antoine Pitrou31119e42013-11-22 17:38:12 +01001313
1314 def chmod(self, mode):
1315 """
1316 Change the permissions of the path, like os.chmod().
1317 """
1318 if self._closed:
1319 self._raise_closed()
1320 self._accessor.chmod(self, mode)
1321
1322 def lchmod(self, mode):
1323 """
1324 Like chmod(), except if the path points to a symlink, the symlink's
1325 permissions are changed, rather than its target's.
1326 """
1327 if self._closed:
1328 self._raise_closed()
1329 self._accessor.lchmod(self, mode)
1330
‮zlohhcuB treboRd9e006b2019-05-16 00:02:11 +02001331 def unlink(self, missing_ok=False):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001332 """
1333 Remove this file or link.
1334 If the path is a directory, use rmdir() instead.
1335 """
1336 if self._closed:
1337 self._raise_closed()
‮zlohhcuB treboRd9e006b2019-05-16 00:02:11 +02001338 try:
1339 self._accessor.unlink(self)
1340 except FileNotFoundError:
1341 if not missing_ok:
1342 raise
Antoine Pitrou31119e42013-11-22 17:38:12 +01001343
1344 def rmdir(self):
1345 """
1346 Remove this directory. The directory must be empty.
1347 """
1348 if self._closed:
1349 self._raise_closed()
1350 self._accessor.rmdir(self)
1351
1352 def lstat(self):
1353 """
1354 Like stat(), except if the path points to a symlink, the symlink's
1355 status information is returned, rather than its target's.
1356 """
1357 if self._closed:
1358 self._raise_closed()
1359 return self._accessor.lstat(self)
1360
Joannah Nanjekye6b5b0132019-05-04 11:27:10 -04001361 def link_to(self, target):
1362 """
1363 Create a hard link pointing to a path named target.
1364 """
1365 if self._closed:
1366 self._raise_closed()
1367 self._accessor.link_to(self, target)
1368
Antoine Pitrou31119e42013-11-22 17:38:12 +01001369 def rename(self, target):
1370 """
hui shang088a09a2019-09-11 21:26:49 +08001371 Rename this path to the given path,
1372 and return a new Path instance pointing to the given path.
Antoine Pitrou31119e42013-11-22 17:38:12 +01001373 """
1374 if self._closed:
1375 self._raise_closed()
1376 self._accessor.rename(self, target)
hui shang088a09a2019-09-11 21:26:49 +08001377 return self.__class__(target)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001378
1379 def replace(self, target):
1380 """
1381 Rename this path to the given path, clobbering the existing
hui shang088a09a2019-09-11 21:26:49 +08001382 destination if it exists, and return a new Path instance
1383 pointing to the given path.
Antoine Pitrou31119e42013-11-22 17:38:12 +01001384 """
1385 if self._closed:
1386 self._raise_closed()
1387 self._accessor.replace(self, target)
hui shang088a09a2019-09-11 21:26:49 +08001388 return self.__class__(target)
Antoine Pitrou31119e42013-11-22 17:38:12 +01001389
1390 def symlink_to(self, target, target_is_directory=False):
1391 """
1392 Make this path a symlink pointing to the given path.
1393 Note the order of arguments (self, target) is the reverse of os.symlink's.
1394 """
1395 if self._closed:
1396 self._raise_closed()
1397 self._accessor.symlink(target, self, target_is_directory)
1398
1399 # Convenience functions for querying the stat results
1400
1401 def exists(self):
1402 """
1403 Whether this path exists.
1404 """
1405 try:
1406 self.stat()
1407 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -08001408 if not _ignore_error(e):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001409 raise
1410 return False
Serhiy Storchaka0185f342018-09-18 11:28:51 +03001411 except ValueError:
1412 # Non-encodable path
1413 return False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001414 return True
1415
1416 def is_dir(self):
1417 """
1418 Whether this path is a directory.
1419 """
1420 try:
1421 return S_ISDIR(self.stat().st_mode)
1422 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -08001423 if not _ignore_error(e):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001424 raise
1425 # Path doesn't exist or is a broken symlink
1426 # (see https://bitbucket.org/pitrou/pathlib/issue/12/)
1427 return False
Serhiy Storchaka0185f342018-09-18 11:28:51 +03001428 except ValueError:
1429 # Non-encodable path
1430 return False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001431
1432 def is_file(self):
1433 """
1434 Whether this path is a regular file (also True for symlinks pointing
1435 to regular files).
1436 """
1437 try:
1438 return S_ISREG(self.stat().st_mode)
1439 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -08001440 if not _ignore_error(e):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001441 raise
1442 # Path doesn't exist or is a broken symlink
1443 # (see https://bitbucket.org/pitrou/pathlib/issue/12/)
1444 return False
Serhiy Storchaka0185f342018-09-18 11:28:51 +03001445 except ValueError:
1446 # Non-encodable path
1447 return False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001448
Cooper Lees173ff4a2017-08-01 15:35:45 -07001449 def is_mount(self):
1450 """
1451 Check if this path is a POSIX mount point
1452 """
1453 # Need to exist and be a dir
1454 if not self.exists() or not self.is_dir():
1455 return False
1456
1457 parent = Path(self.parent)
1458 try:
1459 parent_dev = parent.stat().st_dev
1460 except OSError:
1461 return False
1462
1463 dev = self.stat().st_dev
1464 if dev != parent_dev:
1465 return True
1466 ino = self.stat().st_ino
1467 parent_ino = parent.stat().st_ino
1468 return ino == parent_ino
1469
Antoine Pitrou31119e42013-11-22 17:38:12 +01001470 def is_symlink(self):
1471 """
1472 Whether this path is a symbolic link.
1473 """
1474 try:
1475 return S_ISLNK(self.lstat().st_mode)
1476 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -08001477 if not _ignore_error(e):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001478 raise
1479 # Path doesn't exist
1480 return False
Serhiy Storchaka0185f342018-09-18 11:28:51 +03001481 except ValueError:
1482 # Non-encodable path
1483 return False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001484
1485 def is_block_device(self):
1486 """
1487 Whether this path is a block device.
1488 """
1489 try:
1490 return S_ISBLK(self.stat().st_mode)
1491 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -08001492 if not _ignore_error(e):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001493 raise
1494 # Path doesn't exist or is a broken symlink
1495 # (see https://bitbucket.org/pitrou/pathlib/issue/12/)
1496 return False
Serhiy Storchaka0185f342018-09-18 11:28:51 +03001497 except ValueError:
1498 # Non-encodable path
1499 return False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001500
1501 def is_char_device(self):
1502 """
1503 Whether this path is a character device.
1504 """
1505 try:
1506 return S_ISCHR(self.stat().st_mode)
1507 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -08001508 if not _ignore_error(e):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001509 raise
1510 # Path doesn't exist or is a broken symlink
1511 # (see https://bitbucket.org/pitrou/pathlib/issue/12/)
1512 return False
Serhiy Storchaka0185f342018-09-18 11:28:51 +03001513 except ValueError:
1514 # Non-encodable path
1515 return False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001516
1517 def is_fifo(self):
1518 """
1519 Whether this path is a FIFO.
1520 """
1521 try:
1522 return S_ISFIFO(self.stat().st_mode)
1523 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -08001524 if not _ignore_error(e):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001525 raise
1526 # Path doesn't exist or is a broken symlink
1527 # (see https://bitbucket.org/pitrou/pathlib/issue/12/)
1528 return False
Serhiy Storchaka0185f342018-09-18 11:28:51 +03001529 except ValueError:
1530 # Non-encodable path
1531 return False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001532
1533 def is_socket(self):
1534 """
1535 Whether this path is a socket.
1536 """
1537 try:
1538 return S_ISSOCK(self.stat().st_mode)
1539 except OSError as e:
Steve Dower2f6fae62019-02-03 23:08:18 -08001540 if not _ignore_error(e):
Antoine Pitrou31119e42013-11-22 17:38:12 +01001541 raise
1542 # Path doesn't exist or is a broken symlink
1543 # (see https://bitbucket.org/pitrou/pathlib/issue/12/)
1544 return False
Serhiy Storchaka0185f342018-09-18 11:28:51 +03001545 except ValueError:
1546 # Non-encodable path
1547 return False
Antoine Pitrou31119e42013-11-22 17:38:12 +01001548
Antoine Pitrou8477ed62014-12-30 20:54:45 +01001549 def expanduser(self):
1550 """ Return a new path with expanded ~ and ~user constructs
1551 (as returned by os.path.expanduser)
1552 """
1553 if (not (self._drv or self._root) and
1554 self._parts and self._parts[0][:1] == '~'):
1555 homedir = self._flavour.gethomedir(self._parts[0][1:])
1556 return self._from_parts([homedir] + self._parts[1:])
1557
1558 return self
1559
Antoine Pitrou31119e42013-11-22 17:38:12 +01001560
1561class PosixPath(Path, PurePosixPath):
chasondfa015c2018-02-19 08:36:32 +09001562 """Path subclass for non-Windows systems.
1563
1564 On a POSIX system, instantiating a Path should return this object.
1565 """
Antoine Pitrou31119e42013-11-22 17:38:12 +01001566 __slots__ = ()
1567
1568class WindowsPath(Path, PureWindowsPath):
chasondfa015c2018-02-19 08:36:32 +09001569 """Path subclass for Windows systems.
1570
1571 On a Windows system, instantiating a Path should return this object.
1572 """
Antoine Pitrou31119e42013-11-22 17:38:12 +01001573 __slots__ = ()
Berker Peksag04d42292016-03-11 23:07:27 +02001574
1575 def owner(self):
1576 raise NotImplementedError("Path.owner() is unsupported on this system")
1577
1578 def group(self):
1579 raise NotImplementedError("Path.group() is unsupported on this system")
Cooper Lees173ff4a2017-08-01 15:35:45 -07001580
1581 def is_mount(self):
1582 raise NotImplementedError("Path.is_mount() is unsupported on this system")