prove

CsvLoader.py

@@ -1,4 +1,4 @@
-#!/usr/bin/python3
+#!.venv/bin/python
 
 import sys
 import os
@@ -6,11 +6,11 @@ import pika
 import logging
 import csv
 import re
-import mariadb
+import mysql.connector as mysql
 import shutil
 
-from utils.timefmt import timestamp_fmt as ts
-from utils.timefmt import date_refmt as df
+from utils.time import timestamp_fmt as ts
+from utils.time import date_refmt as df
 from utils.config import set_config as setting
 
 
@@ -34,23 +34,19 @@ class sqlraw:
 
     def write_db(self):
         try:
-            conn = mariadb.connect(**self.config, database=self.dbname)
-        except mariadb.Error as err:
+            conn = mysql.connect(**self.config, database=self.dbname)
+        except Exception as err:
             logging.error(
-                "PID {:>5} >> Error to connet to DB {} - System error {}.".format(
-                    os.getpid(), self.dbname, err
-                )
+                f"PID {os.getpid():>5} >> Error to connet to DB {self.dbname} - System error {err}."
             )
             sys.exit(1)
 
         cur = conn.cursor()
         try:
             cur.execute(self.sql)
-        except mariadb.ProgrammingError as err:
+        except Exception as err:
             logging.error(
-                "PID {:>5} >> Error write into DB {} - System error {}.".format(
-                    os.getpid(), self.dbname, err
-                )
+                f"PID {os.getpid():>5} >> Error write into DB {self.dbname} - System error {err}."
             )
             print(err)
             sys.exit(1)