From a21a4cc81cf06dca2607011c1c3df7d63c95e735 2014-03-14 17:06:02
From: MinRK <benjaminrk@gmail.com>
Date: 2014-03-14 17:06:02
Subject: [PATCH] fallback on copy, copyfile if copy2 fails

in notebook manager

---

diff --git a/IPython/html/services/notebooks/filenbmanager.py b/IPython/html/services/notebooks/filenbmanager.py
index 78500ff..ae9881f 100644
--- a/IPython/html/services/notebooks/filenbmanager.py
+++ b/IPython/html/services/notebooks/filenbmanager.py
@@ -85,7 +85,25 @@ class FileNotebookManager(NotebookManager):
                 os.mkdir(new)
             except:
                 raise TraitError("Couldn't create checkpoint dir %r" % new)
-
+    
+    def _copy(self, src, dest):
+        """copy src to dest
+        
+        try copy2, fallback to copy, then finally copyfile
+        """
+        exc = None
+        for name in ('copy2', 'copy', 'copyfile'):
+            cp = getattr(shutil, name)
+            try:
+                cp(src, dest)
+            except OSError as e:
+                exc = e # assignment required for Python 3
+                self.log.debug("%s(%r,%r) failed: %s", name, src, dest, e)
+            else:
+                return
+        # will only get here if all copies fail:
+        raise exc
+    
     def get_notebook_names(self, path=''):
         """List all notebook names in the notebook dir and path."""
         path = path.strip('/')
@@ -432,7 +450,7 @@ class FileNotebookManager(NotebookManager):
         self.log.debug("creating checkpoint for notebook %s", name)
         if not os.path.exists(self.checkpoint_dir):
             os.mkdir(self.checkpoint_dir)
-        shutil.copy2(nb_path, cp_path)
+        self._copy(nb_path, cp_path)
         
         # return the checkpoint info
         return self.get_checkpoint_model(checkpoint_id, name, path)
@@ -465,7 +483,7 @@ class FileNotebookManager(NotebookManager):
         # ensure notebook is readable (never restore from an unreadable notebook)
         with io.open(cp_path, 'r', encoding='utf-8') as f:
             current.read(f, u'json')
-        shutil.copy2(cp_path, nb_path)
+        self._copy(cp_path, nb_path)
         self.log.debug("copying %s -> %s", cp_path, nb_path)
     
     def delete_checkpoint(self, checkpoint_id, name, path=''):