From 52638600e201c80bd3589f0ccefca72f8f9e6ba1 Mon Sep 17 00:00:00 2001
From: Navan Chauhan <navanchauhan@gmail.com>
Date: Sun, 27 Sep 2020 14:10:20 +0530
Subject: moved configuration to misc
---
app/dock-manual.py | 15 ++++-----------
app/dock-single.py | 11 ++---------
2 files changed, 6 insertions(+), 20 deletions(-)
(limited to 'app')
diff --git a/app/dock-manual.py b/app/dock-manual.py
index 442b76d..2d69294 100644
--- a/app/dock-manual.py
+++ b/app/dock-manual.py
@@ -2,16 +2,9 @@ import mysql.connector as con
from misc.common import get3DModel, CopyContentOfFolder, RemoveAllFilesMatching
from misc.email import email
import os
-import configparser
import sys
-iniConfig = configparser.ConfigParser()
-iniConfig.read('config.ini')
-
-try:
- iniConfig['DATABASE']
-except KeyError:
- iniConfig.read("../config.ini")
+from misc.config import iniConfig
mycon = con.connect(host=iniConfig['DATABASE']['HOST'],user=iniConfig['DATABASE']['USER'],password=iniConfig['DATABASE']['PASSWORD'],port=iniConfig['DATABASE']['PORT'],database=iniConfig['DATABASE']['NAME'])
mycursor = mycon.cursor()
@@ -76,7 +69,7 @@ with tempfile.TemporaryDirectory() as directory:
make_archive(zi, 'zip', directory)
try:
copyfile("report.pdf",os.path.join(reportDirectory,(str(jobID)+".pdf")))
- except:
+ except FileNotFoundError:
reason = "Could not generate the report, this could be because of a failed docking job. Please check the ZIP archive for the configuration and converted PDBQTs and try submitting manually. "
email(toaddr,jobID,date,description,zipArchive=zi,reason=reason)
mycursor.execute('UPDATE curieweb set done=1 where id="%s"' % (jobID))
@@ -92,7 +85,7 @@ with tempfile.TemporaryDirectory() as directory:
os.system("collada2gltf -i model.dae -o model.gltf")
try:
copyfile("model.gltf",os.path.join(modelDirectory,(str(jobID)+".gltf")))
- except:
+ except FileNotFoundError:
print("Does not have Collada2GLTF Installed")
email(toaddr,jobID,date,description,zipArchive=zi)
mycursor.execute('UPDATE curieweb set done=1 where id="%s"' % (jobID))
@@ -106,7 +99,7 @@ with tempfile.TemporaryDirectory() as directory:
os.system("docker run -it --rm -v $(pwd):/usr/app navanchauhan/usd-from-gltf:latest model.gltf model.usdz")
try:
copyfile("model.usdz",os.path.join(modelDirectory,(str(jobID)+".usdz")))
- except:
+ except FileNotFoundError:
print("Could not generate USDZ file")
email(toaddr,jobID,date,description,zipArchive=zi)
mycursor.execute('UPDATE curieweb set done=1 where id="%s"' % (jobID))
diff --git a/app/dock-single.py b/app/dock-single.py
index 54ade78..c09d262 100644
--- a/app/dock-single.py
+++ b/app/dock-single.py
@@ -5,17 +5,10 @@ import os
import sys
from argparse import ArgumentParser
from collections import namedtuple
-from misc.common import get3DModel, CopyContentOfFolder, RemoveAllFilesMatching
import mysql.connector as con
+from misc.common import get3DModel, CopyContentOfFolder, RemoveAllFilesMatching
from misc.email import email
-import configparser
-iniConfig = configparser.ConfigParser()
-iniConfig.read('config.ini')
-
-try:
- iniConfig['DATABASE']
-except KeyError:
- iniConfig.read("../config.ini")
+from misc.config import iniConfig
mycon = con.connect(host=iniConfig['DATABASE']['HOST'],user=iniConfig['DATABASE']['USER'],password=iniConfig['DATABASE']['PASSWORD'],port=iniConfig['DATABASE']['PORT'],database=iniConfig['DATABASE']['NAME'])
mycursor = mycon.cursor()
--
cgit v1.2.3