diff options
| -rw-r--r-- | gn3/db/traits.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/gn3/db/traits.py b/gn3/db/traits.py index fc1463c..3134d31 100644 --- a/gn3/db/traits.py +++ b/gn3/db/traits.py @@ -88,8 +88,32 @@ def insert_publication_data(riset_id: int, strains: List, conn: Any): "%s AND Name in (%s)" % (riset_id, ", ".join(f"'{strain}'" for strain in strains))) strain_ids = cursor.fetchall() + data_id, *_ = cursor.execute("SELECT max(Id) " + "from PublishData").fetchone() + if strain_ids: strain_ids = list(chain(*strain_ids)) + for strain, strain_id in zip(strains, strain_ids): + # TODO: Fetch this values correctly using sql + value, variance, trait_np = 1, 1, 1 + if value: + cursor.execute( + "INSERT INTO PublishData values (%d, %d, %s)" % + (data_id, strain_id, value)) + if variance: + cursor.execute( + "INSERT INTO PublishSE values (%d, %d, %s)" % + (data_id, strain_id, variance)) + if trait_np: + cursor.execute( + "INSERT INTO NStrain values (%d, %d, %s)" % + (data_id, strain_id, trait_np)) + inbred_set_id = lookup_webqtldataset_name(riset_name="", + conn=conn) + cursor.execute( + "SELECT max(Sequence) FROM PublishXRef where " + "InbredSetId = %d and PhenotypeId = %d and PublicationId = %d" % + (inbred_set_id, phenotype_id, publication_id)) def insert_phenotype(phenotype: Optional[Dict], conn: Any): |