From b296e4d2771088d2fd567eca19c2696d8248dbd8 Mon Sep 17 00:00:00 2001 From: Navan Chauhan Date: Tue, 7 Jul 2020 11:41:42 +0530 Subject: Moved tests in a seperate folder --- tests/dbTestFiller.py | 69 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 69 insertions(+) create mode 100644 tests/dbTestFiller.py (limited to 'tests/dbTestFiller.py') 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 -- cgit v1.2.3