def deldoc(self): """Deletes specified document in the database. """ mode, kwargs = self.mode, self.kwargs dbname = dbu.get_dbname(**kwargs) client = self.client() if not self.check_database(client, dbname): return detname = kwargs.get('detector', None) if detname is None: logger.warning( '%s needs in the collection name. Please specify the detector name.' % (mode)) colname = detname db, fs = dbu.db_and_fs(client, dbname) colnames = dbu.collection_names(db) if not (colname in colnames): # dbu.collection_exists(db, colname) logger.warning('db "%s" does not have collection "%s"' % (db.name, str(colname))) return col = dbu.collection(db, colname) logger.info('command mode: "%s" db: "%s" collection: "%s"' % (mode, db.name, str(colname))) defs = self.defs ctype = kwargs.get('ctype', None) run = kwargs.get('run', None) tsec = kwargs.get('time_sec', None) tstamp = kwargs.get('time_stamp', None) vers = kwargs.get('version', None) confirm = kwargs.get('confirm', False) query = {'detector': detname} if ctype != defs['ctype']: query['ctype'] = ctype if run != defs['run']: query['run'] = run if vers != defs['version']: query['version'] = vers #if tsec != defs['time_sec'] : query['time_sec'] = tsec if is_in_command_line('-s', '--time_sec'): query['time_sec'] = tsec if is_in_command_line('-t', '--time_stamp'): query['time_stamp'] = tstamp logger.info('query: %s' % str(query)) docs = dbu.find_docs(col, query) if docs is None or docs.count() == 0: logger.warning('Can not find document for query: %s' % str(query)) return for i, doc in enumerate(docs): msg = ' deldoc %2d:'%i + doc['time_stamp'] + ' ' + str(doc['time_sec'])\ + ' %s'%doc['ctype'].ljust(16) + ' %4d'%doc['run'] + ' ' + str(doc['id_data']) logger.info(msg) if confirm: dbu.delete_document_from_collection(col, doc['_id']) dbu.del_document_data(doc, fs) if not confirm: dbu.request_confirmation()
def delete_documents(dbname, colname, doc_ids): """Delete documents with _id-s in doc_ids from dbname, colname """ #logger.debug('Deleting documents:\n %s' % ('\n '.join(doc_ids))) client = connect_client() db, fs = dbu.db_and_fs(client, dbname) col = collection(db, colname) #msg = 'Deleted documents from db: %s col: %s' % (dbname, colname) for s in doc_ids: oid = ObjectId(s) doc = dbu.find_doc(col, query={'_id': oid}) if doc is None: continue #msg += '\n %s and its data' % doc.get('_id', 'N/A') dbu.del_document_data(doc, fs) dbu.delete_document_from_collection(col, oid)