diff --git a/app/misc.ini b/app/misc.ini
index e9ce9b6..875fa7d 100644
--- a/app/misc.ini
+++ b/app/misc.ini
@@ -5,4 +5,8 @@ OD02 = Failed to add charges to molecule
OD03 = Failed to write the converted PDBQT file
PL02 = Failed to import PLIP, please ensure that you have correctly installed PLIP
-PL01 = Failed to fetch the PDB, please check the PDB Code
\ No newline at end of file
+PL01 = Failed to fetch the PDB, please check the PDB Code
+
+DB01 = This Job ID does not exist 😠. If you think this is an error, please contact us.
+
+CW01 = Looks like an invalid PDB ID
\ No newline at end of file
diff --git a/app/templates/error.html b/app/templates/error.html
index 3073037..b605430 100644
--- a/app/templates/error.html
+++ b/app/templates/error.html
@@ -5,7 +5,7 @@
Oh Snap! An error occured
Error Code: {{code}}
Error Description: {{description}}
-
+
+
{% endblock %}
\ No newline at end of file
diff --git a/app/templates/job_status_error.html b/app/templates/job_status_error.html
deleted file mode 100644
index b6d16a7..0000000
--- a/app/templates/job_status_error.html
+++ /dev/null
@@ -1,15 +0,0 @@
-{% extends 'base.html' %}
-
-{% block main %}
- Job ID: {{ job }}
- This Job ID does not exist 😠. If you think this is an error, please contact us.
-
-{% endblock %}
\ No newline at end of file
diff --git a/app/views.py b/app/views.py
index ef3c2ba..dfd33e2 100644
--- a/app/views.py
+++ b/app/views.py
@@ -114,7 +114,7 @@ def status():
mycursor.execute(sqlQuery)
records = mycursor.fetchall()
if records == []:
- return render_template('job_status_error.html',job=jobID)
+ return render_template('error.html',code="DB01",description=errors['DB01'])
r = records[0]
protein_name = r[1]
ligand_name = r[2]
@@ -346,6 +346,9 @@ def dock_upload_single():
name = form.name.data
email = form.email.data
+ if len(pdb) != 4:
+ return render_template("error.html",code="CW01",description=errors['CW01'])
+
import mysql.connector as con
mycon = con.connect(host=app.config['DB_HOST'],user=app.config['DB_USER'],password=app.config['DB_PASSWORD'],port=app.config['DB_PORT'],database=app.config['DB_NAME'])
mycursor = mycon.cursor()
@@ -356,7 +359,7 @@ def dock_upload_single():
insert_tuple = (jobID,email,pdb,smile,name,description)
mycursor.execute(sqlQuery,insert_tuple)
mycon.commit()
-
+
log(("Description",description),"DEBUG")
cwd = os.path.join(os.getcwd(),"app")