1 files changed, 69 insertions, 0 deletions
diff --git a/tests/dbTestFiller.py b/tests/dbTestFiller.py
new file mode 100644
index 0000000..2046b04
--- /dev/null
+++ b/tests/dbTestFiller.py
@@ -0,0 +1,69 @@
+import mysql.connector as con
+
+debug = False
+host = "navanspi.duckdns.org"
+done = 1
+
+if debug != False:
+    host = "192.168.1.6"
+    done = 0
+
+mycon = con.connect(host=host,user="curieweb",password="curie-web-russian-54",port=3306,database="curie")
+mycursor = mycon.cursor()
+
+try:
+    mycursor.execute("create table curieweb (    id varchar(16) primary key, email nvarchar(255) NOT NULL,   protein LONGBLOB NOT NULL, protein_name VARCHAR(255),  ligand_pdbqt LONGBLOB,   ligand_smile VARCHAR(255),   ligand_name VARCHAR(255),   config LONGBLOB NOT NULL, date DATE, description VARCHAR(255),   done TINYINT DEFAULT 0)")
+except con.ProgrammingError:
+    print("Table Already Exists!")
+
+#try:
+#    mycursor.execute("insert into curieweb values (1,'navanchauhan@gmail.com','lu.pdbqt','test.pdbqt','owo.txt',CURDATE(),'CURIE WEB TASK',0)")
+#except con.IntegrityError:
+#    print("Duplicate Entry For Primary Key!")
+
+from random import choice, shuffle
+from string import digits, ascii_lowercase
+
+def gen_word(N, min_N_dig, min_N_low):
+    choose_from = [digits]*min_N_dig + [ascii_lowercase]*min_N_low
+    choose_from.extend([digits + ascii_lowercase] * (N-min_N_low-min_N_dig))
+    chars = [choice(bet) for bet in choose_from]
+    shuffle(chars)
+    return ''.join(chars)
+
+def convertToBinaryData(filename):
+    with open(filename, 'rb') as file:
+        binaryData = file.read()
+    return binaryData
+
+from os.path import basename
+
+ligand = convertToBinaryData("./files/Eucalyptol.pdbqt")
+receptor = convertToBinaryData("./files/6LU7.pdbqt")
+config = convertToBinaryData("./files/6LU7.txt")
+ligandName = "Eucalyptol"
+receptorName = "6LU7"
+
+sqlQuery = "insert into curieweb (id, email, protein, protein_name, ligand_pdbqt, ligand_name,date, config,done) values (%s,%s,%s,%s,%s,%s,CURDATE(),%s,%s) "
+
+jobID = gen_word(16, 1, 1)
+print("Succesfuly submitted Job ID:",jobID)
+
+insert_tuple = (jobID,"navanchauhan@gmail.com",receptor,receptorName,ligand,ligandName,config,done)
+
+try:
+    mycursor.execute(sqlQuery,insert_tuple)
+except con.IntegrityError:
+    print("Oops, Collision occured. Generating new Job ID and trying again.")
+    jobID = gen_word(16, 1, 1)
+    insert_tuple = (jobID,"navanchauhan@gmail.com",receptor,receptorName,ligand,ligandName,config,done)
+    mycursor.execute(sqlQuery,insert_tuple)
+
+print("Removing Test Query")
+
+q = 'delete from curieweb where id="%s"' % (jobID)
+mycursor.execute(q)
+
+mycon.commit()
+
+print("Database working perfectly")
+\ No newline at end of file