No longer using deprecated db.collection_names() function

armadillica · Apr 26, 2019 · ae6fbf5 · ae6fbf5
1 parent cd21c4e
commit ae6fbf5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/flamenco/__init__.py b/flamenco/__init__.py
@@ -190,7 +190,8 @@ def _create_collections(self, db):
         )
 
         # flamenco_tasks
-        if 'flamenco_tasks' not in db.collection_names(include_system_collections=False):
+        collection_names = {coll['name'] for coll in db.list_collections()}
+        if 'flamenco_tasks' not in collection_names:
             self._log.info('Creating flamenco_tasks collection.')
             db.create_collection('flamenco_tasks',
                                  storageEngine={
@@ -214,7 +215,7 @@ def _create_collections(self, db):
         )
 
         # Manager linking keys
-        if 'flamenco_manager_linking_keys' not in db.list_collection_names():
+        if 'flamenco_manager_linking_keys' not in collection_names:
             self._log.info('Creating flamenco_manager_linking_keys collection.')
             db.create_collection('flamenco_manager_linking_keys')
         else: