blob: 8122cfad3b1920355e44f39ec44b8584aa561d90 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
"""Functions for accessing the database relating to datasets."""
import MySQLdb as mdb
from MySQLdb.cursors import DictCursor
def geno_dataset_by_species_and_population(
conn: mdb.Connection,
speciesid: int,
populationid: int) -> tuple[dict, ...]:
"""Retrieve all genotypes datasets by species and population"""
with conn.cursor(cursorclass=DictCursor) as cursor:
cursor.execute(
"SELECT gf.* FROM InbredSet AS iset INNER JOIN GenoFreeze AS gf "
"ON iset.InbredSetId=gf.InbredSetId "
"WHERE iset.SpeciesId=%(sid)s AND iset.InbredSetId=%(pid)s",
{"sid": speciesid, "pid": populationid})
return tuple(dict(row) for row in cursor.fetchall())
def geno_dataset_by_id(conn: mdb.Connection, dataset_id: int) -> dict:
"""Retrieve genotype dataset by ID"""
with conn.cursor(cursorclass=DictCursor) as cursor:
cursor.execute("SELECT * FROM GenoFreeze WHERE Id=%s",
(dataset_id,))
return dict(cursor.fetchone())
|
