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
|
from __future__ import annotations
import sys
import warnings
from typing import TYPE_CHECKING
from typing import Any
from upath.core import UPath
from upath.types import UNSET_DEFAULT
from upath.types import JoinablePathLike
from upath.types import WritablePathLike
if TYPE_CHECKING:
from typing import Literal
if sys.version_info >= (3, 11):
from typing import Self
from typing import Unpack
else:
from typing_extensions import Self
from typing_extensions import Unpack
from upath._chain import FSSpecChainParser
from upath.types.storage_options import SMBStorageOptions
class SMBPath(UPath):
__slots__ = ()
if TYPE_CHECKING:
def __init__(
self,
*args: JoinablePathLike,
protocol: Literal["smb"] | None = ...,
chain_parser: FSSpecChainParser = ...,
**storage_options: Unpack[SMBStorageOptions],
) -> None: ...
@property
def path(self) -> str:
path = super().path
if len(path) > 1:
return path.removesuffix("/")
# At root level, return "/" to match anchor
if not path and self._relative_base is None:
return self.anchor
return path
def __str__(self) -> str:
path_str = super().__str__()
if path_str.startswith("smb:///"):
return path_str.removesuffix("/")
return path_str
def mkdir(
self,
mode: int = 0o777,
parents: bool = False,
exist_ok: bool = False,
) -> None:
# smbclient does not support setting mode externally
from smbprotocol.exceptions import SMBOSError
if parents and not exist_ok and self.exists():
raise FileExistsError(str(self))
try:
self.fs.mkdir(
self.path,
create_parents=parents,
)
except SMBOSError:
if not exist_ok:
raise FileExistsError(str(self))
if not self.is_dir():
raise FileExistsError(str(self))
def rename(
self,
target: WritablePathLike,
*,
recursive: bool = UNSET_DEFAULT,
maxdepth: int | None = UNSET_DEFAULT,
**kwargs: Any,
) -> Self:
if recursive is not UNSET_DEFAULT:
warnings.warn(
"SMBPath.rename(): recursive is currently ignored.",
UserWarning,
stacklevel=2,
)
if maxdepth is not UNSET_DEFAULT:
warnings.warn(
"SMBPath.rename(): maxdepth is currently ignored.",
UserWarning,
stacklevel=2,
)
return super().rename(target, **kwargs)
|
