def _get_db_for_doc_type(self, doc_type): if doc_type not in self._dbs: couch_db = get_db_by_doc_type(doc_type) callback = LoaderCallback(self.success_counter, self.stdout) db = IterDB(couch_db, new_edits=False, callback=callback) db.__enter__() self._dbs[doc_type] = db return self._dbs[doc_type]
def _get_db_for_doc_type(self, doc_type): if doc_type not in self._dbs: couch_db = get_db_by_doc_type(doc_type) if couch_db is None: raise DocumentClassNotFound('No Document class with name "{}" could be found.'.format(doc_type)) callback = LoaderCallback(self.success_counter, self.stdout) db = IterDB(couch_db, new_edits=False, callback=callback) db.__enter__() self._dbs[doc_type] = db return self._dbs[doc_type]