print "Running queries, count " + str(z)
idxStart = time.time()
cursor.execute("SELECT * FROM locus l, snp s WHERE l.snp_id = s.id AND s.rsid = 'rs8788'")
idxEnd = time.time()
result.qryByRsid = idxEnd - idxStart
idxStart = time.time()
cursor.execute("SELECT count(s.id) FROM locus l, snp s WHERE l.snp_id = s.id AND s.has_sig = true")
idxEnd = time.time()
result.qryByClinSig = idxEnd - idxStart
idxStart = time.time()
cursor.execute("SELECT count(distinct s.rsid) FROM locus l, snp s WHERE l.snp_id = s.id AND l.gene = 'GRIN2B'")
idxEnd = time.time()
result.qryByGene = idxEnd - idxStart
idxStart = time.time()
cursor.execute("SELECT count(distinct s.rsid) FROM locus l, snp s WHERE l.snp_id = s.id AND l.gene = 'GRIN2B' AND s.has_sig = true")
idxEnd = time.time()
result.qryByGeneSig = idxEnd - idxStart
resultsFile.write(result.toString() + '\n')
# Close pgsql cursor
cursor.close()
resultsFile.close()
postgresConnection.close()
print "Run complete."
result.tag = tag
print "Running queries, count " + str(z)
qryStart = time.time()
mongoCollection.find({"rsid":"rs8788"})
qryEnd = time.time()
result.qryByRsid = qryEnd-qryStart
qryStart = time.time()
temptotal = mongoCollection.find({"has_sig":True}).count()
qryEnd = time.time()
result.qryByClinSig = qryEnd-qryStart
qryStart = time.time()
temptotal = mongoCollection.find({"loci.gene":"GRIN2B"}).count()
qryEnd = time.time()
result.qryByGene = qryEnd-qryStart
qryStart = time.time()
temptotal = mongoCollection.find({"has_sig":True,"loci.gene":"GRIN2B"}).count()
qryEnd = time.time()
result.qryByGeneSig = qryEnd-qryStart
resultsFile.write(result.toString() + '\n')
if remote:
try:
print "Sending to GDocs..."
gs.login()
ws.append_row(result.stringArr())
except:
print "Unable to send to GDocs, continuing..."
