From a4c06312c4158b0054ae1d21c3cdd3d3aa632d50 Mon Sep 17 00:00:00 2001
From: manojjahgirdar <manoj.jahgirdar@in.ibm.com>
Date: Tue, 18 Jan 2022 11:38:03 +0530
Subject: [PATCH] Updated code to incorporate ssl authentication with Db2

---
 app.py | 54 ++++++++++++++++++++++++++++++++++--------------------
 1 file changed, 34 insertions(+), 20 deletions(-)

diff --git a/app.py b/app.py
index 0724b52..3a1d15d 100644
--- a/app.py
+++ b/app.py
@@ -4,6 +4,7 @@
 import requests
 from ibm_watson import AssistantV2
 from ibm_cloud_sdk_core.authenticators import IAMAuthenticator
+import base64
 try:
     import ibm_db
 except:
@@ -22,25 +23,36 @@
 with open('ibm-db2-credentials.json', 'r') as credentialsFile:
     credentials1 = json.loads(credentialsFile.read())
 
-dsn_driver = "IBM DB2 ODBC DRIVER"
-dsn_database = credentials1['db']
-dsn_hostname = credentials1['host']
-dsn_port = "50000"
-dsn_uid = credentials1['username']
-dsn_pwd = credentials1['password']
-
-dsn = (
-    "DRIVER={{IBM DB2 ODBC DRIVER}};"
-    "DATABASE={0};"
-    "HOSTNAME={1};"
-    "PORT={2};"
-    "PROTOCOL=TCPIP;"
-    "UID={3};"
-    "PWD={4};").format(dsn_database, dsn_hostname, dsn_port, dsn_uid, dsn_pwd)
 try:
-    conn = ibm_db.connect(dsn, "", "")
+    dsn_database = credentials1['connection']['db2']['database']
+    dsn_hostname = credentials1['connection']['db2']['hosts'][0]['hostname']
+    dsn_port = credentials1['connection']['db2']['hosts'][0]['port']
+    dsn_uid = credentials1['connection']['db2']['authentication']['username']
+    dsn_pwd = credentials1['connection']['db2']['authentication']['password']
+
+    certificate = credentials1['connection']['db2']['certificate']['certificate_base64']
+    ssl_certificate_bytes = base64.b64decode(certificate)
+    ssl_certificate = ssl_certificate_bytes.decode('ascii')
 except:
-    pass
+    print('No DB2 credentials found')
+
+db2_conn_cert = os.path.join(os.path.expanduser('~'),'ibm-db2-ssl.cert')
+with open(db2_conn_cert, "w") as f:
+    f.write(ssl_certificate)
+
+dsn = 'DATABASE={db};HOSTNAME={host};PORT={port};PROTOCOL=TCPIP;UID={uid};PWD={pwd};SECURITY=SSL;SSLServerCertificate={cert}'.format(
+    db=dsn_database,
+    host=dsn_hostname,
+    port=dsn_port,
+    uid=dsn_uid,
+    pwd=dsn_pwd,
+    cert=db2_conn_cert
+)
+
+try:
+    conn = ibm_db.connect(dsn, "", "")
+except Exception as e:
+    print(e)
 
 #########################
 # Create Orders Table
@@ -52,8 +64,8 @@
     ADDRESS varchar(255)	); ' 
 try:
     ibm_db.exec_immediate(conn, table)
-except:
-    pass
+except Exception as e:
+    print("Create Table error-> ",e)
 
 
 with open('watson-assistant-credentials.json', 'r') as credentialsFile:
@@ -93,7 +105,7 @@ def destroySession():
         assistant_id=assistantid, session_id=sessionid).get_result()
         print(response)
     except Exception as e:
-        pass
+        print(e)
 
 # assistant.delete_session(skillid, "<YOUR SESSION ID>").get_result()
 
@@ -234,3 +246,5 @@ def index():
 if __name__ == "__main__":
     app.secret_key = os.urandom(12)
     app.run(debug=True, host='0.0.0.0', port=port)
+
+    
\ No newline at end of file