add support for FileNotFoundError

untitaker · untitaker · commit b41a4fd95396 · 2015-02-08T14:23:36.000+01:00
diff --git a/atomicwrites/__init__.py b/atomicwrites/__init__.py
@@ -12,9 +12,13 @@
 if PY2:
     class FileExistsError(OSError):
         errno = errno.EEXIST
+
+    class FileNotFoundError(OSError):
+        errno = errno.ENOENT
 else:
     # For some reason we have to redefine this, or users can't import it.
     FileExistsError = FileExistsError
+    FileNotFoundError = FileNotFoundError
 
 
 if sys.platform != 'win32':
@@ -42,14 +46,19 @@ def _move_atomic(src, dst):
     import pywintypes
 
     _windows_default_flags = win32file.MOVEFILE_WRITE_THROUGH
+    _windows_error_table = {
+        183: FileExistsError,
+        3: FileNotFoundError
+    }
 
     @contextlib.contextmanager
     def handle_errors():
         try:
             yield
         except pywintypes.error as e:
-            if e.winerror == 183:
-                raise FileExistsError(str(e))
+            native_cls = _windows_error_table.get(e.winerror, None)
+            if native_cls is not None:
+                raise native_cls(str(e))
             else:
                 raise
 
diff --git a/docs/index.rst b/docs/index.rst
@@ -7,10 +7,19 @@ API
 
 .. autofunction:: atomic_write
 
+
+Exceptions
+----------
+
 .. exception:: FileExistsError
 
-   This is Python 3's builtin ``FileExistsError``, under Python 2 it's a
-   subclass of ``OSError`` with similar semantics.
+    This is Python 3's builtin ``FileExistsError``, under Python 2 it's a
+    subclass of ``OSError`` with similar semantics.
+
+.. exception:: FileNotFoundError
+
+    This is Python 3's builtin ``FileNotFoundError``, under Python 2 it's a
+    subclass of ``OSError`` with similar semantics.
 
 Low-level API
 -------------
