def report_connection(idb, inv_db):
"""Check connection, check is inventory, check inventory is as expected
"""
roles = idb['admin'].command(SON({"connectionStatus":int(1)}))
can_write = 'readWrite' in [el['role'] for el in roles['authInfo']['authenticatedUserRoles'] if el['db']=='inventory']
inv_ok = inv.validate_mongodb(inv_db)
lockout = get_lockout_state()
return {'can_write':can_write, 'inv_ok': inv_ok, 'global_lock':lockout}
def delete_all_tables(db):
""" Delete all tables specified by inv Note that other names can be present, see inv.validate_mongod"""
if not inv.validate_mongodb(db):
raise TypeError("db does not match Inventory structure")
return
tbls = inv.get_inv_table_names()
for tbl in tbls:
try:
delete_contents(db, tbl)
delete_table(db, tbl)
except Exception as e:
inv.log_dest.error("Error deleting "+ tbl + ' ' +str(e))
def delete_all_tables(db):
""" Delete all tables specified by inv Note that other names can be present, see inv.validate_mongod"""
if not inv.validate_mongodb(db):
raise TypeError("db does not match Inventory structure")
return
tbls = inv.get_inv_table_names()
for tbl in tbls:
try:
delete_contents(db, tbl)
delete_table(db, tbl)
except Exception as e:
inv.log_dest.error("Error deleting " + tbl + ' ' + str(e))
def delete_table(db, tbl_name, check=True):
"""Delete tbl_name (must be empty) """
if not inv.validate_mongodb(db) and check:
raise TypeError("db does not match Inventory structure")
return
#Grab the possible table names
tbl_names = inv.get_inv_table_names()
if tbl_name in tbl_names:
try:
assert(db[tbl_name].find_one() is None) #Check content is gone
db[tbl_name].drop()
except Exception as e:
inv.log_dest.error("Error dropping "+ tbl_name+' '+ str(e))
def delete_table(db, tbl_name, check=True):
"""Delete tbl_name (must be empty) """
if not inv.validate_mongodb(db) and check:
raise TypeError("db does not match Inventory structure")
return
#Grab the possible table names
tbl_names = inv.get_inv_table_names()
if tbl_name in tbl_names:
try:
assert (db[tbl_name].find_one() is None) #Check content is gone
db[tbl_name].drop()
except Exception as e:
inv.log_dest.error("Error dropping " + tbl_name + ' ' + str(e))
def delete_contents(db, tbl_name, check=True):
"""Delete contents of tbl_name """
if not inv.validate_mongodb(db) and check:
raise TypeError("db does not match Inventory structure")
return
#Grab the possible table names
tbl_names = inv.get_inv_table_names()
if tbl_name in tbl_names:
try:
db[tbl_name].remove()
except Exception as e:
inv.log_dest.error("Error deleting from "+ tbl_name+' '+ str(e)+", dropping")
#Capped tables, e.g rollback, can only be dropped, so try that
try:
db[tbl_name].drop()
except Exception as e:
inv.log_dest.error("Error dropping "+ tbl_name+' '+ str(e))
def delete_contents(db, tbl_name, check=True):
"""Delete contents of tbl_name """
if not inv.validate_mongodb(db) and check:
raise TypeError("db does not match Inventory structure")
return
#Grab the possible table names
tbl_names = inv.get_inv_table_names()
if tbl_name in tbl_names:
try:
db[tbl_name].remove()
except Exception as e:
inv.log_dest.error("Error deleting from " + tbl_name + ' ' +
str(e) + ", dropping")
#Capped tables, e.g rollback, can only be dropped, so try that
try:
db[tbl_name].drop()
except Exception as e:
inv.log_dest.error("Error dropping " + tbl_name + ' ' + str(e))
def delete_by_collection(inv_db, db_name, coll_name):
"""Remove collection entry and all its fields"""
if not inv.validate_mongodb(inv_db):
raise TypeError("db does not match Inventory structure")
return
try:
_db_id = invc.get_db_id(inv_db, db_name)
_c_id = invc.get_coll_id(inv_db, _db_id['id'], coll_name)
except Exception as e:
inv.log_dest.error("Error getting collection " + str(e))
return {'err':True, 'id':0, 'exist':False}
#Remove fields entries matching _c_id
delete_collection_data(inv_db, _c_id['id'], tbl='auto')
delete_collection_data(inv_db, _c_id['id'], tbl='human')
delete_collection_data(inv_db, _c_id['id'], tbl='records')
try:
inv_db[inv.ALL_STRUC.coll_ids[inv.STR_NAME]].remove({'_id':_c_id['id']})
except Exception as e:
inv.log_dest.error("Error removing collection " + str(e))
def delete_by_collection(inv_db, db_name, coll_name):
"""Remove collection entry and all its fields"""
if not inv.validate_mongodb(inv_db):
raise TypeError("db does not match Inventory structure")
return
try:
_db_id = invc.get_db_id(inv_db, db_name)
_c_id = invc.get_coll_id(inv_db, _db_id['id'], coll_name)
except Exception as e:
inv.log_dest.error("Error getting collection " + str(e))
return {'err': True, 'id': 0, 'exist': False}
#Remove fields entries matching _c_id
delete_collection_data(inv_db, _c_id['id'], tbl='auto')
delete_collection_data(inv_db, _c_id['id'], tbl='human')
delete_collection_data(inv_db, _c_id['id'], tbl='records')
try:
inv_db[inv.ALL_STRUC.coll_ids[inv.STR_NAME]].remove(
{'_id': _c_id['id']})
except Exception as e:
inv.log_dest.error("Error removing collection " + str(e))