feat(hbi_utils.py): since most of tables are made private, need to ch…

…ange the way of accessing them.
redhat-performance · Nov 8, 2024 · dd7a4fb · dd7a4fb
1 parent 122bb3a
commit dd7a4fb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/opl/hbi_utils.py b/opl/hbi_utils.py
@@ -114,7 +114,7 @@ def wait_for_next_second():
 
 def fetch_records_count(inventory):
     inventory_cursor = inventory.cursor()
-    inventory_cursor.execute("select count(*) as exact_count from hosts")
+    inventory_cursor.execute("select count(*) as exact_count from hbi.hosts")
     for i in inventory_cursor.fetchone():
         existing_records = int(i)
 
@@ -306,7 +306,7 @@ def cleanup(args, status_data):
     inventory_cursor = inventory.cursor()
 
     logging.info("Truncating Inventory DB 'hosts' table")
-    inventory_cursor.execute("TRUNCATE hosts CASCADE")
+    inventory_cursor.execute("TRUNCATE hbi.hosts CASCADE")
     inventory.commit()
 
     status_data.set_now("parameters.inventory_db.table_hosts.truncated_at")