models.py 2.18 KB
from sqlalchemy.orm import sessionmaker

from .base_models import CommonModel
from tangsel.pbb.models import (Sppt as Invoice, PembayaranSppt as Pembayaran,
                                  Kelurahan, Kecamatan, Dati2 as Kabupaten,
                                  TempatPembayaran, Propinsi,
                                  DatObjekPajak as ObjekPajak)

session_factory = sessionmaker()
class Models:
    def __init__(
            self, tabel_dat_objek_pajak=True):
        # self.Base = Base
        # if payment_db_schema is None:
        #     payment_db_schema = db_schema

        # class Invoice(Base, CommonModel):
        #     __tablename__ = 'sppt'
        #     __table_args__ = dict(schema=db_schema, autoload=True)
        self.Invoice = Invoice

        # class Pembayaran(Base, CommonModel):
        #     __tablename__ = 'pembayaran_sppt'
        #     __table_args__ = dict(schema=payment_db_schema, autoload=True)
        self.Pembayaran = Pembayaran

        # class Kelurahan(Base, CommonModel):
        #     __tablename__ = 'ref_kelurahan'
        #     __table_args__ = dict(schema=db_schema, autoload=True)
        self.Kelurahan = Kelurahan

        # class Kecamatan(Base, CommonModel):
        #     __tablename__ = 'ref_kecamatan'
        #     __table_args__ = dict(schema=db_schema, autoload=True)
        self.Kecamatan = Kecamatan

        # class Kabupaten(Base, CommonModel):
        #     __tablename__ = 'ref_dati2'
        #     __table_args__ = dict(schema=db_schema, autoload=True)
        self.Kabupaten = Kabupaten

        # class Propinsi(Base, CommonModel):
        #     __tablename__ = 'ref_propinsi'
        #     __table_args__ = dict(schema=db_schema, autoload=True)
        self.Propinsi = Propinsi

        # class TempatPembayaran(Base, CommonModel):
        #     __tablename__ = 'tempat_pembayaran'
        #     __table_args__ = dict(schema=db_schema, autoload=True)
        self.TempatPembayaran = TempatPembayaran

        if tabel_dat_objek_pajak:
            # class ObjekPajak(Base, CommonModel):
            #     __tablename__ = 'dat_objek_pajak'
            #     __table_args__ = dict(schema=db_schema, autoload=True)
            self.ObjekPajak = ObjekPajak