Merge pull request untitaker#48 from LovecraftianHorror/pathlib-support

untitaker · web-flow · commit ba24a05b55aa · 2020-04-28T23:38:54.000+02:00
Add support for pathlib paths Fix untitaker#47
diff --git a/atomicwrites/__init__.py b/atomicwrites/__init__.py
@@ -9,6 +9,12 @@
 except ImportError:
     fcntl = None
 
+# `fspath` was added in Python 3.6
+try:
+    from os import fspath
+except ImportError:
+    fspath = None
+
 __version__ = '1.3.0'
 
 
@@ -137,6 +143,10 @@ def __init__(self, path, mode=DEFAULT_MODE, overwrite=False,
         if 'w' not in mode:
             raise ValueError('AtomicWriters can only be written to.')
 
+        # Attempt to convert `path` to `str` or `bytes`
+        if fspath is not None:
+            path = fspath(path)
+
         self._path = path
         self._mode = mode
         self._overwrite = overwrite
