python-fsutil 0.11.0__py3-none-any.whl → 0.13.0__py3-none-any.whl

fsutil/__init__.py

@@ -80,6 +80,7 @@ __all__ = [
     "get_file_size_formatted",
     "get_filename",
     "get_parent_dir",
+    "get_permissions",
     "get_unique_name",
     "is_dir",
     "is_empty",
@@ -113,6 +114,7 @@ __all__ = [
     "replace_file",
     "search_dirs",
     "search_files",
+    "set_permissions",
     "split_filename",
     "split_filepath",
     "split_path",
@@ -788,6 +790,17 @@ def get_parent_dir(path: PathIn, *, levels: int = 1) -> str:
     return join_path(path, *([os.pardir] * max(1, levels)))
 
 
+def get_permissions(path: PathIn) -> int:
+    """
+    Get the file/directory permissions.
+    """
+    path = _get_path(path)
+    assert_exists(path)
+    st_mode = os.stat(path).st_mode
+    permissions = int(str(oct(st_mode & 0o777))[2:])
+    return permissions
+
+
 def get_unique_name(
     path: PathIn,
     *,
@@ -797,7 +810,7 @@ def get_unique_name(
     separator: str = "-",
 ) -> str:
     """
-    Gets a unique name for a directory/file ath the given directory path.
+    Get a unique name for a directory/file ath the given directory path.
     """
     path = _get_path(path)
     assert_dir(path)
@@ -1282,6 +1295,16 @@ def search_files(path: PathIn, pattern: str = "**/*.*") -> list[str]:
     return _filter_paths(path, _search_paths(path, pattern), predicate=is_file)
 
 
+def set_permissions(path: PathIn, value: int) -> None:
+    """
+    Set the file/directory permissions.
+    """
+    path = _get_path(path)
+    assert_exists(path)
+    permissions = int(str(value), 8) & 0o777
+    os.chmod(path, permissions)
+
+
 def split_filename(path: PathIn) -> tuple[str, str]:
     """
     Split a filename and returns its basename and extension.
@@ -1314,12 +1337,58 @@ def split_path(path: PathIn) -> list[str]:
     return names
 
 
+def _write_file_atomic(
+    path: PathIn,
+    content: str,
+    *,
+    append: bool = False,
+    encoding: str = "utf-8",
+) -> None:
+    path = _get_path(path)
+    mode = "a" if append else "w"
+    if append:
+        content = read_file(path, encoding=encoding) + content
+    dirpath, _ = split_filepath(path)
+    try:
+        with tempfile.NamedTemporaryFile(
+            mode=mode,
+            dir=dirpath,
+            delete=True,
+            encoding=encoding,
+        ) as file:
+            file.write(content)
+            file.flush()
+            os.fsync(file.fileno())
+            temp_path = file.name
+            if exists(path):
+                set_permissions(temp_path, get_permissions(path))
+            os.replace(temp_path, path)
+    except FileNotFoundError:
+        # success - the NamedTemporaryFile has not been able
+        # to remove the temp file on __exit__ because the temp file
+        # has replaced atomically the file at path.
+        pass
+
+
+def _write_file_non_atomic(
+    path: PathIn,
+    content: str,
+    *,
+    append: bool = False,
+    encoding: str = "utf-8",
+) -> None:
+    mode = "a" if append else "w"
+    with open(path, mode, encoding=encoding) as file:
+        file.write(content)
+
+
 def write_file(
     path: PathIn,
     content: str,
     *,
     append: bool = False,
     encoding: str = "utf-8",
+    atomic: bool = False,
 ) -> None:
     """
     Write file with the specified content at the given path.
@@ -1327,14 +1396,20 @@ def write_file(
     path = _get_path(path)
     assert_not_dir(path)
     make_dirs_for_file(path)
-    mode = "a" if append else "w"
-    with open(path, mode, encoding=encoding) as file:
-        file.write(content)
+    write_file_func = _write_file_atomic if atomic else _write_file_non_atomic
+    write_file_func(
+        path,
+        content,
+        append=append,
+        encoding=encoding,
+    )
 
 
 def write_file_json(
     path: PathIn,
     data: Any,
+    encoding: str = "utf-8",
+    atomic: bool = False,
     **kwargs: Any,
 ) -> None:
     """
@@ -1351,4 +1426,10 @@ def write_file_json(
 
     kwargs.setdefault("default", default_encoder)
     content = json.dumps(data, **kwargs)
-    write_file(path, content)
+    write_file(
+        path,
+        content,
+        append=False,
+        encoding=encoding,
+        atomic=atomic,
+    )

fsutil/metadata.py

@@ -4,4 +4,4 @@ __description__ = "high-level file-system operations for lazy devs."
 __email__ = "fabio.caccamo@gmail.com"
 __license__ = "MIT"
 __title__ = "python-fsutil"
-__version__ = "0.11.0"
+__version__ = "0.13.0"

{python_fsutil-0.11.0.dist-info → python_fsutil-0.13.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-fsutil
-Version: 0.11.0
+Version: 0.13.0
 Summary: high-level file-system operations for lazy devs.
 Author-email: Fabio Caccamo <fabio.caccamo@gmail.com>
 Maintainer-email: Fabio Caccamo <fabio.caccamo@gmail.com>
@@ -132,6 +132,7 @@ import fsutil
 -   [`get_file_size_formatted`](#get_file_size_formatted)
 -   [`get_filename`](#get_filename)
 -   [`get_parent_dir`](#get_parent_dir)
+-   [`get_permissions`](#get_permissions)
 -   [`get_unique_name`](#get_unique_name)
 -   [`is_dir`](#is_dir)
 -   [`is_empty`](#is_empty)
@@ -165,6 +166,7 @@ import fsutil
 -   [`replace_file`](#replace_file)
 -   [`search_dirs`](#search_dirs)
 -   [`search_files`](#search_files)
+-   [`set_permissions`](#set_permissions)
 -   [`split_filename`](#split_filename)
 -   [`split_filepath`](#split_filepath)
 -   [`split_path`](#split_path)
@@ -494,10 +496,17 @@ filename = fsutil.get_filename(path)
 parent_dir = fsutil.get_parent_dir(path, levels=1)
 ```
 
+#### `get_permissions`
+
+```python
+# Get the file/directory permissions.
+permissions = fsutil.get_permissions(path)
+```
+
 #### `get_unique_name`
 
 ```python
-# Gets a unique name for a directory/file ath the given directory path.
+# Get a unique name for a directory/file ath the given directory path.
 unique_name = fsutil.get_unique_name(path, prefix="", suffix="", extension="", separator="-")
 ```
 
@@ -747,6 +756,13 @@ dirs = fsutil.search_dirs(path, pattern="**/*")
 files = fsutil.search_files(path, pattern="**/*.*")
 ```
 
+#### `set_permissions`
+
+```python
+# Set the file/directory permissions.
+fsutil.set_permissions(path, 700)
+```
+
 #### `split_filename`
 
 ```python
@@ -772,14 +788,14 @@ path_names = fsutil.split_path(path)
 
 ```python
 # Write file with the specified content at the given path.
-fsutil.write_file(path, content, append=False, encoding="utf-8")
+fsutil.write_file(path, content, append=False, encoding="utf-8", atomic=False)
 ```
 
 #### `write_file_json`
 
 ```python
 # Write a json file at the given path with the specified data encoded in json format.
-fsutil.write_file_json(path, data, skipkeys=False, ensure_ascii=True, check_circular=True, allow_nan=True, cls=None, indent=None, separators=None, default=None, sort_keys=False)
+fsutil.write_file_json(path, data, encoding="utf-8", atomic=False, skipkeys=False, ensure_ascii=True, check_circular=True, allow_nan=True, cls=None, indent=None, separators=None, default=None, sort_keys=False)
 ```
 
 ## Testing

python_fsutil-0.13.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+fsutil/__init__.py,sha256=IpqVSsROxOpfRTEpHEgmB8Y0Ig0OwLcOEOvl_xNRlCg,39444
+fsutil/metadata.py,sha256=K36K01JFKjpV25tf7qT9WchCaxT2E7DfEJ2Dx-3OQjk,266
+fsutil/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_fsutil-0.13.0.dist-info/LICENSE.txt,sha256=IcPfOD_lonKcLMnPRhiWdQUPW_LYW-5TTDljwi3UXQg,1078
+python_fsutil-0.13.0.dist-info/METADATA,sha256=AKjGMjwCxQkDP5gbvq-q2XUow99v_FL53oHh5ewS3vs,24628
+python_fsutil-0.13.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+python_fsutil-0.13.0.dist-info/top_level.txt,sha256=KGTSjNLPWmxnZt8BFMZQrpnhkO7DsKrYWoXK-Sf3wBQ,7
+python_fsutil-0.13.0.dist-info/RECORD,,

{python_fsutil-0.11.0.dist-info → python_fsutil-0.13.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.41.3)
+Generator: bdist_wheel (0.42.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

python_fsutil-0.11.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-fsutil/__init__.py,sha256=BfBvX3rKqyo_UiiovROK1KRyiUlFaHhsdwTpoeqqBvg,37421
-fsutil/metadata.py,sha256=LaJ9f5wd3szp7udufkvrJDJItKAP0J5PuTYCYZC83XQ,266
-fsutil/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_fsutil-0.11.0.dist-info/LICENSE.txt,sha256=IcPfOD_lonKcLMnPRhiWdQUPW_LYW-5TTDljwi3UXQg,1078
-python_fsutil-0.11.0.dist-info/METADATA,sha256=3cI2HrIDbgjcKgCXw41E1_AjZIDuAx7wtZcOnmos_S0,24268
-python_fsutil-0.11.0.dist-info/WHEEL,sha256=Xo9-1PvkuimrydujYJAjF7pCkriuXBpUPEjma1nZyJ0,92
-python_fsutil-0.11.0.dist-info/top_level.txt,sha256=KGTSjNLPWmxnZt8BFMZQrpnhkO7DsKrYWoXK-Sf3wBQ,7
-python_fsutil-0.11.0.dist-info/RECORD,,