ubelt.util_path module¶
Functions for working with filesystem paths.
The expandpath()
function expands the tilde to $HOME and environment
variables to their values.
The augpath()
function creates variants of an existing path without
having to spend multiple lines of code splitting it up and stitching it back
together.
The shrinkuser()
function replaces your home directory with a tilde.
The userhome()
function reports the home directory of the current user of
the operating system.
The ensuredir()
function operates like mkdir -p
in unix.
The Path
object is an extension of pathlib.Path
that contains
extra convinience methods corresponding to the extra functional methods in this
module.
- class ubelt.util_path.Path(*args, **kwargs)[source]¶
Bases:
pathlib.PosixPath
An extension of
pathlib.Path
with extra convinience methods- ensuredir(mode=511)[source]¶
Concise alias of
self.mkdir(parents=True, exist_ok=True)
- Returns
returns itself
- Return type
Example
>>> import ubelt as ub >>> cache_dpath = ub.ensure_app_cache_dir('ubelt') >>> dpath = ub.Path(join(cache_dpath, 'ensuredir')) >>> if dpath.exists(): ... os.rmdir(dpath) >>> assert not dpath.exists() >>> dpath.ensuredir() >>> assert dpath.exists() >>> dpath.rmdir()
- expandvars()[source]¶
As discussed in CPythonIssue21301, CPython wont be adding expandvars to pathlib. I think this is a mistake, so I added it in this extension.
- Returns
path with expanded environment variables
- Return type
References
- CPythonIssue21301
- expand()[source]¶
Expands user tilde and environment variables.
Concise alias of Path(os.path.expandvars(self.expanduser()))
- Returns
path with expanded environment variables and tildes
- Return type
Example
>>> import ubelt as ub >>> #home_v1 = ub.Path('$HOME').expand() >>> home_v2 = ub.Path('~/').expand() >>> assert isinstance(home_v2, ub.Path) >>> # xdoctest: +REQUIRES(PY3) >>> home_v3 = ub.Path.home() >>> #print('home_v1 = {!r}'.format(home_v1)) >>> print('home_v2 = {!r}'.format(home_v2)) >>> print('home_v3 = {!r}'.format(home_v3)) >>> assert home_v3 == home_v2 # == home_v1
- shrinkuser(home='~')[source]¶
Inverse of
os.path.expanduser()
.- Parameters
home (str, default=’~’) – symbol used to replace the home path. Defaults to ‘~’, but you might want to use ‘$HOME’ or ‘%USERPROFILE%’ instead.
- Returns
path - shortened path replacing the home directory with a tilde
- Return type
Example
>>> import ubelt as ub >>> path = ub.Path('~').expand() >>> assert str(path.shrinkuser()) == '~' >>> assert str(ub.Path((str(path) + '1')).shrinkuser()) == str(path) + '1' >>> assert str((path / '1').shrinkuser()) == join('~', '1') >>> assert str((path / '1').shrinkuser('$HOME')) == join('$HOME', '1') >>> assert str(ub.Path('.').shrinkuser()) == '.'
- augment(suffix='', prefix='', ext=None, stem=None, dpath=None, relative=None, multidot=False)[source]¶
Create a new path with a different extension, basename, directory, prefix, and/or suffix.
See
augpath()
for more details.- Parameters
suffix (str, default=’’) – placed between the stem and extension
prefix (str, default=’’) – placed in front of the stem
ext (str | None, default=None) – if specified, replaces the extension
stem (str | None, default=None) – if specified, replaces the stem (i.e. basename without extension). Note: named base in
augpath()
.dpath (str | PathLike | None, default=None) – if specified, replaces the specified “relative” directory, which by default is the parent directory.
relative (str | PathLike | None, default=None) – Replaces
relative
withdpath
inpath
. Has no effect ifdpath
is not specified. Defaults to the dirname of the inputpath
. experimental not currently implemented.multidot (bool, default=False) – Allows extensions to contain multiple dots. Specifically, if False, everything after the last dot in the basename is the extension. If True, everything after the first dot in the basename is the extension.
- Returns
augmented path
- Return type
Example
>>> import ubelt as ub >>> path = ub.Path('foo.bar') >>> suffix = '_suff' >>> prefix = 'pref_' >>> ext = '.baz' >>> newpath = path.augment(suffix, prefix, ext=ext, stem='bar') >>> print('newpath = {!r}'.format(newpath)) newpath = Path('pref_bar_suff.baz')
- class ubelt.util_path.TempDir[source]¶
Bases:
object
Context for creating and cleaning up temporary directories.
Note
This exists because
tempfile.TemporaryDirectory
was introduced in Python 3.2. Thus once ubelt no longer supports python 2.7, this class will be deprecated.Example
>>> from ubelt.util_path import * # NOQA >>> with TempDir() as self: >>> dpath = self.dpath >>> assert exists(dpath) >>> assert not exists(dpath)
Example
>>> from ubelt.util_path import * # NOQA >>> self = TempDir() >>> dpath = self.ensure() >>> assert exists(dpath) >>> self.cleanup() >>> assert not exists(dpath)
- ubelt.util_path.augpath(path, suffix='', prefix='', ext=None, base=None, dpath=None, relative=None, multidot=False)[source]¶
Create a new path with a different extension, basename, directory, prefix, and/or suffix.
A prefix is inserted before the basename. A suffix is inserted between the basename and the extension. The basename and extension can be replaced with a new one. Essentially a path is broken down into components (dpath, base, ext), and then recombined as (dpath, prefix, base, suffix, ext) after replacing any specified component.
- Parameters
path (str | PathLike) – a path to augment
suffix (str, default=’’) – placed between the basename and extension
prefix (str, default=’’) – placed in front of the basename
ext (str | None, default=None) – if specified, replaces the extension
base (str | None, default=None) – if specified, replaces the basename without extension. Note: this is refered to as stem in
ub.Path
.dpath (str | PathLike | None, default=None) – if specified, replaces the specified “relative” directory, which by default is the parent directory.
relative (str | PathLike | None, default=None) – Replaces
relative
withdpath
inpath
. Has no effect ifdpath
is not specified. Defaults to the dirname of the inputpath
. experimental not currently implemented.multidot (bool, default=False) – Allows extensions to contain multiple dots. Specifically, if False, everything after the last dot in the basename is the extension. If True, everything after the first dot in the basename is the extension.
- Returns
augmented path
- Return type
Example
>>> import ubelt as ub >>> path = 'foo.bar' >>> suffix = '_suff' >>> prefix = 'pref_' >>> ext = '.baz' >>> newpath = ub.augpath(path, suffix, prefix, ext=ext, base='bar') >>> print('newpath = %s' % (newpath,)) newpath = pref_bar_suff.baz
Example
>>> from ubelt.util_path import * # NOQA >>> augpath('foo.bar') 'foo.bar' >>> augpath('foo.bar', ext='.BAZ') 'foo.BAZ' >>> augpath('foo.bar', suffix='_') 'foo_.bar' >>> augpath('foo.bar', prefix='_') '_foo.bar' >>> augpath('foo.bar', base='baz') 'baz.bar' >>> augpath('foo.tar.gz', ext='.zip', multidot=True) foo.zip >>> augpath('foo.tar.gz', ext='.zip', multidot=False) foo.tar.zip >>> augpath('foo.tar.gz', suffix='_new', multidot=True) foo_new.tar.gz
- ubelt.util_path.shrinkuser(path, home='~')[source]¶
Inverse of
os.path.expanduser()
.- Parameters
path (str | PathLike) – path in system file structure
home (str, default=’~’) – symbol used to replace the home path. Defaults to ‘~’, but you might want to use ‘$HOME’ or ‘%USERPROFILE%’ instead.
- Returns
path - shortened path replacing the home directory with a tilde
- Return type
Example
>>> from ubelt.util_path import * # NOQA >>> path = expanduser('~') >>> assert path != '~' >>> assert shrinkuser(path) == '~' >>> assert shrinkuser(path + '1') == path + '1' >>> assert shrinkuser(path + '/1') == join('~', '1') >>> assert shrinkuser(path + '/1', '$HOME') == join('$HOME', '1') >>> assert shrinkuser('.') == '.'
- ubelt.util_path.userhome(username=None)[source]¶
Returns the path to some user’s home directory.
- Parameters
username (str | None, default=None) – name of a user on the system. If not specified, the current user is inferred.
- Returns
userhome_dpath - path to the specified home directory
- Return type
- Raises
Example
>>> from ubelt.util_path import * # NOQA >>> import getpass >>> username = getpass.getuser() >>> assert userhome() == expanduser('~') >>> assert userhome(username) == expanduser('~')
- ubelt.util_path.ensuredir(dpath, mode=1023, verbose=0, recreate=False)[source]¶
Ensures that directory will exist. Creates new dir with sticky bits by default
- Parameters
dpath (str | PathLike | Tuple[str | PathLike]) – dir to ensure. Can also be a tuple to send to join
mode (int, default=0o1777) – octal mode of directory
verbose (int, default=0) – verbosity
recreate (bool, default=False) – if True removes the directory and all of its contents and creates a fresh new directory. USE CAREFULLY.
- Returns
path - the ensured directory
- Return type
- SeeAlso:
Note
This function is not thread-safe in Python2
Example
>>> from ubelt.util_path import * # NOQA >>> import ubelt as ub >>> cache_dpath = ub.ensure_app_cache_dir('ubelt') >>> dpath = join(cache_dpath, 'ensuredir') >>> if exists(dpath): ... os.rmdir(dpath) >>> assert not exists(dpath) >>> ub.ensuredir(dpath) >>> assert exists(dpath) >>> os.rmdir(dpath)
- ubelt.util_path.expandpath(path)[source]¶
Shell-like environment variable and tilde path expansion.
Less aggressive than truepath. Only expands environs and tilde. Does not change relative paths to absolute paths.
- Parameters
path (str | PathLike) – string representation of a path
- Returns
expanded path
- Return type
Example
>>> from ubelt.util_path import * # NOQA >>> import ubelt as ub >>> assert normpath(ub.expandpath('~/foo')) == join(ub.userhome(), 'foo') >>> assert ub.expandpath('foo') == 'foo'