def test_atomic_on_outmost_connection_that_is_in_transaction():
conn = ConnectionPlus(sqlite3.connect(':memory:'))
conn.execute('BEGIN')
assert True is conn.in_transaction
match_str = re.escape('SQLite connection has uncommitted transactions. '
'Please commit those before starting an atomic '
'transaction.')
with pytest.raises(RuntimeError, match=match_str):
with atomic(conn):
pass
def get_parameters(conn: ConnectionPlus, run_id: int) -> List[ParamSpec]:
"""
Get the list of param specs for run
Args:
conn: the connection to the sqlite database
run_id: The id of the run
Returns:
A list of param specs for this run
"""
sql = f"""
SELECT parameter FROM layouts WHERE run_id={run_id}
"""
c = conn.execute(sql)
param_names_temp = many_many(c, 'parameter')
param_names = [p[0] for p in param_names_temp]
param_names = cast(List[str], param_names)
parspecs = []
for param_name in param_names:
parspecs.append(get_paramspec(conn, run_id, param_name))
return parspecs
def get_paramspec(conn: ConnectionPlus, run_id: int,
param_name: str) -> ParamSpec:
"""
Get the ParamSpec object for the given parameter name
in the given run
Args:
conn: Connection to the database
run_id: The run id
param_name: The name of the parameter
"""
# get table name
sql = f"""
SELECT result_table_name FROM runs WHERE run_id = {run_id}
"""
c = conn.execute(sql)
result_table_name = one(c, 'result_table_name')
# get the data type
sql = f"""
PRAGMA TABLE_INFO("{result_table_name}")
"""
c = conn.execute(sql)
for row in c.fetchall():
if row['name'] == param_name:
param_type = row['type']
break
# get everything else
sql = f"""
SELECT * FROM layouts
WHERE parameter="{param_name}" and run_id={run_id}
"""
c = conn.execute(sql)
resp = many(c, 'layout_id', 'run_id', 'parameter', 'label', 'unit',
'inferred_from')
(layout_id, _, _, label, unit, inferred_from_string) = resp
if inferred_from_string:
inferred_from = inferred_from_string.split(', ')
else:
inferred_from = []
deps = get_dependencies(conn, layout_id)
depends_on: Optional[List[str]]
if len(deps) == 0:
depends_on = None
else:
dps: List[int] = [dp[0] for dp in deps]
ax_nums: List[int] = [dp[1] for dp in deps]
depends_on = []
for _, dp in sorted(zip(ax_nums, dps)):
sql = f"""
SELECT parameter FROM layouts WHERE layout_id = {dp}
"""
c = conn.execute(sql)
depends_on.append(one(c, 'parameter'))
parspec = ParamSpec(param_name, param_type, label, unit, inferred_from,
depends_on)
return parspec