posisi.py

import colander
import os
from opensipkd.tools.report import csv_response, open_rml_pdf, open_rml_row, pdf_response, file_response
from deform import widget
from pyramid.view import view_config
from sqlalchemy import or_
from sqlalchemy.orm import aliased
from opensipkd.tools import get_random_string

from opensipkd.models import DBSession as PartnerDBSession, DBSession, \
    ResCompany
from opensipkd.models import Departemen, Jabatan
from opensipkd.models import Partner, PartnerDepartemen
from opensipkd.tools import dmy, date_from_str
from . import widget_os, button_import
from ..views import ColumnDT, DataTables, BaseView
from pyramid.path import AssetResolver
from . import BaseView, button_import, get_params

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

class AddSchema(colander.Schema):
    nama_widget = widget.AutocompleteInputWidget(
        size=60,
        values='/partner/hon/act',
        min_length=2,
        style="z-index: 100000 !important;")

    @colander.deferred
    def departemen_widget(node, kw):
        departemen_list = DBSession.query(Departemen.id, Departemen.nama).all()
        choices = [(str(departemen.id), departemen.nama) for departemen in departemen_list]
        return widget.CheckboxChoiceWidget(values=choices, multiple=True)  # Enable multiple selection

    @colander.deferred
    def jabatan_widget(node, kw):
        jabatan_list = DBSession.query(Jabatan.id, Jabatan.nama).all()
        choices = [('', '- Pilih Jabatan -')] + [(str(jabatan.id), jabatan.nama) for jabatan in jabatan_list]
        return widget.SelectWidget(values=choices)

    partner_id = colander.SchemaNode(
        colander.Integer(),
        oid="partner_id",
        widget=widget.HiddenWidget(),
        title="Partner ID")
    nama = colander.SchemaNode(
        colander.String(),
        widget=nama_widget,
        oid="partner_nm",
        title="Nama")
    departemen_id = colander.SchemaNode(
        colander.Integer(),
        oid="departemen_id",
        widget=widget.HiddenWidget(),
        missing=colander.drop)
    departemen = colander.SchemaNode(
        colander.Set(),  # Menggunakan Set untuk multiple values
        widget=departemen_widget,
        oid="departemen_nm",
        title="Departemen *")
    jabatan_id = colander.SchemaNode(
        colander.Integer(),
        oid="jabatan_id",
        widget=jabatan_widget,
        title="Jabatan *",
        missing=colander.drop)
    mulai = colander.SchemaNode(
        colander.String(),
        oid="mulai",
        widget=widget_os.DateInputWidget(css_class="date", format="dd-mm-yyyy"),
        title="Mulai")
    selesai = colander.SchemaNode(
        colander.String(),
        oid="selesai",
        widget=widget_os.DateInputWidget(css_class="date", format="dd-mm-yyyy",
                                         style="z-index: 9999 !important;"),
        title="Selesai")

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

class ListSchema(colander.Schema):
    id = colander.SchemaNode(colander.Integer(), title="Action")
    nama = colander.SchemaNode(
        colander.String(), field=Partner.nama)
    mulai = colander.SchemaNode(
        colander.String())
    selesai = colander.SchemaNode(
        colander.String())

class ViewPartner(BaseView):
    def __init__(self, request):
        super(ViewPartner, self).__init__(request)
        self.list_schema = ListSchema
        self.table = PartnerDepartemen
        self.add_schema = AddSchema
        self.edit_schema = EditSchema
        self.list_route = 'partner-departemen'
        self.list_buttons = self.list_buttons + self.list_report + (button_import,)
        path = os.path.dirname(__file__)
        path = os.path.dirname(path)
        self.report_file = os.path.join(path, 'reports', 'partnerdep.jrxml')

    def form_validator(self, form, value):
        print(f"Validating form with values: {value}")  # Debugging
        def err_departemen():
            raise colander.Invalid(form, 'Departemen Kerja Belum Dipilih')

        def err_jabatan():
            raise colander.Invalid(form, 'Jabatan Belum Dipilih')

        def err_partner():
            raise colander.Invalid(form, 'Partner Belum Dipilih')

        # Validasi partner_id
        if not value['partner_id']:
            print("Validation failed: Partner ID kosong")
            err_partner()

        # Validasi jabatan_id
        if not value['jabatan_id']:
            print("Validation failed: Jabatan kosong")
            err_jabatan()

        # Validasi departemen
        if not value['departemen'] or value['departemen'] == set():
            print("Validation failed: Departemen kosong")
            err_departemen()

    def save_request(self, values, row=None):
        print(f"Saving with values: {values}")  # Debugging
        request = self.req
        if 'id' in request.matchdict:
            values['id'] = request.matchdict['id']
        if 'mulai' in values:
            values['mulai'] = date_from_str(values['mulai']) if values['mulai'] else None
        if 'selesai' in values:
            values['selesai'] = date_from_str(values['selesai']) if values['selesai'] else None

        # Ambil jabatan_id dari form
        if 'jabatan_id' in values and values['jabatan_id']:
            values['jabatan_id'] = int(values['jabatan_id'])  # Konversi ke integer
        else:
            values['jabatan_id'] = None

        # Pastikan partner_id ada dan valid
        partner_id = values.get('partner_id')
        if not partner_id:
            request.session.flash('Partner ID wajib diisi.', 'error')
            return None

        # Simpan setiap departemen yang dipilih sebagai baris baru
        if 'departemen' in values and values['departemen']:
            # Jika sedang mengedit, hapus entri lama untuk partner_id dan jabatan_id ini
            if 'id' in values:
                DBSession.query(PartnerDepartemen).filter_by(
                    partner_id=partner_id,
                    jabatan_id=values['jabatan_id']
                ).delete()

            # Tambahkan entri baru untuk setiap departemen yang dipilih
            for dept_id in values['departemen']:
                new_row = PartnerDepartemen(
                    partner_id=partner_id,
                    departemen_id=int(dept_id),
                    jabatan_id=values['jabatan_id'],
                    mulai=values['mulai'],
                    selesai=values['selesai']
                )
                DBSession.add(new_row)
            DBSession.flush()
            request.session.flash('Posisi Partner sudah disimpan.')
            return None
        else:
            request.session.flash('Tidak ada departemen yang dipilih.', 'error')
            return None

    @view_config(route_name='partner-departemen', renderer='templates/table.pt',
                 permission='view')
    def view_list(self):
        return super().view_list()
        
    def get_module_path(self, module_name):
        a = AssetResolver(module_name)
        resolver = a.resolve('')
        return resolver.abspath()

    def pdf_response(self, **kwargs):
        # print("PDF RESPONSE======================================")
        from opensipkd.base.tools.report import jasper_export
        logo = self.get_module_path('opensipkd.base')
        logo = os.path.join(logo, 'static', 'img', 'logo.png')
        parameters = {
            "judul": get_params('company', "openSIPKD"), 
            "logo": logo
            }
        print(parameters)
        filename = jasper_export(self.report_file, parameters=parameters)
        return file_response(self.req, filename=filename[0])
        
    @staticmethod
    def query_register():
        return DBSession.query(PartnerDepartemen.partner_id, PartnerDepartemen.departemen_id, PartnerDepartemen.departemen, PartnerDepartemen.jabatan_id, PartnerDepartemen.mulai, PartnerDepartemen.selesai).order_by(PartnerDepartemen.partner_id)

    def csv_response(self, **kwargs):
        query = self.query_register()
        row = query.first()
        header = row._mapping.keys()
        rows = [list(item) for item in query.all()]
        filename = f"{get_random_string(16)}.csv"
        value = {
            'header': header,
            'rows': rows,
        }
        return csv_response(self.req, value, filename)

    @view_config(route_name='partner-departemen-act', renderer='json',
                 permission='view')
    def view_act(self):
        # request = self.req
        # ses = request.session
        # params = request.params
        # url_dict = request.matchdict
        # struktural = aliased(Jabatan)
        # if url_dict['act'] == 'grid':
            # columns = [
                # ColumnDT(PartnerDepartemen.id, mData='id'),
                # ColumnDT(Partner.kode, mData='nik'),
                # ColumnDT(Partner.nama, mData='nama'),
                # ColumnDT(Departemen.nama, mData='departemen'),
                # ColumnDT(Jabatan.nama, mData='jabatan'),
                # ColumnDT(ResCompany.nama, mData='jenis'),
                # ColumnDT(struktural.nama, mData='struktural_nm'),
                # ColumnDT(PartnerDepartemen.mulai, mData='mulai'),
                # ColumnDT(PartnerDepartemen.selesai, mData='selesai'),
            # ]
            # q = PartnerDBSession.query().select_from(PartnerDepartemen) \
                # .join(Departemen,
                      # PartnerDepartemen.departemen_id == Departemen.id) \
                # .outerjoin(ResCompany, Departemen.company_id == ResCompany.id) \
                # .outerjoin(Partner, Partner.id == PartnerDepartemen.partner_id) \
                # .outerjoin(Jabatan,
                           # (PartnerDepartemen.jabatan_id == Jabatan.id)) \
                # .outerjoin(struktural,
                           # (PartnerDepartemen.jabatan_id == struktural.id)) \
                # .order_by(Partner.nama)
            # if self.req.user.company_id:
                # q = q.filter(Departemen.company_id == self.req.user.company_id)
            # row_table = DataTables(request.GET, q, columns)
            # return row_table.output_result()

        # elif url_dict['act'] == 'hon_departemen':
            # term = 'term' in params and params['term'] or ''
            # prefix = 'prefix' in params and params['prefix'] or ''
            # q = PartnerDBSession.query(Partner.id, Partner.nik, Partner.nama,
                                       # PartnerDepartemen.jabatan_id,
                                       # Jabatan.nama.label('jabatan_nm'),
                                       # ).join(PartnerDepartemen) \
                # .join(Jabatan, (PartnerDepartemen.jabatan_id == Jabatan.id)) \
                # .filter(Partner.nama.ilike('%%%s%%' % term)) \
                # .filter(PartnerDepartemen.departemen_id == ses['departemen_id']) \
                # .filter(or_(Jabatan.kode == '101', Jabatan.kode == '102')) \
                # .order_by(Partner.nama)
            # if self.req.user.company_id:
                # q = q.filter(Departemen.company_id == self.req.user.company_id)
            # rows = q.all()
            # r = []
            # keys = ('id', 'value', 'nik', 'nama', 'jabatan_id', 'jabatan_nm')
            # for k in rows:
                # values = (k[0], k[2] + (" - ") + k[4], k[1], k[2], k[3], k[4])
                # r.append(dict(zip(keys, values)))
            # return r

        # elif url_dict['act'] == 'hon_jabatan':
            # term = 'term' in params and params['term'] or ''
            # prefix = 'prefix' in params and params['prefix'] or ''
            # partner_id = 'partner_id' in params and params['partner_id'] or 0
            # q = PartnerDBSession.query(Partner.id, Partner.nik, Partner.nama,
                                       # PartnerDepartemen.jabatan_id,
                                       # Jabatan.nama.label('jabatan_nm'),
                                       # ).join(PartnerDepartemen) \
                # .join(Jabatan, (PartnerDepartemen.jabatan_id == Jabatan.id)) \
                # .filter(Partner.nama.ilike('%%%s%%' % term)) \
                # .filter(PartnerDepartemen.departemen_id == ses['departemen_id']) \
                # .order_by(Partner.nama)
            # if self.req.user.company_id:
                # q = q.filter(Departemen.company_id == self.req.user.company_id)
            # rows = q.all()
            # r = []
            # keys = ('id', 'value', 'nik', 'nama', 'jabatan_id', 'jabatan_nm')
            # for k in rows:
                # values = (k[0], k[2] + (" - ") + k[4], k[1], k[2], k[3], k[4])
                # r.append(dict(zip(keys, values)))
            # return r

        # elif url_dict['act'] == 'hon_skpkd':
            # term = 'term' in params and params['term'] or ''
            # prefix = 'prefix' in params and params['prefix'] or ''
            # q = PartnerDBSession.query(Partner.id, Partner.nik, Partner.nama,
                                       # PartnerDepartemen.jabatan_id,
                                       # Jabatan.nama.label('jabatan_nm'),
                                       # ).join(PartnerDepartemen) \
                # .join(Jabatan, (PartnerDepartemen.jabatan_id == Jabatan.id)) \
                # .filter(Partner.nama.ilike('%%%s%%' % term)) \
                # .filter(PartnerDepartemen.departemen_id == ses['departemen_id']) \
                # .filter(or_(Jabatan.kode == '101', Jabatan.kode == '102',
                           # Jabatan.kode == '103', Jabatan.kode == '104')) \
                # .order_by(Partner.nama)
            # if self.req.user.company_id:
                # q = q.filter(Departemen.company_id == self.req.user.company_id)
            # rows = q.all()
            # r = []
            # keys = ('id', 'value', 'nik', 'nama', 'jabatan_id', 'jabatan_nm')
            # for k in rows:
                # values = (k[0], k[2] + (" - ") + k[4], k[1], k[2], k[3], k[4])
                # r.append(dict(zip(keys, values)))
            # return r

        # elif url_dict['act'] == 'hon_jabatannm':
            # term = 'term' in params and params['term'] or ''
            # q = DBSession.query(Jabatan.id, Jabatan.kode, Jabatan.nama,
                                # Jabatan.jenis) \
                # .filter(Jabatan.nama.ilike('%%%s%%' % term)) \
                # .order_by(Jabatan.nama)
            # if self.req.user.company_id:
                # q = q.filter(Departemen.company_id == self.req.user.company_id)
            # rows = q.all()
            # r = []
            # for k in rows:
                # if k[3] == 1:
                    # nama_jenis = 'Struktural'
                # elif k[3] == 2:
                    # nama_jenis = 'Fungsional'
                # else:
                    # nama_jenis = 'Keuangan'
                # d = {}
                # d['id'] = k[0]
                # d['value'] = k[2] + ' (' + nama_jenis + ')'
                # d['kode'] = k[1]
                # d['nama'] = k[2]
                # r.append(d)
            # return r
        return super(ViewPartner, self).view_act()

    def get_values(self, row, istime=False):
        values = super().get_values(row, istime)
        values['nama'] = row.partner.nama if row.partner else ''
        values['jabatan_id'] = str(row.jabatan_id) if row.jabatan_id else ''
        # Ambil semua departemen yang terkait dengan partner_id dan jabatan_id ini
        departemen_ids = DBSession.query(PartnerDepartemen.departemen_id).filter_by(
            partner_id=row.partner_id,
            jabatan_id=row.jabatan_id
        ).all()
        values['departemen'] = [str(dept_id[0]) for dept_id in departemen_ids] if departemen_ids else []
        values['mulai'] = dmy(row.mulai) if row.mulai else ''
        values['selesai'] = dmy(row.selesai) if row.selesai else ''
        return values

    @view_config(route_name='partner-departemen-view',
                 renderer='templates/form.pt', permission='partner-departemen')
    def view_view(self):
        return super().view_view()

    def before_add(self):
        values = {'departemen_id': self.ses['departemen_id'],
                  'departemen_nm': self.ses['departemen_nm'],
                  'departemen_kd': self.ses['departemen_kd']}
        return values

    @view_config(route_name='partner-departemen-add',
                 renderer='templates/form.pt',
                 permission='partner-departemen')
    def view_add(self):
        print(f"POST data: {self.req.POST}")  # Debugging data yang dikirim dari form
        return super().view_add()

    @view_config(route_name='partner-departemen-edit',
                 renderer='templates/form.pt',
                 permission='partner-departemen')
    def view_edt(self):
        return super().view_edit()

    @view_config(route_name='partner-departemen-delete',
                 renderer='templates/form.pt',
                 permission='partner-departemen')
    def view_del(self):
        return super().view_delete()
        
    @view_config(route_name='partner-departemen-upload',
                 renderer='templates/form.pt', permission='partner-departemen')
    def view_upload(self):
        self.upload_keys = ["mulai"]
        return super().view_upload(exts=(".csv", ".tsv"))