UI changes

AppCode/MatCreditHandler.py

@@ -6,30 +6,24 @@ import mysql.connector
 class MatCreditHandler:
 
     def __init__(self):
-        db = DBConfig()
-        self.conn = db.get_db_connection()
+        # db = DBConfig()
+        self.conn = DBConfig.get_db_connection()
         self.cursor = self.conn.cursor(dictionary=True)
 
-    @staticmethod
-    def fetch_all():
-        conn = DBConfig.get_db_connection()
-        cur = conn.cursor(dictionary=True)
-
+    # get all Mat credit data
+    def fetch_all(self):
         try:
-            # Stored Procedure returns TWO result sets
-            cur.callproc("GetMatCedit")
-
-            result_sets = cur.stored_results()
-
-            mat_rows = next(result_sets).fetchall()
-            utilization_rows = next(result_sets).fetchall()
-
-            return mat_rows, utilization_rows
+           self.cursor.callproc("GetMatCedit")
+           result_sets = self.cursor.stored_results()
+           mat_rows = next(result_sets).fetchall()
+           utilization_rows = next(result_sets).fetchall()
 
+           return mat_rows, utilization_rows
         finally:
-            cur.close()
-            conn.close()
+            self.cursor.close()
+            self.cursor.close()
 
+    # Save Mat credit data single row
     @staticmethod
     def save_single(data):
         conn = DBConfig.get_db_connection()
@@ -81,6 +75,7 @@ class MatCreditHandler:
             conn.close()
 
     
+    # save all Mat credit data 
     @staticmethod
     def save_bulk(rows):
         conn = DBConfig.get_db_connection()
@@ -120,4 +115,9 @@ class MatCreditHandler:
 
         finally:
             cur.close()
-            conn.close()
+            conn.close()
+
+    # CLOSE CONNECTION
+    def close(self):
+        self.cursor.close()
+        self.conn.close()