diff options
Diffstat (limited to 'linuxnamespaces')
| -rw-r--r-- | linuxnamespaces/atlocation.py | 25 |
1 files changed, 22 insertions, 3 deletions
diff --git a/linuxnamespaces/atlocation.py b/linuxnamespaces/atlocation.py index be9d17c..6b28ec3 100644 --- a/linuxnamespaces/atlocation.py +++ b/linuxnamespaces/atlocation.py @@ -341,14 +341,33 @@ class AtLocation: ), ) - def mkdir(self, mode: int = 0o777) -> None: - """Wrapper for os.mkdir supplying path and dir_fd.""" + def mkdir( + self, mode: int = 0o777, parents: bool = False, exist_ok: bool = False + ) -> None: + """Wrapper for os.mkdir supplying path and dir_fd. It also supports + the parents and exist_ok arguments from pathlib.Path.mkdir. + """ if self.flags != AtFlags.NONE: raise NotImplementedError( "mkdir is not supported for an AtLocation with flags" ) assert self.location - os.mkdir(self.location, mode, dir_fd=self.fd_or_none) + try: + os.mkdir(self.location, mode, dir_fd=self.fd_or_none) + except FileNotFoundError: + if not parents: + raise + parentlocation = os.path.dirname(self.location) + if not parentlocation: + raise + AtLocation(self.fd, parentlocation, self.flags).mkdir( + parents=True, exist_ok=True + ) + self.mkdir(mode, False, exist_ok) + except OSError: + # Like pathlib, avoid checking EEXISTS as there may be more reasons + if not exist_ok or not self.is_dir(): + raise def mknod(self, mode: int = 0o600, device: int = 0) -> None: """Wrapper for os.mknod supplying path and dir_fd.""" |