82d417b9e6
Fresh start for the StarlingX automation framework. Change-Id: Ie265e0791024f45f71faad6315c2b91b022934d1
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
from framework.database.connection.database_operation_manager import DatabaseOperationManager
|
|
from psycopg2.extras import RealDictCursor
|
|
|
|
|
|
class RunOperation:
|
|
"""
|
|
This class allows you to perform Run Database Operations
|
|
"""
|
|
|
|
def __init__(self):
|
|
self.database_operation_manager = DatabaseOperationManager()
|
|
|
|
def create_run(self, run_name: str, run_type_id: int, release: str) -> int:
|
|
"""
|
|
Creates a run
|
|
Args:
|
|
run_name (): the name of the run
|
|
run_type_id (): the run_type_id
|
|
release (): the release
|
|
|
|
Returns:
|
|
|
|
"""
|
|
insert_query = "INSERT INTO run " "(run_name, run_type_id, release) " f"VALUES ('{run_name}',{run_type_id},'{release}') RETURNING run_id"
|
|
|
|
results = self.database_operation_manager.execute_query(insert_query, cursor_factory=RealDictCursor)
|
|
|
|
if results:
|
|
return results[0]['run_id'] # can only ever be 1 result
|
|
|
|
raise ValueError("Unable to insert the run and get a run id.")
|