1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
|
from __future__ import annotations
import os
from plumbum.machines.local import local
from plumbum.path.base import Path
from plumbum.path.local import LocalPath
def delete(*paths):
"""Deletes the given paths. The arguments can be either strings,
:class:`local paths <plumbum.path.local.LocalPath>`,
:class:`remote paths <plumbum.path.remote.RemotePath>`, or iterables of such.
No error is raised if any of the paths does not exist (it is silently ignored)
"""
for p in paths:
if isinstance(p, Path):
p.delete()
elif isinstance(p, str):
local.path(p).delete()
elif hasattr(p, "__iter__"):
delete(*p)
else:
raise TypeError(f"Cannot delete {p!r}")
def _move(src, dst):
ret = copy(src, dst)
delete(src)
return ret
def move(src, dst):
"""Moves the source path onto the destination path; ``src`` and ``dst`` can be either
strings, :class:`LocalPaths <plumbum.path.local.LocalPath>` or
:class:`RemotePath <plumbum.path.remote.RemotePath>`; any combination of the three will
work.
.. versionadded:: 1.3
``src`` can also be a list of strings/paths, in which case ``dst`` must not exist or be a directory.
"""
if not isinstance(dst, Path):
dst = local.path(dst)
if isinstance(src, (tuple, list)):
if not dst.exists():
dst.mkdir()
elif not dst.is_dir():
raise ValueError(
f"When using multiple sources, dst {dst!r} must be a directory"
)
for src2 in src:
move(src2, dst)
return dst
if not isinstance(src, Path):
src = local.path(src)
if isinstance(src, LocalPath):
return src.move(dst) if isinstance(dst, LocalPath) else _move(src, dst)
if isinstance(dst, LocalPath):
return _move(src, dst)
if src.remote == dst.remote:
return src.move(dst)
return _move(src, dst)
def copy(src, dst):
"""
Copy (recursively) the source path onto the destination path; ``src`` and ``dst`` can be
either strings, :class:`LocalPaths <plumbum.path.local.LocalPath>` or
:class:`RemotePath <plumbum.path.remote.RemotePath>`; any combination of the three will
work.
.. versionadded:: 1.3
``src`` can also be a list of strings/paths, in which case ``dst`` must not exist or be a directory.
"""
if not isinstance(dst, Path):
dst = local.path(dst)
if isinstance(src, (tuple, list)):
if not dst.exists():
dst.mkdir()
elif not dst.is_dir():
raise ValueError(
f"When using multiple sources, dst {dst!r} must be a directory"
)
for src2 in src:
copy(src2, dst)
return dst
if not isinstance(src, Path):
src = local.path(src)
if isinstance(src, LocalPath):
if isinstance(dst, LocalPath):
return src.copy(dst)
dst.remote.upload(src, dst)
return dst
if isinstance(dst, LocalPath):
src.remote.download(src, dst)
return dst
if src.remote == dst.remote:
return src.copy(dst)
with local.tempdir() as tmp:
copy(src, tmp)
copy(tmp / src.name, dst)
return dst
def gui_open(filename):
"""This selects the proper gui open function. This can
also be achieved with webbrowser, but that is not supported."""
if hasattr(os, "startfile"):
os.startfile(filename)
else:
local.get("xdg-open", "open")(filename)
|