82d417b9e6
Fresh start for the StarlingX automation framework. Change-Id: Ie265e0791024f45f71faad6315c2b91b022934d1
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
from framework.database.connection.database_operation_manager import DatabaseOperationManager
|
|
from framework.logging.automation_logger import get_logger
|
|
|
|
|
|
class LabOperation:
|
|
"""
|
|
Lab Operation class
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.database_operation_manager = DatabaseOperationManager()
|
|
|
|
def does_lab_exist(self, lab_name: str) -> bool:
|
|
"""
|
|
Returns true if a lab with the given name exists
|
|
Args:
|
|
lab_name (): the name of the lab
|
|
|
|
Returns: True if it exists, False otherwise
|
|
|
|
"""
|
|
get_lab_id_query = f"SELECT lab_info_id FROM lab_info where lab_name='{lab_name}'"
|
|
result = self.database_operation_manager.execute_query(get_lab_id_query)
|
|
|
|
return len(result) > 0
|
|
|
|
def insert_lab(self, lab_name: str):
|
|
"""
|
|
Insert lab into database
|
|
Args:
|
|
lab_name (): the lab name
|
|
|
|
Returns:
|
|
|
|
"""
|
|
if not self.does_lab_exist(lab_name): # Only add the lab if it doesn't exist.
|
|
|
|
create_lab_query = f"INSERT INTO lab_info (lab_name) VALUES ('{lab_name}') RETURNING lab_info_id"
|
|
result = self.database_operation_manager.execute_query(create_lab_query)
|
|
return result[0][0]
|
|
else:
|
|
get_logger().log_error(f"WARNING: This lab is already in the database! Lab Name: {lab_name}")
|
|
return -1
|
|
|
|
def get_lab_id(self, lab_name):
|
|
"""
|
|
This function will transform the lab_name passed in into the equivalent lab id.
|
|
Args:
|
|
lab_name (str): is the name of the lab.
|
|
Returns: The id associated with the lab.
|
|
"""
|
|
# Get the lab id from the database.
|
|
get_lab_id_query = f"SELECT lab_info_id FROM lab_info where lab_name='{lab_name}'"
|
|
|
|
result = self.database_operation_manager.execute_query(get_lab_id_query)
|
|
if result:
|
|
if len(result) > 1:
|
|
get_logger().log_info(f"WARNING: We have found more than one result matching the Lab Name: {lab_name}")
|
|
return result[0][0] # First Row, First Entry
|
|
else:
|
|
get_logger().log_error(f"ERROR: No lab with the name {lab_name} exists.")
|
|
return -1
|