fix: input cursor ref without connection

Sieboldianus · Jan 11, 2019 · 6b01fc5 · 6b01fc5
1 parent bfb3bb0
commit 6b01fc5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/lbsntransform/__main__.py b/lbsntransform/__main__.py
@@ -54,8 +54,7 @@ def main():
         loc_filelist = LoadData.read_local_files(config)
     else:
         # establish input connection
-        db_input = LoadData.initialize_input_connection(config)
-        cursor_input = db_input[1]
+        cursor_input = LoadData.initialize_input_connection(config)
 
     # start settings
     processed_records = 0

diff --git a/lbsntransform/classes/load_data.py b/lbsntransform/classes/load_data.py
@@ -182,7 +182,7 @@ def initialize_input_connection(cfg):
                                         cfg.dbpassword_input,
                                         True  # ReadOnly Mode
                                         )
-        cursor_input = input_connection.connect()
+        conn_input, cursor_input = input_connection.connect()
         return cursor_input
 
     @staticmethod

diff --git a/lbsntransform/classes/store_csv.py b/lbsntransform/classes/store_csv.py
@@ -119,7 +119,7 @@ def clean_csv_batches(self, batched_records):
                 self.sort_files(filelist)
                 if len(filelist) > 1:
                     print(
-                        f'Cleaning & merging output files..{x}'
+                        f'Cleaning & merging output files..{x_cnt}'
                         f'/{len(batched_records)}', end='\r')
                     self.merge_files(filelist, type_name)
                 else: