bphtb.py 2.92 KB
import sys
import transaction
from opensipkd.waktu import dmyhms
from opensipkd.bphtb.models.default import (
    IsoPayment,
    IsoReversal,
    Payment,
    Invoice,
    Perolehan,
    Customer,
    )
from opensipkd.bphtb.services.base import get_db_session
from iso8583_web.models.meta import Base as BaseConf
from opensipkd.iso8583.bjb.bphtb import Doc
from ..models import (
    Base,
    Bphtb,
    )
from .common import (
    get_iso,
    get_keys,
    App as BaseApp,
    init_db as base_init_db,
    )


class App(BaseApp):
    iso_class = Doc
    report_orm = Bphtb
    iso_reversal_orm = IsoReversal

    def __init__(self, argv):
        super().__init__(argv)
        if not self.pid:
            return
        self.base_q_inv = self.prod_session.query(
                Invoice, Customer, Perolehan).filter(
                Invoice.ppat_id == Customer.id,
                Invoice.perolehan_id == Perolehan.id)

    def get_db_session(self):  # Override
        return get_db_session()

    def run_payment(self):  # Override
        last = self.get_last_id('bphtb payment last id')
        q_iso = self.prod_session.query(IsoPayment, Payment).filter(
                IsoPayment.id == Payment.id,
                IsoPayment.id > last.as_int())
        for row_iso, row_pay in q_iso.order_by(IsoPayment.id):
            iso = get_iso(row_iso.iso_request, Doc, self.option.debug)
            if self.get_report(iso):
                continue
            tgl_bayar = row_iso.tgl.date()
            q_inv = self.base_q_inv.filter(Invoice.id == row_pay.sspd_id)
            inv, ppat, perolehan = q_inv.first()
            d = get_keys(iso)
            s_tgl = dmyhms(row_iso.tgl)
            self.log.info(
                f'Tgl bayar {s_tgl}, Nomor bayar {d["nomor_bayar"]}, '
                f'STAN {d["stan"]}, NTB {d["ntb"]}, Channel {d["channel"]}')
            rpt = Bphtb(
                    stan=d['stan'], ntb=d['ntb'], tgl=tgl_bayar,
                    jam=row_iso.tgl.time(),
                    invoice_id=iso.get_invoice_id().strip(),
                    nop=row_pay.nop, wp_nama=row_pay.wp_nama,
                    wp_alamat=row_pay.wp_alamat, op_alamat=inv.op_alamat,
                    npop=row_pay.npop, bumi_luas=row_pay.bumi_luas,
                    bng_luas=row_pay.bng_luas, nilai_bphtb=row_pay.bayar,
                    jenis_perolehan=perolehan.nama, ppat=ppat.nama,
                    channel_id=iso.get_channel().strip(),
                    channel_nama=d['channel'])
            last.nilai = str(row_iso.id)
            print(f'last.nilai {last.nilai}')
            with transaction.manager:
                self.rpt_session.add(rpt)
                self.rpt_session.add(last)

    def run_reversal(self):  # Override
        super().run_reversal('bphtb reversal last date')


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)