• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1r"""OS routines for NT or Posix depending on what system we're on.
2
3This exports:
4  - all functions from posix or nt, e.g. unlink, stat, etc.
5  - os.path is either posixpath or ntpath
6  - os.name is either 'posix' or 'nt'
7  - os.curdir is a string representing the current directory (always '.')
8  - os.pardir is a string representing the parent directory (always '..')
9  - os.sep is the (or a most common) pathname separator ('/' or '\\')
10  - os.extsep is the extension separator (always '.')
11  - os.altsep is the alternate pathname separator (None or '/')
12  - os.pathsep is the component separator used in $PATH etc
13  - os.linesep is the line separator in text files ('\r' or '\n' or '\r\n')
14  - os.defpath is the default search path for executables
15  - os.devnull is the file path of the null device ('/dev/null', etc.)
16
17Programs that import and use 'os' stand a better chance of being
18portable between different platforms.  Of course, they must then
19only use functions that are defined by all platforms (e.g., unlink
20and opendir), and leave all pathname manipulation to os.path
21(e.g., split and join).
22"""
23
24#'
25import abc
26import sys
27import stat as st
28
29from _collections_abc import _check_methods
30
31GenericAlias = type(list[int])
32
33_names = sys.builtin_module_names
34
35# Note:  more names are added to __all__ later.
36__all__ = ["altsep", "curdir", "pardir", "sep", "pathsep", "linesep",
37           "defpath", "name", "path", "devnull", "SEEK_SET", "SEEK_CUR",
38           "SEEK_END", "fsencode", "fsdecode", "get_exec_path", "fdopen",
39           "popen", "extsep"]
40
41def _exists(name):
42    return name in globals()
43
44def _get_exports_list(module):
45    try:
46        return list(module.__all__)
47    except AttributeError:
48        return [n for n in dir(module) if n[0] != '_']
49
50# Any new dependencies of the os module and/or changes in path separator
51# requires updating importlib as well.
52if 'posix' in _names:
53    name = 'posix'
54    linesep = '\n'
55    from posix import *
56    try:
57        from posix import _exit
58        __all__.append('_exit')
59    except ImportError:
60        pass
61    import posixpath as path
62
63    try:
64        from posix import _have_functions
65    except ImportError:
66        pass
67
68    import posix
69    __all__.extend(_get_exports_list(posix))
70    del posix
71
72elif 'nt' in _names:
73    name = 'nt'
74    linesep = '\r\n'
75    from nt import *
76    try:
77        from nt import _exit
78        __all__.append('_exit')
79    except ImportError:
80        pass
81    import ntpath as path
82
83    import nt
84    __all__.extend(_get_exports_list(nt))
85    del nt
86
87    try:
88        from nt import _have_functions
89    except ImportError:
90        pass
91
92else:
93    raise ImportError('no os specific module found')
94
95sys.modules['os.path'] = path
96from os.path import (curdir, pardir, sep, pathsep, defpath, extsep, altsep,
97    devnull)
98
99del _names
100
101
102if _exists("_have_functions"):
103    _globals = globals()
104    def _add(str, fn):
105        if (fn in _globals) and (str in _have_functions):
106            _set.add(_globals[fn])
107
108    _set = set()
109    _add("HAVE_FACCESSAT",  "access")
110    _add("HAVE_FCHMODAT",   "chmod")
111    _add("HAVE_FCHOWNAT",   "chown")
112    _add("HAVE_FSTATAT",    "stat")
113    _add("HAVE_FUTIMESAT",  "utime")
114    _add("HAVE_LINKAT",     "link")
115    _add("HAVE_MKDIRAT",    "mkdir")
116    _add("HAVE_MKFIFOAT",   "mkfifo")
117    _add("HAVE_MKNODAT",    "mknod")
118    _add("HAVE_OPENAT",     "open")
119    _add("HAVE_READLINKAT", "readlink")
120    _add("HAVE_RENAMEAT",   "rename")
121    _add("HAVE_SYMLINKAT",  "symlink")
122    _add("HAVE_UNLINKAT",   "unlink")
123    _add("HAVE_UNLINKAT",   "rmdir")
124    _add("HAVE_UTIMENSAT",  "utime")
125    supports_dir_fd = _set
126
127    _set = set()
128    _add("HAVE_FACCESSAT",  "access")
129    supports_effective_ids = _set
130
131    _set = set()
132    _add("HAVE_FCHDIR",     "chdir")
133    _add("HAVE_FCHMOD",     "chmod")
134    _add("HAVE_FCHOWN",     "chown")
135    _add("HAVE_FDOPENDIR",  "listdir")
136    _add("HAVE_FDOPENDIR",  "scandir")
137    _add("HAVE_FEXECVE",    "execve")
138    _set.add(stat) # fstat always works
139    _add("HAVE_FTRUNCATE",  "truncate")
140    _add("HAVE_FUTIMENS",   "utime")
141    _add("HAVE_FUTIMES",    "utime")
142    _add("HAVE_FPATHCONF",  "pathconf")
143    if _exists("statvfs") and _exists("fstatvfs"): # mac os x10.3
144        _add("HAVE_FSTATVFS", "statvfs")
145    supports_fd = _set
146
147    _set = set()
148    _add("HAVE_FACCESSAT",  "access")
149    # Some platforms don't support lchmod().  Often the function exists
150    # anyway, as a stub that always returns ENOSUP or perhaps EOPNOTSUPP.
151    # (No, I don't know why that's a good design.)  ./configure will detect
152    # this and reject it--so HAVE_LCHMOD still won't be defined on such
153    # platforms.  This is Very Helpful.
154    #
155    # However, sometimes platforms without a working lchmod() *do* have
156    # fchmodat().  (Examples: Linux kernel 3.2 with glibc 2.15,
157    # OpenIndiana 3.x.)  And fchmodat() has a flag that theoretically makes
158    # it behave like lchmod().  So in theory it would be a suitable
159    # replacement for lchmod().  But when lchmod() doesn't work, fchmodat()'s
160    # flag doesn't work *either*.  Sadly ./configure isn't sophisticated
161    # enough to detect this condition--it only determines whether or not
162    # fchmodat() minimally works.
163    #
164    # Therefore we simply ignore fchmodat() when deciding whether or not
165    # os.chmod supports follow_symlinks.  Just checking lchmod() is
166    # sufficient.  After all--if you have a working fchmodat(), your
167    # lchmod() almost certainly works too.
168    #
169    # _add("HAVE_FCHMODAT",   "chmod")
170    _add("HAVE_FCHOWNAT",   "chown")
171    _add("HAVE_FSTATAT",    "stat")
172    _add("HAVE_LCHFLAGS",   "chflags")
173    _add("HAVE_LCHMOD",     "chmod")
174    if _exists("lchown"): # mac os x10.3
175        _add("HAVE_LCHOWN", "chown")
176    _add("HAVE_LINKAT",     "link")
177    _add("HAVE_LUTIMES",    "utime")
178    _add("HAVE_LSTAT",      "stat")
179    _add("HAVE_FSTATAT",    "stat")
180    _add("HAVE_UTIMENSAT",  "utime")
181    _add("MS_WINDOWS",      "stat")
182    supports_follow_symlinks = _set
183
184    del _set
185    del _have_functions
186    del _globals
187    del _add
188
189
190# Python uses fixed values for the SEEK_ constants; they are mapped
191# to native constants if necessary in posixmodule.c
192# Other possible SEEK values are directly imported from posixmodule.c
193SEEK_SET = 0
194SEEK_CUR = 1
195SEEK_END = 2
196
197# Super directory utilities.
198# (Inspired by Eric Raymond; the doc strings are mostly his)
199
200def makedirs(name, mode=0o777, exist_ok=False):
201    """makedirs(name [, mode=0o777][, exist_ok=False])
202
203    Super-mkdir; create a leaf directory and all intermediate ones.  Works like
204    mkdir, except that any intermediate path segment (not just the rightmost)
205    will be created if it does not exist. If the target directory already
206    exists, raise an OSError if exist_ok is False. Otherwise no exception is
207    raised.  This is recursive.
208
209    """
210    head, tail = path.split(name)
211    if not tail:
212        head, tail = path.split(head)
213    if head and tail and not path.exists(head):
214        try:
215            makedirs(head, exist_ok=exist_ok)
216        except FileExistsError:
217            # Defeats race condition when another thread created the path
218            pass
219        cdir = curdir
220        if isinstance(tail, bytes):
221            cdir = bytes(curdir, 'ASCII')
222        if tail == cdir:           # xxx/newdir/. exists if xxx/newdir exists
223            return
224    try:
225        mkdir(name, mode)
226    except OSError:
227        # Cannot rely on checking for EEXIST, since the operating system
228        # could give priority to other errors like EACCES or EROFS
229        if not exist_ok or not path.isdir(name):
230            raise
231
232def removedirs(name):
233    """removedirs(name)
234
235    Super-rmdir; remove a leaf directory and all empty intermediate
236    ones.  Works like rmdir except that, if the leaf directory is
237    successfully removed, directories corresponding to rightmost path
238    segments will be pruned away until either the whole path is
239    consumed or an error occurs.  Errors during this latter phase are
240    ignored -- they generally mean that a directory was not empty.
241
242    """
243    rmdir(name)
244    head, tail = path.split(name)
245    if not tail:
246        head, tail = path.split(head)
247    while head and tail:
248        try:
249            rmdir(head)
250        except OSError:
251            break
252        head, tail = path.split(head)
253
254def renames(old, new):
255    """renames(old, new)
256
257    Super-rename; create directories as necessary and delete any left
258    empty.  Works like rename, except creation of any intermediate
259    directories needed to make the new pathname good is attempted
260    first.  After the rename, directories corresponding to rightmost
261    path segments of the old name will be pruned until either the
262    whole path is consumed or a nonempty directory is found.
263
264    Note: this function can fail with the new directory structure made
265    if you lack permissions needed to unlink the leaf directory or
266    file.
267
268    """
269    head, tail = path.split(new)
270    if head and tail and not path.exists(head):
271        makedirs(head)
272    rename(old, new)
273    head, tail = path.split(old)
274    if head and tail:
275        try:
276            removedirs(head)
277        except OSError:
278            pass
279
280__all__.extend(["makedirs", "removedirs", "renames"])
281
282def walk(top, topdown=True, onerror=None, followlinks=False):
283    """Directory tree generator.
284
285    For each directory in the directory tree rooted at top (including top
286    itself, but excluding '.' and '..'), yields a 3-tuple
287
288        dirpath, dirnames, filenames
289
290    dirpath is a string, the path to the directory.  dirnames is a list of
291    the names of the subdirectories in dirpath (excluding '.' and '..').
292    filenames is a list of the names of the non-directory files in dirpath.
293    Note that the names in the lists are just names, with no path components.
294    To get a full path (which begins with top) to a file or directory in
295    dirpath, do os.path.join(dirpath, name).
296
297    If optional arg 'topdown' is true or not specified, the triple for a
298    directory is generated before the triples for any of its subdirectories
299    (directories are generated top down).  If topdown is false, the triple
300    for a directory is generated after the triples for all of its
301    subdirectories (directories are generated bottom up).
302
303    When topdown is true, the caller can modify the dirnames list in-place
304    (e.g., via del or slice assignment), and walk will only recurse into the
305    subdirectories whose names remain in dirnames; this can be used to prune the
306    search, or to impose a specific order of visiting.  Modifying dirnames when
307    topdown is false has no effect on the behavior of os.walk(), since the
308    directories in dirnames have already been generated by the time dirnames
309    itself is generated. No matter the value of topdown, the list of
310    subdirectories is retrieved before the tuples for the directory and its
311    subdirectories are generated.
312
313    By default errors from the os.scandir() call are ignored.  If
314    optional arg 'onerror' is specified, it should be a function; it
315    will be called with one argument, an OSError instance.  It can
316    report the error to continue with the walk, or raise the exception
317    to abort the walk.  Note that the filename is available as the
318    filename attribute of the exception object.
319
320    By default, os.walk does not follow symbolic links to subdirectories on
321    systems that support them.  In order to get this functionality, set the
322    optional argument 'followlinks' to true.
323
324    Caution:  if you pass a relative pathname for top, don't change the
325    current working directory between resumptions of walk.  walk never
326    changes the current directory, and assumes that the client doesn't
327    either.
328
329    Example:
330
331    import os
332    from os.path import join, getsize
333    for root, dirs, files in os.walk('python/Lib/email'):
334        print(root, "consumes", end="")
335        print(sum(getsize(join(root, name)) for name in files), end="")
336        print("bytes in", len(files), "non-directory files")
337        if 'CVS' in dirs:
338            dirs.remove('CVS')  # don't visit CVS directories
339
340    """
341    sys.audit("os.walk", top, topdown, onerror, followlinks)
342    return _walk(fspath(top), topdown, onerror, followlinks)
343
344def _walk(top, topdown, onerror, followlinks):
345    dirs = []
346    nondirs = []
347    walk_dirs = []
348
349    # We may not have read permission for top, in which case we can't
350    # get a list of the files the directory contains.  os.walk
351    # always suppressed the exception then, rather than blow up for a
352    # minor reason when (say) a thousand readable directories are still
353    # left to visit.  That logic is copied here.
354    try:
355        # Note that scandir is global in this module due
356        # to earlier import-*.
357        scandir_it = scandir(top)
358    except OSError as error:
359        if onerror is not None:
360            onerror(error)
361        return
362
363    with scandir_it:
364        while True:
365            try:
366                try:
367                    entry = next(scandir_it)
368                except StopIteration:
369                    break
370            except OSError as error:
371                if onerror is not None:
372                    onerror(error)
373                return
374
375            try:
376                is_dir = entry.is_dir()
377            except OSError:
378                # If is_dir() raises an OSError, consider that the entry is not
379                # a directory, same behaviour than os.path.isdir().
380                is_dir = False
381
382            if is_dir:
383                dirs.append(entry.name)
384            else:
385                nondirs.append(entry.name)
386
387            if not topdown and is_dir:
388                # Bottom-up: recurse into sub-directory, but exclude symlinks to
389                # directories if followlinks is False
390                if followlinks:
391                    walk_into = True
392                else:
393                    try:
394                        is_symlink = entry.is_symlink()
395                    except OSError:
396                        # If is_symlink() raises an OSError, consider that the
397                        # entry is not a symbolic link, same behaviour than
398                        # os.path.islink().
399                        is_symlink = False
400                    walk_into = not is_symlink
401
402                if walk_into:
403                    walk_dirs.append(entry.path)
404
405    # Yield before recursion if going top down
406    if topdown:
407        yield top, dirs, nondirs
408
409        # Recurse into sub-directories
410        islink, join = path.islink, path.join
411        for dirname in dirs:
412            new_path = join(top, dirname)
413            # Issue #23605: os.path.islink() is used instead of caching
414            # entry.is_symlink() result during the loop on os.scandir() because
415            # the caller can replace the directory entry during the "yield"
416            # above.
417            if followlinks or not islink(new_path):
418                yield from _walk(new_path, topdown, onerror, followlinks)
419    else:
420        # Recurse into sub-directories
421        for new_path in walk_dirs:
422            yield from _walk(new_path, topdown, onerror, followlinks)
423        # Yield after recursion if going bottom up
424        yield top, dirs, nondirs
425
426__all__.append("walk")
427
428if {open, stat} <= supports_dir_fd and {scandir, stat} <= supports_fd:
429
430    def fwalk(top=".", topdown=True, onerror=None, *, follow_symlinks=False, dir_fd=None):
431        """Directory tree generator.
432
433        This behaves exactly like walk(), except that it yields a 4-tuple
434
435            dirpath, dirnames, filenames, dirfd
436
437        `dirpath`, `dirnames` and `filenames` are identical to walk() output,
438        and `dirfd` is a file descriptor referring to the directory `dirpath`.
439
440        The advantage of fwalk() over walk() is that it's safe against symlink
441        races (when follow_symlinks is False).
442
443        If dir_fd is not None, it should be a file descriptor open to a directory,
444          and top should be relative; top will then be relative to that directory.
445          (dir_fd is always supported for fwalk.)
446
447        Caution:
448        Since fwalk() yields file descriptors, those are only valid until the
449        next iteration step, so you should dup() them if you want to keep them
450        for a longer period.
451
452        Example:
453
454        import os
455        for root, dirs, files, rootfd in os.fwalk('python/Lib/email'):
456            print(root, "consumes", end="")
457            print(sum(os.stat(name, dir_fd=rootfd).st_size for name in files),
458                  end="")
459            print("bytes in", len(files), "non-directory files")
460            if 'CVS' in dirs:
461                dirs.remove('CVS')  # don't visit CVS directories
462        """
463        sys.audit("os.fwalk", top, topdown, onerror, follow_symlinks, dir_fd)
464        if not isinstance(top, int) or not hasattr(top, '__index__'):
465            top = fspath(top)
466        # Note: To guard against symlink races, we use the standard
467        # lstat()/open()/fstat() trick.
468        if not follow_symlinks:
469            orig_st = stat(top, follow_symlinks=False, dir_fd=dir_fd)
470        topfd = open(top, O_RDONLY, dir_fd=dir_fd)
471        try:
472            if (follow_symlinks or (st.S_ISDIR(orig_st.st_mode) and
473                                    path.samestat(orig_st, stat(topfd)))):
474                yield from _fwalk(topfd, top, isinstance(top, bytes),
475                                  topdown, onerror, follow_symlinks)
476        finally:
477            close(topfd)
478
479    def _fwalk(topfd, toppath, isbytes, topdown, onerror, follow_symlinks):
480        # Note: This uses O(depth of the directory tree) file descriptors: if
481        # necessary, it can be adapted to only require O(1) FDs, see issue
482        # #13734.
483
484        scandir_it = scandir(topfd)
485        dirs = []
486        nondirs = []
487        entries = None if topdown or follow_symlinks else []
488        for entry in scandir_it:
489            name = entry.name
490            if isbytes:
491                name = fsencode(name)
492            try:
493                if entry.is_dir():
494                    dirs.append(name)
495                    if entries is not None:
496                        entries.append(entry)
497                else:
498                    nondirs.append(name)
499            except OSError:
500                try:
501                    # Add dangling symlinks, ignore disappeared files
502                    if entry.is_symlink():
503                        nondirs.append(name)
504                except OSError:
505                    pass
506
507        if topdown:
508            yield toppath, dirs, nondirs, topfd
509
510        for name in dirs if entries is None else zip(dirs, entries):
511            try:
512                if not follow_symlinks:
513                    if topdown:
514                        orig_st = stat(name, dir_fd=topfd, follow_symlinks=False)
515                    else:
516                        assert entries is not None
517                        name, entry = name
518                        orig_st = entry.stat(follow_symlinks=False)
519                dirfd = open(name, O_RDONLY, dir_fd=topfd)
520            except OSError as err:
521                if onerror is not None:
522                    onerror(err)
523                continue
524            try:
525                if follow_symlinks or path.samestat(orig_st, stat(dirfd)):
526                    dirpath = path.join(toppath, name)
527                    yield from _fwalk(dirfd, dirpath, isbytes,
528                                      topdown, onerror, follow_symlinks)
529            finally:
530                close(dirfd)
531
532        if not topdown:
533            yield toppath, dirs, nondirs, topfd
534
535    __all__.append("fwalk")
536
537def execl(file, *args):
538    """execl(file, *args)
539
540    Execute the executable file with argument list args, replacing the
541    current process. """
542    execv(file, args)
543
544def execle(file, *args):
545    """execle(file, *args, env)
546
547    Execute the executable file with argument list args and
548    environment env, replacing the current process. """
549    env = args[-1]
550    execve(file, args[:-1], env)
551
552def execlp(file, *args):
553    """execlp(file, *args)
554
555    Execute the executable file (which is searched for along $PATH)
556    with argument list args, replacing the current process. """
557    execvp(file, args)
558
559def execlpe(file, *args):
560    """execlpe(file, *args, env)
561
562    Execute the executable file (which is searched for along $PATH)
563    with argument list args and environment env, replacing the current
564    process. """
565    env = args[-1]
566    execvpe(file, args[:-1], env)
567
568def execvp(file, args):
569    """execvp(file, args)
570
571    Execute the executable file (which is searched for along $PATH)
572    with argument list args, replacing the current process.
573    args may be a list or tuple of strings. """
574    _execvpe(file, args)
575
576def execvpe(file, args, env):
577    """execvpe(file, args, env)
578
579    Execute the executable file (which is searched for along $PATH)
580    with argument list args and environment env, replacing the
581    current process.
582    args may be a list or tuple of strings. """
583    _execvpe(file, args, env)
584
585__all__.extend(["execl","execle","execlp","execlpe","execvp","execvpe"])
586
587def _execvpe(file, args, env=None):
588    if env is not None:
589        exec_func = execve
590        argrest = (args, env)
591    else:
592        exec_func = execv
593        argrest = (args,)
594        env = environ
595
596    if path.dirname(file):
597        exec_func(file, *argrest)
598        return
599    saved_exc = None
600    path_list = get_exec_path(env)
601    if name != 'nt':
602        file = fsencode(file)
603        path_list = map(fsencode, path_list)
604    for dir in path_list:
605        fullname = path.join(dir, file)
606        try:
607            exec_func(fullname, *argrest)
608        except (FileNotFoundError, NotADirectoryError) as e:
609            last_exc = e
610        except OSError as e:
611            last_exc = e
612            if saved_exc is None:
613                saved_exc = e
614    if saved_exc is not None:
615        raise saved_exc
616    raise last_exc
617
618
619def get_exec_path(env=None):
620    """Returns the sequence of directories that will be searched for the
621    named executable (similar to a shell) when launching a process.
622
623    *env* must be an environment variable dict or None.  If *env* is None,
624    os.environ will be used.
625    """
626    # Use a local import instead of a global import to limit the number of
627    # modules loaded at startup: the os module is always loaded at startup by
628    # Python. It may also avoid a bootstrap issue.
629    import warnings
630
631    if env is None:
632        env = environ
633
634    # {b'PATH': ...}.get('PATH') and {'PATH': ...}.get(b'PATH') emit a
635    # BytesWarning when using python -b or python -bb: ignore the warning
636    with warnings.catch_warnings():
637        warnings.simplefilter("ignore", BytesWarning)
638
639        try:
640            path_list = env.get('PATH')
641        except TypeError:
642            path_list = None
643
644        if supports_bytes_environ:
645            try:
646                path_listb = env[b'PATH']
647            except (KeyError, TypeError):
648                pass
649            else:
650                if path_list is not None:
651                    raise ValueError(
652                        "env cannot contain 'PATH' and b'PATH' keys")
653                path_list = path_listb
654
655            if path_list is not None and isinstance(path_list, bytes):
656                path_list = fsdecode(path_list)
657
658    if path_list is None:
659        path_list = defpath
660    return path_list.split(pathsep)
661
662
663# Change environ to automatically call putenv() and unsetenv()
664from _collections_abc import MutableMapping, Mapping
665
666class _Environ(MutableMapping):
667    def __init__(self, data, encodekey, decodekey, encodevalue, decodevalue):
668        self.encodekey = encodekey
669        self.decodekey = decodekey
670        self.encodevalue = encodevalue
671        self.decodevalue = decodevalue
672        self._data = data
673
674    def __getitem__(self, key):
675        try:
676            value = self._data[self.encodekey(key)]
677        except KeyError:
678            # raise KeyError with the original key value
679            raise KeyError(key) from None
680        return self.decodevalue(value)
681
682    def __setitem__(self, key, value):
683        key = self.encodekey(key)
684        value = self.encodevalue(value)
685        putenv(key, value)
686        self._data[key] = value
687
688    def __delitem__(self, key):
689        encodedkey = self.encodekey(key)
690        unsetenv(encodedkey)
691        try:
692            del self._data[encodedkey]
693        except KeyError:
694            # raise KeyError with the original key value
695            raise KeyError(key) from None
696
697    def __iter__(self):
698        # list() from dict object is an atomic operation
699        keys = list(self._data)
700        for key in keys:
701            yield self.decodekey(key)
702
703    def __len__(self):
704        return len(self._data)
705
706    def __repr__(self):
707        return 'environ({{{}}})'.format(', '.join(
708            ('{!r}: {!r}'.format(self.decodekey(key), self.decodevalue(value))
709            for key, value in self._data.items())))
710
711    def copy(self):
712        return dict(self)
713
714    def setdefault(self, key, value):
715        if key not in self:
716            self[key] = value
717        return self[key]
718
719    def __ior__(self, other):
720        self.update(other)
721        return self
722
723    def __or__(self, other):
724        if not isinstance(other, Mapping):
725            return NotImplemented
726        new = dict(self)
727        new.update(other)
728        return new
729
730    def __ror__(self, other):
731        if not isinstance(other, Mapping):
732            return NotImplemented
733        new = dict(other)
734        new.update(self)
735        return new
736
737def _createenviron():
738    if name == 'nt':
739        # Where Env Var Names Must Be UPPERCASE
740        def check_str(value):
741            if not isinstance(value, str):
742                raise TypeError("str expected, not %s" % type(value).__name__)
743            return value
744        encode = check_str
745        decode = str
746        def encodekey(key):
747            return encode(key).upper()
748        data = {}
749        for key, value in environ.items():
750            data[encodekey(key)] = value
751    else:
752        # Where Env Var Names Can Be Mixed Case
753        encoding = sys.getfilesystemencoding()
754        def encode(value):
755            if not isinstance(value, str):
756                raise TypeError("str expected, not %s" % type(value).__name__)
757            return value.encode(encoding, 'surrogateescape')
758        def decode(value):
759            return value.decode(encoding, 'surrogateescape')
760        encodekey = encode
761        data = environ
762    return _Environ(data,
763        encodekey, decode,
764        encode, decode)
765
766# unicode environ
767environ = _createenviron()
768del _createenviron
769
770
771def getenv(key, default=None):
772    """Get an environment variable, return None if it doesn't exist.
773    The optional second argument can specify an alternate default.
774    key, default and the result are str."""
775    return environ.get(key, default)
776
777supports_bytes_environ = (name != 'nt')
778__all__.extend(("getenv", "supports_bytes_environ"))
779
780if supports_bytes_environ:
781    def _check_bytes(value):
782        if not isinstance(value, bytes):
783            raise TypeError("bytes expected, not %s" % type(value).__name__)
784        return value
785
786    # bytes environ
787    environb = _Environ(environ._data,
788        _check_bytes, bytes,
789        _check_bytes, bytes)
790    del _check_bytes
791
792    def getenvb(key, default=None):
793        """Get an environment variable, return None if it doesn't exist.
794        The optional second argument can specify an alternate default.
795        key, default and the result are bytes."""
796        return environb.get(key, default)
797
798    __all__.extend(("environb", "getenvb"))
799
800def _fscodec():
801    encoding = sys.getfilesystemencoding()
802    errors = sys.getfilesystemencodeerrors()
803
804    def fsencode(filename):
805        """Encode filename (an os.PathLike, bytes, or str) to the filesystem
806        encoding with 'surrogateescape' error handler, return bytes unchanged.
807        On Windows, use 'strict' error handler if the file system encoding is
808        'mbcs' (which is the default encoding).
809        """
810        filename = fspath(filename)  # Does type-checking of `filename`.
811        if isinstance(filename, str):
812            return filename.encode(encoding, errors)
813        else:
814            return filename
815
816    def fsdecode(filename):
817        """Decode filename (an os.PathLike, bytes, or str) from the filesystem
818        encoding with 'surrogateescape' error handler, return str unchanged. On
819        Windows, use 'strict' error handler if the file system encoding is
820        'mbcs' (which is the default encoding).
821        """
822        filename = fspath(filename)  # Does type-checking of `filename`.
823        if isinstance(filename, bytes):
824            return filename.decode(encoding, errors)
825        else:
826            return filename
827
828    return fsencode, fsdecode
829
830fsencode, fsdecode = _fscodec()
831del _fscodec
832
833# Supply spawn*() (probably only for Unix)
834if _exists("fork") and not _exists("spawnv") and _exists("execv"):
835
836    P_WAIT = 0
837    P_NOWAIT = P_NOWAITO = 1
838
839    __all__.extend(["P_WAIT", "P_NOWAIT", "P_NOWAITO"])
840
841    # XXX Should we support P_DETACH?  I suppose it could fork()**2
842    # and close the std I/O streams.  Also, P_OVERLAY is the same
843    # as execv*()?
844
845    def _spawnvef(mode, file, args, env, func):
846        # Internal helper; func is the exec*() function to use
847        if not isinstance(args, (tuple, list)):
848            raise TypeError('argv must be a tuple or a list')
849        if not args or not args[0]:
850            raise ValueError('argv first element cannot be empty')
851        pid = fork()
852        if not pid:
853            # Child
854            try:
855                if env is None:
856                    func(file, args)
857                else:
858                    func(file, args, env)
859            except:
860                _exit(127)
861        else:
862            # Parent
863            if mode == P_NOWAIT:
864                return pid # Caller is responsible for waiting!
865            while 1:
866                wpid, sts = waitpid(pid, 0)
867                if WIFSTOPPED(sts):
868                    continue
869
870                return waitstatus_to_exitcode(sts)
871
872    def spawnv(mode, file, args):
873        """spawnv(mode, file, args) -> integer
874
875Execute file with arguments from args in a subprocess.
876If mode == P_NOWAIT return the pid of the process.
877If mode == P_WAIT return the process's exit code if it exits normally;
878otherwise return -SIG, where SIG is the signal that killed it. """
879        return _spawnvef(mode, file, args, None, execv)
880
881    def spawnve(mode, file, args, env):
882        """spawnve(mode, file, args, env) -> integer
883
884Execute file with arguments from args in a subprocess with the
885specified environment.
886If mode == P_NOWAIT return the pid of the process.
887If mode == P_WAIT return the process's exit code if it exits normally;
888otherwise return -SIG, where SIG is the signal that killed it. """
889        return _spawnvef(mode, file, args, env, execve)
890
891    # Note: spawnvp[e] isn't currently supported on Windows
892
893    def spawnvp(mode, file, args):
894        """spawnvp(mode, file, args) -> integer
895
896Execute file (which is looked for along $PATH) with arguments from
897args in a subprocess.
898If mode == P_NOWAIT return the pid of the process.
899If mode == P_WAIT return the process's exit code if it exits normally;
900otherwise return -SIG, where SIG is the signal that killed it. """
901        return _spawnvef(mode, file, args, None, execvp)
902
903    def spawnvpe(mode, file, args, env):
904        """spawnvpe(mode, file, args, env) -> integer
905
906Execute file (which is looked for along $PATH) with arguments from
907args in a subprocess with the supplied environment.
908If mode == P_NOWAIT return the pid of the process.
909If mode == P_WAIT return the process's exit code if it exits normally;
910otherwise return -SIG, where SIG is the signal that killed it. """
911        return _spawnvef(mode, file, args, env, execvpe)
912
913
914    __all__.extend(["spawnv", "spawnve", "spawnvp", "spawnvpe"])
915
916
917if _exists("spawnv"):
918    # These aren't supplied by the basic Windows code
919    # but can be easily implemented in Python
920
921    def spawnl(mode, file, *args):
922        """spawnl(mode, file, *args) -> integer
923
924Execute file with arguments from args in a subprocess.
925If mode == P_NOWAIT return the pid of the process.
926If mode == P_WAIT return the process's exit code if it exits normally;
927otherwise return -SIG, where SIG is the signal that killed it. """
928        return spawnv(mode, file, args)
929
930    def spawnle(mode, file, *args):
931        """spawnle(mode, file, *args, env) -> integer
932
933Execute file with arguments from args in a subprocess with the
934supplied environment.
935If mode == P_NOWAIT return the pid of the process.
936If mode == P_WAIT return the process's exit code if it exits normally;
937otherwise return -SIG, where SIG is the signal that killed it. """
938        env = args[-1]
939        return spawnve(mode, file, args[:-1], env)
940
941
942    __all__.extend(["spawnl", "spawnle"])
943
944
945if _exists("spawnvp"):
946    # At the moment, Windows doesn't implement spawnvp[e],
947    # so it won't have spawnlp[e] either.
948    def spawnlp(mode, file, *args):
949        """spawnlp(mode, file, *args) -> integer
950
951Execute file (which is looked for along $PATH) with arguments from
952args in a subprocess with the supplied environment.
953If mode == P_NOWAIT return the pid of the process.
954If mode == P_WAIT return the process's exit code if it exits normally;
955otherwise return -SIG, where SIG is the signal that killed it. """
956        return spawnvp(mode, file, args)
957
958    def spawnlpe(mode, file, *args):
959        """spawnlpe(mode, file, *args, env) -> integer
960
961Execute file (which is looked for along $PATH) with arguments from
962args in a subprocess with the supplied environment.
963If mode == P_NOWAIT return the pid of the process.
964If mode == P_WAIT return the process's exit code if it exits normally;
965otherwise return -SIG, where SIG is the signal that killed it. """
966        env = args[-1]
967        return spawnvpe(mode, file, args[:-1], env)
968
969
970    __all__.extend(["spawnlp", "spawnlpe"])
971
972
973# Supply os.popen()
974def popen(cmd, mode="r", buffering=-1):
975    if not isinstance(cmd, str):
976        raise TypeError("invalid cmd type (%s, expected string)" % type(cmd))
977    if mode not in ("r", "w"):
978        raise ValueError("invalid mode %r" % mode)
979    if buffering == 0 or buffering is None:
980        raise ValueError("popen() does not support unbuffered streams")
981    import subprocess, io
982    if mode == "r":
983        proc = subprocess.Popen(cmd,
984                                shell=True,
985                                stdout=subprocess.PIPE,
986                                bufsize=buffering)
987        return _wrap_close(io.TextIOWrapper(proc.stdout), proc)
988    else:
989        proc = subprocess.Popen(cmd,
990                                shell=True,
991                                stdin=subprocess.PIPE,
992                                bufsize=buffering)
993        return _wrap_close(io.TextIOWrapper(proc.stdin), proc)
994
995# Helper for popen() -- a proxy for a file whose close waits for the process
996class _wrap_close:
997    def __init__(self, stream, proc):
998        self._stream = stream
999        self._proc = proc
1000    def close(self):
1001        self._stream.close()
1002        returncode = self._proc.wait()
1003        if returncode == 0:
1004            return None
1005        if name == 'nt':
1006            return returncode
1007        else:
1008            return returncode << 8  # Shift left to match old behavior
1009    def __enter__(self):
1010        return self
1011    def __exit__(self, *args):
1012        self.close()
1013    def __getattr__(self, name):
1014        return getattr(self._stream, name)
1015    def __iter__(self):
1016        return iter(self._stream)
1017
1018# Supply os.fdopen()
1019def fdopen(fd, *args, **kwargs):
1020    if not isinstance(fd, int):
1021        raise TypeError("invalid fd type (%s, expected integer)" % type(fd))
1022    import io
1023    return io.open(fd, *args, **kwargs)
1024
1025
1026# For testing purposes, make sure the function is available when the C
1027# implementation exists.
1028def _fspath(path):
1029    """Return the path representation of a path-like object.
1030
1031    If str or bytes is passed in, it is returned unchanged. Otherwise the
1032    os.PathLike interface is used to get the path representation. If the
1033    path representation is not str or bytes, TypeError is raised. If the
1034    provided path is not str, bytes, or os.PathLike, TypeError is raised.
1035    """
1036    if isinstance(path, (str, bytes)):
1037        return path
1038
1039    # Work from the object's type to match method resolution of other magic
1040    # methods.
1041    path_type = type(path)
1042    try:
1043        path_repr = path_type.__fspath__(path)
1044    except AttributeError:
1045        if hasattr(path_type, '__fspath__'):
1046            raise
1047        else:
1048            raise TypeError("expected str, bytes or os.PathLike object, "
1049                            "not " + path_type.__name__)
1050    if isinstance(path_repr, (str, bytes)):
1051        return path_repr
1052    else:
1053        raise TypeError("expected {}.__fspath__() to return str or bytes, "
1054                        "not {}".format(path_type.__name__,
1055                                        type(path_repr).__name__))
1056
1057# If there is no C implementation, make the pure Python version the
1058# implementation as transparently as possible.
1059if not _exists('fspath'):
1060    fspath = _fspath
1061    fspath.__name__ = "fspath"
1062
1063
1064class PathLike(abc.ABC):
1065
1066    """Abstract base class for implementing the file system path protocol."""
1067
1068    @abc.abstractmethod
1069    def __fspath__(self):
1070        """Return the file system path representation of the object."""
1071        raise NotImplementedError
1072
1073    @classmethod
1074    def __subclasshook__(cls, subclass):
1075        if cls is PathLike:
1076            return _check_methods(subclass, '__fspath__')
1077        return NotImplemented
1078
1079    __class_getitem__ = classmethod(GenericAlias)
1080
1081
1082if name == 'nt':
1083    class _AddedDllDirectory:
1084        def __init__(self, path, cookie, remove_dll_directory):
1085            self.path = path
1086            self._cookie = cookie
1087            self._remove_dll_directory = remove_dll_directory
1088        def close(self):
1089            self._remove_dll_directory(self._cookie)
1090            self.path = None
1091        def __enter__(self):
1092            return self
1093        def __exit__(self, *args):
1094            self.close()
1095        def __repr__(self):
1096            if self.path:
1097                return "<AddedDllDirectory({!r})>".format(self.path)
1098            return "<AddedDllDirectory()>"
1099
1100    def add_dll_directory(path):
1101        """Add a path to the DLL search path.
1102
1103        This search path is used when resolving dependencies for imported
1104        extension modules (the module itself is resolved through sys.path),
1105        and also by ctypes.
1106
1107        Remove the directory by calling close() on the returned object or
1108        using it in a with statement.
1109        """
1110        import nt
1111        cookie = nt._add_dll_directory(path)
1112        return _AddedDllDirectory(
1113            path,
1114            cookie,
1115            nt._remove_dll_directory
1116        )
1117