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
|
from __future__ import with_statement
from plumbum.path.base import Path
from plumbum.lib import six
from plumbum.machines.local import local, 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, six.string_types):
local.path(p).delete()
elif hasattr(p, "__iter__"):
delete(*p)
else:
raise TypeError("Cannot delete %r" % (p,))
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.isdir():
raise ValueError("When using multiple sources, dst %r must be a directory" % (dst,))
for src2 in src:
move(src2, dst)
return dst
elif not isinstance(src, Path):
src = local.path(src)
if isinstance(src, LocalPath):
if isinstance(dst, LocalPath):
return src.move(dst)
else:
return _move(src, dst)
elif isinstance(dst, LocalPath):
return _move(src, dst)
elif src.remote == dst.remote:
return src.move(dst)
else:
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.isdir():
raise ValueError("When using multiple sources, dst %r must be a directory" % (dst,))
for src2 in src:
copy(src2, dst)
return dst
elif not isinstance(src, Path):
src = local.path(src)
if isinstance(src, LocalPath):
if isinstance(dst, LocalPath):
return src.copy(dst)
else:
dst.remote.upload(src, dst)
return dst
elif isinstance(dst, LocalPath):
src.remote.download(src, dst)
return dst
elif src.remote == dst.remote:
return src.copy(dst)
else:
with local.tempdir() as tmp:
copy(src, tmp)
copy(tmp / src.basename, dst)
return dst
|
