webr2.py 3.14 KB
import sys
from sqlalchemy.exc import ProgrammingError
import transaction
from opensipkd.waktu import dmyhms
from opensipkd.iso8583.bjb.webr.structure import PAYMENT_CODE
from opensipkd.iso8583.bjb.webr.models import (
    LogMixin,
    Log,
    )
from iso8583_web.models.meta import Base as BaseConf
from ..models import (
    Base,
    Webr,
    )
from .common import (
    App2 as BaseApp,
    init_db as base_init_db,
    )


ERR_NOT_FOUND = 'Tgl {tgl_bayar} nomor bayar {invoice_id} tidak ada'


class AlternativeLog(Base, LogMixin):
    __table_args__ = dict(schema='webr')


class App(BaseApp):
    field_invoice_id = 'bit_061'
    field_ntb = 'bit_048'
    report_orm = Webr

    def __init__(self, argv):
        super().__init__(argv)
        if not self.pid:
            return
        self.base_q_iso_resp = self.prod_session.query(Log)
        self.log_orm = Log
        try:
            self.base_q_iso_resp.first()
        except ProgrammingError:
            self.prod_session.rollback()
            self.base_q_iso_resp = self.prod_session.query(AlternativeLog)
            self.log_orm = AlternativeLog
        self.base_q_iso_resp = self.base_q_iso_resp.filter_by(
                mti='0210', bit_003=PAYMENT_CODE)

    def is_iso_resp_ok(self, iso_req):
        q = self.base_q_iso_resp.filter_by(
                bit_011=iso_req.bit_011, bit_048=iso_req.bit_048)
        iso_resp = q.order_by(Log.id.desc()).first()
        return iso_resp and iso_resp.bit_039 == '00'

    def run_payment(self):
        last = self.get_last_id('webr2 payment last id')
        q_iso_req = self.prod_session.query(Log).filter_by(
                    mti='0210', bit_003=PAYMENT_CODE).filter(
                    Log.id > last.as_int())
        for iso_req in q_iso_req.order_by(Log.id):
            if self.get_report(iso_req):
                continue
            if not self.is_iso_resp_ok(iso_req):
                continue
            d = self.get_keys(iso_req)
            tgl_bayar = iso_req.created.date()
            s_tgl = dmyhms(iso_req.created)
            self.log.info(
                f'Tgl bayar {s_tgl}, Nomor bayar {d["nomor_bayar"]}, '
                f'STAN {d["stan"]}, NTB {d["ntb"]}, Channel {d["channel"]}')
            rpt = Webr(
                    stan=d['stan'], ntb=d['ntb'], tgl=tgl_bayar,
                    jam=iso_req.created.time(), payment_id=iso_req.id,
                    nomor_bayar=d['nomor_bayar'],
                    nama_wp=iso_req.bit_062_data['Nama Penyetor'],
                    pokok=iso_req.bit_004,
                    denda=0, jml_bayar=iso_req.bit_004,
                    channel_id=iso_req.bit_018.strip(),
                    channel_name=d['channel'])
            last.nilai = str(iso_req.id)
            with transaction.manager:
                self.rpt_session.add(rpt)
                self.rpt_session.add(last)

    def get_log_orm(self):  # Override
        return self.log_orm

    def run_reversal(self):  # Override
        super().run_reversal('webr2 reversal last id')


def main(argv=sys.argv[1:]):
    app = App(argv)
    if app.pid:
        app.run()


def init_db(argv=sys.argv[1:]):
    base_init_db(Base.metadata, BaseConf.metadata, argv)