Merge pull request pandas-dev#389 from manahl/issue-343

Renaming libraries within a database/namespace

Author: bmoscon

CHANGES.md

@@ -3,6 +3,7 @@
 ### 1.49
   * Bugfix:  #384 sentinels missing time data on chunk start/ends in ChunkStore
   * Bugfix:  #382 Remove dependency on cython being pre-installed
+  * Bugfix:  #343 Renaming libraries/collections within a namespace/database
 
 ### 1.48 (2017-06-26)
   * BugFix: Rollback #363, as it breaks multi-index dataframe

arctic/arctic.py

@@ -322,13 +322,21 @@ def rename_library(self, from_lib, to_lib):
         to_lib: str
             The new name of the library
         """
+        to_colname = to_lib
+        if '.' in from_lib and '.' in to_lib:
+            if from_lib.split('.')[0] != to_lib.split('.')[0]:
+                raise ValueError("Collection can only be renamed in the same database")
+            to_colname = to_lib.split('.')[1]
+
         l = ArcticLibraryBinding(self, from_lib)
         colname = l.get_top_level_collection().name
-        logger.info('Dropping collection: %s' % colname)
-        l._db[colname].rename(to_lib)
+
+        logger.info('Renaming collection: %s' % colname)
+        l._db[colname].rename(to_colname)
         for coll in l._db.collection_names():
             if coll.startswith(colname + '.'):
-                l._db[coll].rename(coll.replace(from_lib, to_lib))
+                l._db[coll].rename(coll.replace(colname, to_colname))
+
         if from_lib in self._library_cache:
             del self._library_cache[from_lib]
             del self._library_cache[l.get_name()]

tests/integration/test_arctic.py

@@ -168,6 +168,25 @@ def test_lib_rename(arctic):
     assert('test' not in arctic.list_libraries())
 
 
+def test_lib_rename_namespace(arctic):
+    arctic.initialize_library('namespace.test')
+    l = arctic['namespace.test']
+    l.write('test_data', 'abc')
+
+    with pytest.raises(ValueError) as e:
+        arctic.rename_library('namespace.test', 'new_namespace.test')
+    assert('Collection can only be renamed in the same database' in str(e))
+
+    arctic.rename_library('namespace.test', 'namespace.newlib')
+    l = arctic['namespace.newlib']
+    assert(l.read('test_data').data == 'abc')
+
+    with pytest.raises(LibraryNotFoundException) as e:
+        l = arctic['namespace.test']
+    assert('Library namespace.test' in str(e))
+    assert('namespace.test' not in arctic.list_libraries())
+
+
 def test_lib_type(arctic):
     arctic.initialize_library('test')
     assert(arctic.get_library_type('test') == VERSION_STORE)