From 09b3bb38fc7305c9b47c29bc90ebc9c636827307 Mon Sep 17 00:00:00 2001
From: Hector Fernandez <hector@iatec.biz>
Date: Wed, 17 Jun 2020 09:44:56 +0200
Subject: SOPA0000: Added support for saving results (strings or files) in the
 DB. It also saves a final DMESG file.

---
 test-cli/test/helpers/psqldb.py | 29 ++++++++++++++++++-----------
 1 file changed, 18 insertions(+), 11 deletions(-)

(limited to 'test-cli/test/helpers/psqldb.py')

diff --git a/test-cli/test/helpers/psqldb.py b/test-cli/test/helpers/psqldb.py
index 1d0e422..c98338a 100644
--- a/test-cli/test/helpers/psqldb.py
+++ b/test-cli/test/helpers/psqldb.py
@@ -1,29 +1,30 @@
 import psycopg2
 
+
 class PgSQLConnection(object):
     """aaaaaaa"""
 
     __conection_object = None
     __db_config = {'dbname': 'testsrv', 'host': '192.168.2.171',
-                 'password': 'Idkfa2009', 'port': 5432, 'user': 'admin'}
+                   'password': 'Idkfa2009', 'port': 5432, 'user': 'admin'}
 
-    def __init__ (self, connect_str = None):
+    def __init__(self, connect_str=None):
         self.__conection_object = None
         if connect_str is not None:
             self.__db_config = connect_str
         else:
             self.__db_config = {'dbname': 'testsrv', 'host': '192.168.2.171',
-                 'password': 'Idkfa2009', 'port': 5432, 'user': 'admin'}
+                                'password': 'Idkfa2009', 'port': 5432, 'user': 'admin'}
 
-    def db_connect (self, connect_str = None):
+    def db_connect(self, connect_str=None):
         result = False
         try:
             if connect_str == None:
                 self.__conection_object = psycopg2.connect(**self.__db_config)
             else:
-                self.__db_config = connect_str;
+                self.__db_config = connect_str
                 self.__conection_object = psycopg2.connect(**self.__db_config)
-                
+
             self.__conection_object.autocommit = True
             result = True
         except Exception as error:
@@ -37,20 +38,26 @@ class PgSQLConnection(object):
         cur.close()
         return data
 
+    def db_upload_large_file(self, filepath):
+        # a new connection must be created to use large objects
+        __conn = psycopg2.connect(**self.__db_config)
+        lobj = __conn.lobject(oid=0, mode="rw", new_oid=0, new_file=filepath)
+        fileoid = lobj.oid
+        lobj.close()
+        __conn.commit()
+        __conn.close()
+        return fileoid
+
     def commit(self):
         self.__conection_object.commit()
 
-
     def db_close(self):
         if self.__conection_object is not None:
             self.__conection_object.close()
             del self.__conection_object
             self.__conection_object = None
 
-
     def __del__(self):
         if self.__conection_object is not None:
             self.__conection_object.close()
-            #print("disconnected")
-
-
+            # print("disconnected")
-- 
cgit v1.1