def get_rows(
table_name: str,
column_names: Optional[List[str]] = None,
filter_formula: Optional[Mapping] = None,
filter_pairs: Optional[Mapping] = None,
):
"""Get columns in a row selected by filter and/or pairs.
Execute a select query in the database with an optional filter and
pairs and return a subset of columns (or all of them if empty)
:param table_name: Primary key of the workflow storing the data
:param column_names: optional list of columns to select
:param filter_formula: Optional JSON formula to use in the WHERE clause
:param filter_pairs: Pairs key: value to filter in the WHERE clause
:return: cursor resulting from the query
"""
query, fields = get_select_query(
table_name,
column_names=column_names,
filter_formula=filter_formula,
filter_pairs=filter_pairs,
)
# Execute the query
cursor = connection.connection.cursor(cursor_factory=DictCursor)
cursor.execute(query, fields)
return cursor
def get_row(
table_name: str,
key_name: str,
key_value,
column_names: Optional[List[str]] = None,
filter_formula: Optional[Mapping] = None,
filter_pairs: Optional[Mapping] = None,
):
"""Get a single row in the DB with the key name/value pair.
:param table_name: Name of the table
:param key_name: Key name to uniquely identify the row
:param key_value: Key value to uniquely identify the row
:param column_names: Columns to access (all of them if empty)
:param filter_formula: Optional filter formula
:param filter_pairs: Optional dictionary to restrict the clause
:return: Dictionary with the row
"""
key_pair = {key_name: key_value}
if filter_pairs:
filter_pairs = dict(key_pair, **filter_pairs)
else:
filter_pairs = key_pair
query, fields = get_select_query(
table_name,
column_names=column_names,
filter_formula=filter_formula,
filter_pairs=filter_pairs,
)
# Execute the query
cursor = connection.connection.cursor(cursor_factory=DictCursor)
cursor.execute(query, fields)
if cursor.rowcount != 1:
raise Exception('Query returned more than one row.')
return cursor.fetchone()