aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--wqflask/wqflask/correlation/correlation_file_generator.py26
1 files changed, 23 insertions, 3 deletions
diff --git a/wqflask/wqflask/correlation/correlation_file_generator.py b/wqflask/wqflask/correlation/correlation_file_generator.py
index 401b3ba6..e06c4123 100644
--- a/wqflask/wqflask/correlation/correlation_file_generator.py
+++ b/wqflask/wqflask/correlation/correlation_file_generator.py
@@ -2,6 +2,29 @@ from urllib.parse import urlparse
import pymysql as mdb
import os
import csv
+import lmdb
+import os
+import tempfile
+
+
+
+def parse_db_url():
+ """function to parse SQL_URI env variable note:there\
+ is a default value for SQL_URI so a tuple result is\
+ always expected"""
+ parsed_db = urlparse(SQL_URI)
+
+
+ return (
+ parsed_db.hostname, parsed_db.username, parsed_db.password,
+ parsed_db.path[1:], 3306)
+
+
+# This function is deprecated. Use database_connection instead.
+def database_connector():
+ """function to create db connector"""
+ host, user, passwd, db_name, db_port = parse_db_url()
+ return mdb.connect(host=host,user=user,password=passwd,database=db_name)
def get_probesetfreezes(conn, inbredsetid=1):
@@ -107,9 +130,6 @@ def generate_csv_file(conn,db_name,txt_dir,file_name):
# file name ,file expiry,type of storage
- # I want to use lmdb to store the files
- # file name already done #import that
-
# file expiry to be done lt
try: