partner.py

from datetime import datetime
from pyramid.view import view_config
from pyramid.httpexceptions import HTTPFound
import colander
from deform import (
    Form,
    widget,
    ValidationFailure,
    Button
)
from opensipkd.base.views import ColumnDT, DataTables, BaseView
from ..models import DBSession
from ..models import Partner, User

SESS_ADD_FAILED = 'Tambah partner gagal'
SESS_EDIT_FAILED = 'Edit partner gagal'


def form_buttons():
    btn_register = Button(name='simpan', css_class='btn-success', type="submit")
    btn_cancel = Button(name='batal', css_class='btn-warning', type="submit")
    return [btn_cancel, btn_register]


def user_list():
    return DBSession.query(User.id, User.user_name).all()


@colander.deferred
def deferred_user(node, kw):  ##initialize deffered_status
    values = kw.get('daftar_user', [])  # varible for get value daftar_status
    return widget.Select2Widget(values=values)


class AddSchema(colander.Schema):
    kode = colander.SchemaNode(
        colander.String(),
        validator=colander.Length(max=32),
        oid="kode",
        title="Kode")
    nama = colander.SchemaNode(
        colander.String(),
        oid="nama")
    alamat_1 = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="alamat_1")
    alamat_2 = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="alamat_2")
    kelurahan = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="kelurahan")
    kecamatan = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="kecamatan")
    kota = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="kota")
    provinsi = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="provinsi")
    email = colander.SchemaNode(
        colander.String(),
        oid="nik")
    phone = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="nik")
    fax = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="nik")
    mobile = colander.SchemaNode(
        colander.String(),
        oid="nik")
    website = colander.SchemaNode(
        colander.String(),
        missing=colander.drop,
        oid="nik",
        title="Notif Callback URL")
    is_vendor = colander.SchemaNode(
        colander.Boolean(),
        oid="is_vendor",
        title="Vendor")
    is_customer = colander.SchemaNode(
        colander.Boolean(),
        oid="is_ccustomer",
        title="Customer")
    status = colander.SchemaNode(
        colander.Boolean(),
        oid="status")

    user_id = colander.SchemaNode(
        colander.Integer(),
        widget=deferred_user,
        oid="status")

    np_imid = colander.SchemaNode(
        colander.String(),
        oid="np_imid",
        missing=colander.drop,
        title="NicePay iMID"
        )
    np_key = colander.SchemaNode(
        colander.String(),
        oid="np_key",
        missing=colander.drop,
        title="NicePay Key"
        )


class EditSchema(AddSchema):
    id = colander.SchemaNode(colander.String(),
                             missing=colander.drop,
                             widget=widget.HiddenWidget(),
                             )


class ViewPartner(BaseView):
    ########
    # List #
    ########
    @view_config(route_name='api-partner-list', renderer='templates/partner/list.pt',
                 permission='partner')
    def view_list(self):
        return dict(a={})

    ##########
    # Action #
    ##########
    @view_config(route_name='api-partner-act', renderer='json',
                 permission='partner')
    def view_act(self):
        request = self.req
        ses = request.session
        params = request.params
        url_dict = request.matchdict

        if url_dict['act'] == 'grid':
            columns = [
                ColumnDT(Partner.id, mData='id'),
                ColumnDT(Partner.kode, mData='kode'),
                ColumnDT(Partner.nama, mData='nama'),
                ColumnDT(Partner.is_vendor, mData='is_vendor'),
                ColumnDT(Partner.is_customer, mData='is_customer'),
                ColumnDT(Partner.status, mData='status'),
            ]
            query = DBSession.query().select_from(Partner)
            rowTable = DataTables(request.GET, query, columns)
            return rowTable.output_result()

        elif url_dict['act'] == 'hok':
            term = 'term' in params and params['term'] or ''
            prefix = 'prefix' in params and params['prefix'] or ''
            q = DBSession.query(Partner.id, Partner.kode.label('value'),
                                Partner.kode, Partner.nama). \
                filter(Partner.nama.ilike('%%%s%%' % term)). \
                filter(Partner.is_vendor == 1). \
                order_by(Partner.nama)
            row = q.first()
            r = []
            if row:
                keys = row.keys()
                for k in q.all():
                    d = dict(zip(keys, k))
                    r.append(d)
            return r

        elif url_dict['act'] == 'hon':
            term = 'term' in params and params['term'] or ''
            prefix = 'prefix' in params and params['prefix'] or ''
            q = DBSession.query(Partner.id, Partner.nama.label('value'),
                                Partner.kode, Partner.nama). \
                filter(Partner.nama.ilike('%%%s%%' % term)). \
                order_by(Partner.nama)
            r = []
            row = q.first()
            if not row:
                return r

            keys = row.keys()
            for k in q.all():
                d = dict(zip(keys, k))
                r.append(d)
            return r

        elif url_dict['act'] == 'vendor':  # vendor only
            term = 'term' in params and params['term'] or ''
            prefix = 'prefix' in params and params['prefix'] or ''
            q = DBSession.query(Partner.id, Partner.nama.label('value'),
                                Partner.kode, Partner.nama). \
                filter(Partner.nama.ilike('%%%s%%' % term)). \
                filter(Partner.is_vendor == 1). \
                order_by(Partner.nama)
            row = q.first()
            r = []
            if row:
                keys = row.keys()
                for k in q.all():
                    d = dict(zip(keys, k))
                    r.append(d)
            return r

        elif url_dict['act'] == 'customer':  # customer only
            term = 'term' in params and params['term'] or ''
            prefix = 'prefix' in params and params['prefix'] or ''
            q = DBSession.query(Partner.id, Partner.nama.label('value'),
                                Partner.kode, Partner.nama). \
                filter(Partner.nama.ilike('%%%s%%' % term)). \
                filter(Partner.is_customer == 1). \
                order_by(Partner.nama)
            row = q.first()
            r = []
            if row:
                keys = row.keys()
                for k in q.all():
                    d = dict(zip(keys, k))
                    r.append(d)
            return r

    @view_config(route_name='api-partner-add', renderer='templates/partner/add.pt',
                 permission='partner-add')
    def view_add(self):
        request = self.req
        form = get_form(request, AddSchema, buttons=form_buttons())
        if request.POST:
            if 'simpan' in request.POST:
                controls = request.POST.items()
                try:
                    controls = form.validate(controls)
                except ValidationFailure as e:
                    form.render(appstruct=e.cstruct)
                    return dict(form=form)
                save_request(request, dict(controls))
            return route_list(request)
        elif SESS_ADD_FAILED in request.session:
            return self.session_failed(SESS_ADD_FAILED)
        return dict(form=form)
        # return dict(form=form.render())

    ##########
    # Edit #
    ##########
    @view_config(route_name='api-partner-edit', renderer='templates/partner/edt.pt',
                 permission='partner-edit')
    def view_edt(self):
        request = self.req
        q = query_id(request)
        row = q.first()
        if not row:
            return id_not_found(request)
        uid = row.id

        form = get_form(request, EditSchema, buttons=form_buttons())
        if request.POST:
            if 'simpan' in request.POST:
                controls = request.POST.items()
                try:
                    controls = form.validate(controls)
                except ValidationFailure as e:
                    form.render(appstruct=e.cstruct)
                    return dict(form=form)

                save_request(request, dict(controls), row)
            return route_list(request)
        elif SESS_EDIT_FAILED in request.session:
            return self.session_failed(SESS_EDIT_FAILED)
        val = row.to_dict()
        values = {}
        for v in val:
            if val[v]:
                values[v] = val[v]
        form.render(appstruct=values)
        return dict(form=form)

    ##########
    # Delete #
    ##########
    @view_config(route_name='api-partner-delete', renderer='templates/partner/del.pt',
                 permission='partner-del')
    def view_del(self):
        request = self.req
        q = query_id(request)
        row = q.first()

        if not row:
            return id_not_found(request)

        form = Form(colander.Schema(), buttons=('hapus', 'batal'))
        if request.POST:
            if 'hapus' in request.POST:
                msg = 'Partner ID %d %s sudah dihapus.' % (row.id, row.nama)
                q.delete()
                DBSession.flush()
                request.session.flash(msg)
            return route_list(request)
        return dict(row=row, form=form.render())


#######
# Add #
#######
def form_validator(form, value):
    def err_kode():
        raise colander.Invalid(form,
                               'Kode %s sudah digunakan oleh %s' % (
                                   value['kode'], found.nama))

    # def err_login():
    # raise colander.Invalid(form,
    # 'Login %s sudah digunakan oleh kode %s' % (
    # value['user_nm'], found.row.nama))

    if 'id' in form.request.matchdict:
        uid = form.request.matchdict['id']
        q = DBSession.query(Partner).filter_by(id=uid)
        row = q.first()
    else:
        row = None

    q = Partner.query_kode(value['kode'])  # DBSession.query(Partner).filter_by(kode=value['kode'])
    found = q.first()
    if row:
        if found and found.id != row.id:
            err_kode()
    elif found:
        err_kode()


def get_form(request, class_form, buttons=None, row=None):
    schema = class_form(validator=form_validator)
    schema = schema.bind(daftar_user=user_list())
    schema.request = request
    if row:
        schema.deserialize(row)
    if not buttons:
        buttons = ('simpan', 'batal')
    return Form(schema, buttons=buttons)


def save(values, user, row=None):
    if not row:
        row = Partner()
        row.created = datetime.now()
        row.create_uid = user.id
    values['is_vendor'] = 'is_vendor' in values and values['is_vendor'] and 1 or 0
    values['is_customer'] = 'is_customer' in values and values['is_customer'] and 1 or 0
    row.from_dict(values)
    row.updated = datetime.now()
    row.update_uid = user.id
    row.status = 'status' in values and values['status'] and 1 or 0
    DBSession.add(row)
    DBSession.flush()
    return row


def save_request(request, values, row=None):
    if 'id' in request.matchdict:
        values['id'] = request.matchdict['id']
    row = save(values, request.user, row)
    request.session.flash('Partner sudah disimpan.')


def route_list(request):
    return HTTPFound(location=request.route_url('api-partner-list'))


def session_failed(request, session_name):
    r = dict(form=request.session[session_name])
    del request.session[session_name]
    return r


########
# Edit #
########
def query_id(request):
    return DBSession.query(Partner).filter_by(id=request.matchdict['id'])


def id_not_found(request):
    msg = 'Partner ID %s Tidak Ditemukan.' % request.matchdict['id']
    request.session.flash(msg, 'error')
    return route_list(request)