Add rename support for Libraries in Arctic (pandas-dev#225)

Author: bmoscon

arctic/arctic.py

@@ -302,6 +302,28 @@ def check_quota(self, library):
         l = ArcticLibraryBinding(self, library)
         l.check_quota()
 
+    def rename_library(self, from_lib, to_lib):
+        """
+        Renames a library
+
+        Parameters
+        ----------
+        from_lib: str
+            The name of the library to be renamed
+        to_lib: str
+            The new name of the library
+        """
+        l = ArcticLibraryBinding(self, from_lib)
+        colname = l.get_top_level_collection().name
+        logger.info('Dropping collection: %s' % colname)
+        l._db[colname].rename(to_lib)
+        for coll in l._db.collection_names():
+            if coll.startswith(colname + '.'):
+                l._db[coll].rename(coll.replace(from_lib, to_lib))
+        if from_lib in self._library_cache:
+            del self._library_cache[from_lib]
+            del self._library_cache[l.get_name()]
+
 
 class ArcticLibraryBinding(object):
     """

tests/integration/test_arctic.py

@@ -144,3 +144,16 @@ def test_default_mongo_retry_timout():
     with pytest.raises(LibraryNotFoundException):
         Arctic('unresolved-host', serverSelectionTimeoutMS=0)['some.lib']
     assert time.time() - now < 1.
+
+
+def test_lib_rename(arctic):
+    arctic.initialize_library('test')
+    l = arctic['test']
+    l.write('test_data', 'abc')
+    arctic.rename_library('test', 'new_name')
+    l = arctic['new_name']
+    assert(l.read('test_data').data == 'abc')
+    with pytest.raises(LibraryNotFoundException) as e:
+        l = arctic['test']
+    assert('Library test' in str(e))
+    assert('test' not in arctic.list_libraries())