Bug fixed log2db saat Postgres offline

payment_report/scripts/common.py

@@ -44,7 +44,8 @@ BIT_18_NAMES = {
     '6014': 'INTERNETBANKING',
     '6015': 'KIOSK',
     '6016': 'AUTODEBET',
-    '6017': 'MOBILBANKING'}
+    '6017': 'MOBILBANKING',
+    '7012': 'PTPOS'}
 
 # Bit 41, 42, dan 43
 BIT_PROFILE_NAMES = {
@@ -64,7 +65,7 @@ BIT_PROFILE_NAMES = {
     'BLIBLI': 'BLIBLI',
     'ARTPAY': 'ARTPAY'}
 
-BANK_NAMES = {'700': 'PT POS'}
+BANK_NAMES = {'700': 'PTPOS'}
 
 my_registry = dict()
 

payment_report/scripts/log2db.py

@@ -11,7 +11,10 @@ from watchdog.observers import Observer
 from watchdog.events import FileSystemEventHandler
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
-from sqlalchemy.exc import IntegrityError
+from sqlalchemy.exc import (
+    IntegrityError,
+    OperationalError,
+    )
 from ..log_models import LogFile
 from .tools import BacaFile
 
@@ -50,13 +53,21 @@ def read_log(log_file, db_url, tail_mode=True):
             return
         print([line])
         line_id = sha256(line.encode('utf-8')).hexdigest()
-        row = LogFile(line_id=line_id, line=line)
-        try:
-            db_session.add(row)
-            db_session.flush()
-            db_session.commit()
-        except IntegrityError:
-            db_session.rollback()
+        while True:
+            row = LogFile(line_id=line_id, line=line)
+            try:
+                db_session.add(row)
+                db_session.flush()
+                db_session.commit()
+                return
+            except IntegrityError:
+                db_session.rollback()
+                return
+            except OperationalError as e:
+                db_session.rollback()
+                print(e)
+                print('Tunggu, mungkin sedang maintenance ...')
+                sleep(10)
 
     engine = create_engine(db_url)
     factory = sessionmaker(bind=engine)

payment_report/scripts/pbb.py

@@ -119,6 +119,9 @@ class App(BaseApp):
         self.Sppt = self.models.Sppt
         self.base_q_sppt = self.prod_session.query(self.Sppt)
         self.nip_pospbb = self.conf.get('nip_pospbb')
+        self.set_other_db()
+
+    def set_other_db(self):
         if 'h2h_db_url' in self.conf:
             factory = self.get_factory('h2h_db_url')
             self.h2h_session = factory()