return os.terminal_size((columns, lines))
-# Check that a given file can be accessed with the correct mode.
-# Additionally check that `file` is not a directory, as on Windows
-# directories pass the os.access check.
-def _access_check(fn, mode):
- return (os.path.exists(fn) and os.access(fn, mode)
- and not os.path.isdir(fn))
-
def which(cmd, mode=os.F_OK | os.X_OK, path=None):
"""Given a command, mode, and a PATH string, return the path which
conforms to the given mode on the PATH, or None if there is no such
path.
"""
+ # Check that a given file can be accessed with the correct mode.
+ # Additionally check that `file` is not a directory, as on Windows
+ # directories pass the os.access check.
+ def _access_check(fn, mode):
+ return (os.path.exists(fn) and os.access(fn, mode)
+ and not os.path.isdir(fn))
+
# If we're given a path with a directory part, look it up directly rather
# than referring to PATH directories. This includes checking relative to the
# current directory, e.g. ./script
path = os.environ.get("PATH", os.defpath)
if not path:
return None
- if isinstance(cmd, bytes):
- path = os.fsencode(path)
- path = path.split(os.fsencode(os.pathsep))
- else:
- path = os.fsdecode(path)
- path = path.split(os.pathsep)
+ path = path.split(os.pathsep)
if sys.platform == "win32":
# The current directory takes precedence on Windows.
# PATHEXT is necessary to check on Windows.
pathext = os.environ.get("PATHEXT", "").split(os.pathsep)
- if isinstance(cmd, bytes):
- pathext = map(os.fsencode, pathext)
# See if the given file matches any of the expected path extensions.
# This will allow us to short circuit when given "python.exe".
# If it does match, only test that one, otherwise we have to try
os.chmod(self.temp_file.name, stat.S_IXUSR)
self.addCleanup(self.temp_file.close)
self.dir, self.file = os.path.split(self.temp_file.name)
- self.env_path = self.dir
def test_basic(self):
# Given an EXE in a directory, it should be returned.
def test_environ_path(self):
with support.EnvironmentVarGuard() as env:
- env['PATH'] = self.env_path
+ env['PATH'] = self.dir
rv = shutil.which(self.file)
self.assertEqual(rv, self.temp_file.name)
base_dir = os.path.dirname(self.dir)
with support.change_cwd(path=self.dir), \
support.EnvironmentVarGuard() as env:
- env['PATH'] = self.env_path
+ env['PATH'] = self.dir
rv = shutil.which(self.file, path='')
self.assertIsNone(rv)
self.assertIsNone(rv)
-class TestWhichBytes(TestWhich):
- def setUp(self):
- TestWhich.setUp(self)
- self.dir = os.fsencode(self.dir)
- self.file = os.fsencode(self.file)
- self.temp_file.name = os.fsencode(self.temp_file.name)
-
-
class TestMove(unittest.TestCase):
def setUp(self):