user_ext.py 5.27 KB
import os
import colander
from datatables import (ColumnDT, DataTables, )
from deform import (widget, )
from opensipkd.tools.buttons import btn_close, btn_delete, btn_view
from opensipkd.tools.report import open_rml_row, csv_response, open_rml_pdf, pdf_response, file_response
from pyramid.i18n import TranslationStringFactory
from pyramid.view import view_config

from . import BaseView, button_import, get_params # Tombol import
from opensipkd.models import (DBSession, User, ExternalIdentity)

# lib buat mendapatkan full path asset
from pyramid.path import AssetResolver

_ = TranslationStringFactory('user')


class AddSchema(colander.Schema):
    external_user_name = colander.SchemaNode(
        colander.String(), title=_('User Name'))
    provider_name = (colander.SchemaNode(colander.String(), title=_('Provider')))
    local_user_id = (colander.SchemaNode(colander.String(), title=_('User ID')))


class EditSchema(AddSchema):
    external_id = colander.SchemaNode(colander.String(),
                                      widget=widget.TextInputWidget(readonly=True),
                                      missing=colander.drop)


class ListSchema(colander.Schema):
    id = colander.SchemaNode(colander.String(), field="external_id", title="Action")
    external_user_name = colander.SchemaNode(
        colander.String(), title=_('User Name'))
    provider_name = (colander.SchemaNode(colander.String(), title=_('Provider')))
    local_user_id = (colander.SchemaNode(colander.String(), title=_('User ID')))


class UserExt(BaseView):
    def __init__(self, request):
        super(UserExt, self).__init__(request)
        self.edit_schema = EditSchema
        self.list_schema = ListSchema
        self.list_route = "user-ext"
        self.list_buttons =  (btn_close,) + self.list_report + (button_import,)
        self.table = ExternalIdentity

        # Coba cetak pdf
        path = os.path.dirname(__file__)
        path = os.path.dirname(path)
        self.report_file = os.path.join(path, 'reports', 'users-external.jrxml')

    @view_config(
        route_name='user-ext', 
        renderer='templates/table.pt',
        permission='user-view')
    def view_list(self):
        form = super(UserExt, self).view_list(allow_edit=False, allow_delete=False)
        return form

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

    # Fungsi delete
    @view_config(
        route_name='user-ext-delete', 
        renderer='templates/form.pt',
        permission='user-edit')
    def view_delete(self):
        return super(UserExt, self).view_delete()

    # Mendapatkan path gambar
    def get_module_path(self, module_name):
        a = AssetResolver(module_name)
        resolver = a.resolve('')
        return resolver.abspath()

    def pdf_response(self, **kwargs):
        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])

    # Coba csv 
    def csv_response(self, **kwargs):
        query = self.table.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)
    def csv_response(request, value, filename=None):
        if not filename:
            filename = get_random_string(20)+".csv"
        response = request.response
        response.content_type = 'text/csv'
        # response.content_disposition = 'attachment;filename=' + filename
        response.content_disposition = 'filename=' + filename
        if sys.version_info < (3,):
            import StringIO
            fout = StringIO.StringIO()
        else:
            fout = io.StringIO()

        fcsv = csv.writer(fout, delimiter=',', quotechar='"',
                        quoting=csv.QUOTE_MINIMAL)
        fcsv.writerow(value.get('header', []))
        fcsv.writerows(value.get('rows', []))
        response.write(fout.getvalue())
        return response

    @view_config(
        route_name='user-ext-act', 
        renderer='json', 
        permission='user-view')
    def view_act(self):
        return super(UserExt, self).view_act()

    def delete_msg(self, row):
        return f'Data ID {row.external_id} sudah dihapus.'

    def query_id(self):
        return DBSession.query(ExternalIdentity).filter_by(external_id=self.req.matchdict["id"])

    # Fungsi upload
    @view_config(
        route_name='user-ext-upload',
        renderer='templates/form.pt',
        permission='user-upload'
    )
    def view_upload(self):
        self.upload_keys=["local_user_id"]
        return super().view_upload(exts=(".csv", ".tsv"))