base_views.py

import logging
import os
import re
from datetime import datetime
from email.utils import parseaddr

import colander
from datatables import ColumnDT
from dateutil.relativedelta import relativedelta
from deform import (widget, Form, ValidationFailure, FileData, )
from pyramid.httpexceptions import HTTPFound, HTTPNotFound

from opensipkd.base.views.upload import tmpstore
from opensipkd.tools import dmy, get_settings, get_ext, \
    date_from_str, get_random_string
from opensipkd.tools.buttons import btn_save, btn_cancel, btn_close, btn_delete, \
    btn_add, btn_csv, \
    btn_pdf, btn_unpost, btn_post
from opensipkd.tools.captcha import get_captcha
from opensipkd.tools.report import csv_response, file_response
from .common import DataTables
from .. import DBSession, get_params, get_urls
from ..scripts.initializedb import append_csv
from ...detable import DeTable

log = logging.getLogger(__name__)


class UploadSchema(colander.Schema):
    upload = colander.SchemaNode(
        FileData(),
        widget=widget.FileUploadWidget(tmpstore),
        title='Unggah')


class BaseView(object):
    def __init__(self, request):
        self.req = request
        self.ses = self.req.session
        self.db_session = DBSession
        self.params = self.req.params
        self.settings = get_settings()
        # if not request.user:
        if "g_state" in request.cookies:
            request.response.delete_cookie("g_state", '/')

        now = datetime.now()
        # self.dt_awal = self.ses["dt_awal"] if "dt_awal" in self.ses else now
        # self.awal = dmy(self.dt_awal)
        # self.dt_akhir = self.ses["dt_akhir"] if "dt_akhir" in self.ses else now
        # self.akhir = dmy(self.dt_akhir)
        # self.ses["dt_awal"] = self.dt_awal
        # self.ses["dt_akhir"] = self.dt_akhir
        self.tahun = 'tahun' in self.ses and self.ses['tahun'] or now.strftime(
            '%Y')
        self.tahun = 'tahun' in self.params and self.params[
            'tahun'] or self.tahun
        self.ses['tahun'] = self.tahun

        self.bulan = 'bulan' in self.ses and self.ses['bulan'] or now.strftime(
            '%m')

        if 'bulan' in self.params and self.params['bulan']:
            self.bulan = self.params['bulan'].strip().zfill(2)
            dt_awal = date_from_str(
                '{d}-{m}-{y}'.format(y=self.tahun, m=self.bulan, d='01'))
            dt_akhir = dt_awal + \
                       relativedelta(months=1) - relativedelta(days=1)

            self.ses['awal'] = dmy(dt_awal)
            self.ses['akhir'] = dmy(dt_akhir)

        self.ses['bulan'] = int(self.bulan)

        self.posted = 'posted' in self.ses and self.ses['posted'] or 0
        if 'posted' in self.params and self.params['posted']:
            posted = self.params['posted']
            self.posted = ((posted == 'true' or posted == '1') and 1) or (
                    (posted == 'false' or posted == '0') and 0) or 0
        self.ses['posted'] = self.posted

        self.awal = 'awal' in self.ses and self.ses['awal'] or dmy(now)
        awal = 'awal' in self.params and self.params['awal'] or self.awal
        try:
            self.dt_awal = date_from_str(awal)
            self.awal = awal
        except:
            self.dt_awal = date_from_str(self.awal)

        self.ses['awal'] = self.awal
        self.ses['dt_awal'] = self.dt_awal

        self.akhir = 'akhir' in self.ses and self.ses['akhir'] or dmy(now)
        akhir = 'akhir' in self.params and self.params['akhir'] or self.akhir

        try:
            self.dt_akhir = date_from_str(akhir)
            self.akhir = akhir
        except:
            self.dt_akhir = date_from_str(self.akhir)

        self.tahun_awal = 'tahun_awal' in self.ses and self.ses[
            'tahun_awal'] or self.tahun
        self.tahun_awal = 'tahun_awal' in self.params and self.params[
            'tahun_awal'] or self.tahun_awal
        self.ses['tahun_awal'] = self.tahun_awal

        self.tahun_akhir = 'tahun_akhir' in self.ses and self.ses[
            'tahun_akhir'] or self.tahun_awal
        self.tahun_akhir = 'tahun_akhir' in self.params and self.params[
            'tahun_akhir'] or self.tahun_akhir
        self.ses['tahun_akhir'] = self.tahun_akhir

        self.departemen_kd = 'departemen_kd' in self.ses and self.ses[
            'departemen_kd'] or '0.0.00'
        self.departemen_nm = 'departemen_nm' in self.ses and self.ses[
            'departemen_nm'] or 'PILIH UNIT'
        self.departemen_id = 'departemen_id' in self.ses and self.ses[
            'departemen_id'] or 0
        self.ses['departemen_kd'] = self.departemen_kd
        self.ses['departemen_nm'] = self.departemen_nm
        self.ses['departemen_id'] = self.departemen_id

        self.jenis = 'jenis' in self.ses and self.ses['jenis'] or 0
        self.jenis = 'jenis' in self.params and self.params[
            'jenis'] or self.jenis
        self.ses['jenis'] = self.jenis
        self.list_route = 'home'
        self.list_col_defs = ""
        self.list_cols = ""
        # self.list_buttons = 'btn_view, btn_add, btn_edit, btn_delete, ' \
        #                     'btn_close'
        self.list_report = (btn_csv, btn_pdf)
        # self.list_buttons = (btn_view, btn_add, btn_edit, btn_delete, btn_close)
        self.list_buttons = (btn_add, btn_close)
        self.columns = None
        self.form_params = dict(scripts="")
        self.list_url = ''
        self.list_route = ''
        self.list_schema = ""
        self.form_scripts = """
         $('#parent_nm').bind('typeahead:selected', function(obj, datum) {
              $('#parent_id').val(datum.id);
              $('#parent_kd').val(datum.kode);

        });"""
        self.edit_schema = ""
        self.add_schema = ""
        self.upload_schema = UploadSchema
        self.table = ""
        self.home = self.req._host
        self.buttons = None
        self.headers = None
        self.bindings = {}
        self.autocomplete = 'on'
        self.action_suffix = "/grid/act"
        self.upload_keys = ["kode"]
        self.report_file = ""
        self.new_buttons = {}

    def query_register(self, **kwargs):
        pass

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

    def route_list(self, **kwargs):
        msg = kwargs.get("msg")
        error = kwargs.get("error", "")
        list_url = kwargs.get("list_url", None)
        if msg:
            self.ses.flash(msg, error)
        if not list_url:
            list_url = self.req.route_url(self.list_route)
        log.error(list_url)
        if self.headers:
            return HTTPFound(location=get_urls(list_url),
                             headers=self.headers)
        else:
            return HTTPFound(location=get_urls(list_url))

    def form_validator(self, form, value):
        pass

    def get_params(self, params, default=None):
        return get_params(params, default)

    def get_form(self, class_form, row=None, buttons=(btn_save, btn_cancel),
                 **kwargs):
        buttons = self.buttons and self.buttons or buttons
        if "bindings" in kwargs and kwargs["bindings"]:
            bindings = kwargs["bindings"]
        elif self.bindings:
            bindings = self.bindings
        else:
            bindings = self.get_bindings(row)
        form_params = {}
        # form_params["after_bind"] = after_bind
        if "validator" in kwargs and kwargs["validator"]:
            form_params["validator"] = kwargs["validator"]
            # schema = class_form(validator=kwargs["validator"])
        else:
            form_params["validator"] = self.form_validator
            # schema = class_form(validator=self.form_validator)
        if "after_bind" in kwargs and kwargs["after_bind"]:
            form_params["after_bind"] = kwargs["after_bind"]
            # schema = class_form(validator=kwargs["validator"])

        schema = class_form(**form_params)

        schema = schema.bind(request=self.req, **bindings)
        schema.request = self.req
        if row:
            schema.deserialize(row)

        return Form(schema, buttons=buttons, autocomplete=self.autocomplete)

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

    def view_list(self, **kwargs):
        if self.list_schema:
            allow_edit = kwargs.get("allow_edit", True)
            allow_delete = kwargs.get("allow_delete", True)
            allow_post = kwargs.get("allow_post", False)
            allow_unpost = kwargs.get("allow_unpost", False)
            state_save = kwargs.get("state_save", False)
            schema = self.list_schema()
            schema = schema.bind(request=self.req)
            list_url = kwargs.get("list_url", None)
            new_buttons = kwargs.get("new_buttons")
            if not new_buttons:
                new_buttons = self.new_buttons

            if not list_url:
                list_url = self.req.route_url(self.list_route)
            table = DeTable(schema,
                            action=list_url,
                            action_suffix="/grid/act",
                            buttons=self.list_buttons,
                            request=self.req,
                            allow_edit=allow_edit,
                            allow_delete=allow_delete,
                            allow_post=allow_post,
                            allow_unpost=allow_unpost,
                            state_save=state_save,
                            new_buttons=new_buttons
                            )
            resources = table.get_widget_resources()
            # resources=dict(css="", js="")
            return dict(form=table.render(), scripts="", css=resources["css"],
                        js=resources["js"])

        arg = kwargs and kwargs or {}
        arg.update(url=self.list_url, col_defs=self.list_col_defs,
                   cols=self.list_cols, buttons=self.list_buttons)
        return arg

    def get_bindings(self, row=None):
        return {}

    def next_view(self, form, **kwargs):
        return self.route_list()

    def next_edit(self, form, **kwargs):
        return self.route_list(**kwargs)

    def view_view(self, **kwargs):  # row = query_id(request).first()
        request = self.req
        row = self.query_id().first()
        if not row:
            return self.id_not_found()
        bindings = self.get_bindings(row)
        buttons = kwargs.get("buttons", (btn_close,))
        form = self.get_form(self.edit_schema, buttons=buttons,
                             bindings=bindings)
        if request.POST:
            result = self.next_view(form, row=row)
            if result:
                return result
            return self.after_view(row=row)

        values = self.get_values(row)
        if not values:
            return self.route_list("Nilai Data tidak ditemukan", "error")
        form.set_appstruct(values)
        table = self.get_item_table(row)
        resources = form.get_widget_resources()
        is_object = kwargs.get("is_object", False)
        if is_object:
            return dict(form=form,
                        readonly=True,
                        table=table and table.render() or None,
                        scripts=self.form_scripts,
                        css=resources["css"],
                        js=resources["js"],
                        **kwargs
                        )
        return dict(form=form.render(readonly=True),
                    table=table and table.render() or None,
                    scripts=self.form_scripts,
                    css=resources["css"],
                    js=resources["js"],
                    **kwargs
                    )

    def set_post(self, **kwargs):
        pass

    def set_unpost(self, **kwargs):
        pass

    def view_post(self, post_field="status", **kwargs):
        request = self.req
        row = self.query_id().first()
        if not row:
            return self.id_not_found()
        if getattr(row, post_field):
            buttons = (btn_unpost, btn_close)
        else:
            buttons = (btn_post, btn_close)
        return self.view_view(buttons=buttons)

    def view_upload(self, exts=('.png', '.ico')):
        bindings = self.get_bindings()
        form = self.get_form(self.upload_schema, bindings=bindings)
        resources = form.get_widget_resources()
        if self.req.POST:
            if 'save' in self.req.POST:
                input_file = self.req.POST['upload'].file
                filename = self.req.POST['upload'].filename.lower()
                ext = get_ext(filename).lower()
                if ext.lower() not in exts:
                    ext = ", ".join([ex for ex in exts])
                    self.req.session.flash(f'File harus format {ext}', 'error')
                    return dict(form=form.render(),
                                scripts=self.form_scripts, css=resources["css"],
                                js=resources["js"])

                _here = get_params('tmp', '/tmp')
                folder = os.path.join(_here, 'upload')
                if not os.path.exists(folder):
                    os.makedirs(folder)

                fullpath = os.path.join(folder, filename)
                output_file = open(fullpath, 'wb')
                input_file.seek(0)
                while True:
                    data = input_file.read(2 << 16)
                    if not data:
                        break
                    output_file.write(data)
                output_file.close()
                self.save_upload(fullpath)

            elif "cancel" in self.req.POST or 'batal' in self.req.POST or "close" in self.req.POST:
                self.cancel_act()
            else:
                return self.next_add(form, resources=resources)

            return self.route_list()
        return dict(form=form.render(),
                    scripts=self.form_scripts, css=resources["css"],
                    js=resources["js"])

    def get_file(self, filename):
        return open(filename)

    def save_upload(self, file_name):
        return append_csv(self.table, file_name, self.upload_keys,
                          get_file_func=self.get_file, update_exist=True)

    def before_add(self):
        return {}

    def validation_failure(self, value):
        return value

    def cancel_act(self, **kwargs):
        return self.route_list(**kwargs)

    def after_add(self, row=None, **kwargs):
        return self.route_list(**kwargs)

    def after_edit(self, row=None, **kwargs):
        return self.route_list(**kwargs)

    def after_view(self, **kwargs):
        return self.route_list(**kwargs)

    def next_act(self):
        url_dict = self.req.matchdict
        raise HTTPNotFound

    def pdf_response(self, **kwargs):
        from opensipkd.base.tools.report import jasper_export
        filename = jasper_export(self.report_file)
        return file_response(self.req, filename=filename[0])

    def csv_response(self):
        query = self.table.query_register()
        row = query.first()
        header = row.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 list_join(self, query):
        return query

    def list_filter(self, query):
        return query

    def get_list(self):
        url = []
        if not self.columns:
            columns = []
            for d in self.list_schema():
                global_search = True
                if hasattr(d, "searchable"):
                    if d.searchable == False:
                        global_search = False

                if hasattr(d, "field"):
                    if type(d.field) == str:
                        columns.append(
                            ColumnDT(getattr(self.table, d.field),
                                     mData=d.name,
                                     global_search=global_search))
                    else:
                        columns.append(
                            ColumnDT(d.field, mData=d.name,
                                     global_search=global_search
                                     ))
                else:
                    columns.append(
                        ColumnDT(getattr(self.table, d.name), mData=d.name,
                                 global_search=global_search))
                if hasattr(d, "url"):
                    url.append(d.name)
        else:
            columns = self.columns

        query = self.db_session.query().select_from(self.table)
        query = self.list_join(query)
        if self.req.user and self.req.user.company_id and hasattr(
                self.table, "company_id"):
            query = query.filter(
                self.table.company_id == self.req.user.company_id)
        query = self.list_filter(query)
        row_table = DataTables(self.req.GET, query, columns)
        result = row_table.output_result()
        #     for k, v in d.items():
        #         if k in url and v:
        #             link = "/".join([self.home, nik_url, v])
        #             d[k] =f'<a href="{link}" target="_blank">View</a>'
        return result

    def view_act(self, **kwargs):
        url_dict = self.req.matchdict
        if url_dict['act'] == 'grid':
            return self.get_list()

        elif url_dict['act'] == 'csv':
            return self.csv_response()

        elif url_dict['act'] == 'pdf':
            return self.pdf_response()

        else:
            return self.next_act()

    def view_add(self, **kwargs):
        # bindings = self.get_bindings()
        form = self.get_form(self.add_schema, **kwargs)
        table = self.get_item_table(**kwargs)
        resources = form.get_widget_resources()
        is_object = kwargs.get("is_object", False)
        if self.req.POST:
            if 'save' in self.req.POST:
                controls = self.req.POST.items()
                try:
                    c = form.validate(controls)
                except ValidationFailure as e:
                    # value = self.validation_failure(e.cstruct)
                    # value.update(self.before_add())
                    # form.render(appstruct=value)
                    # log.debug(e.cstruct)
                    # log.debug(e.field)
                    # efield = e.field
                    for f in e.field.children:
                        if isinstance(f.typ, colander.Date):
                            e.cstruct[f.name] = date_from_str(
                                e.cstruct[f.name])

                    # for k, v in e.cstruct.items():
                    #     log.debug(hasattr(e.field, k))
                    # if isinstance(f, colander.Date):
                    #     e.cstruct[f] = date_from_str(e.cstruct[f])
                    if is_object:
                        return dict(form=form,
                                    table=table and table.render() or None,
                                    scripts=self.form_scripts,
                                    css=resources["css"],
                                    js=resources["js"],
                                    **kwargs
                                    )

                    return dict(form=form.render(e.cstruct),
                                table=table and table.render() or None,
                                scripts=self.form_scripts, css=resources["css"],
                                js=resources["js"],
                                **kwargs
                                )
                values = dict(c)
                row = self.save_request(values)
                return self.after_add(row=row, **kwargs)
            elif "cancel" in self.req.POST or 'batal' in self.req.POST or "close" in self.req.POST:
                self.cancel_act()
            else:
                return self.next_add(form, table=table, resources=resources)

            return self.route_list(**kwargs)
        values = self.before_add()
        form.set_appstruct(values)
        if is_object:
            return dict(form=form,
                        table=table and table.render() or None,
                        scripts=self.form_scripts,
                        css=resources["css"],
                        js=resources["js"],
                        **kwargs
                        )

        return dict(form=form.render(), table=table and table.render() or None,
                    scripts=self.form_scripts, css=resources["css"],
                    js=resources["js"])

    def save(self, values, user, row=None):
        log.info("Save")
        log.debug(values)
        values.pop("id", None)
        self.ses["old_email"] = user and user.email or None
        if not row:
            row = self.table()
            row.created = datetime.now()
            row.create_uid = user and user.id or None
        else:
            row.updated = datetime.now()
            row.update_uid = user and user.id or None

        row.from_dict(values)
        if hasattr(row, "status"):
            status = "status" in values and values["status"] or 0
            try:
                status = int(status)
            except:
                status = status and 1 or 0
            row.status = status
        self.db_session.add(row)
        self.db_session.flush()
        return row

    def save_request(self, values, row=None):
        for k, v in self.req.GET.items():
            if k not in values:
                if v:
                    values[k] = v
        return self.save(values, self.req.user, row)

    def id_not_found(self, **kwargs):
        msg = f"Data yang dicari Tidak Ditemukan ID:" \
              f" {self.req.matchdict['id']}."
        self.req.session.flash(msg, 'error')
        return self.route_list(**kwargs)

    def get_values(self, row, istime=False):
        d = row.to_dict()
        # if 'tanggal' in d and d['tanggal']:
        #     d["tanggal"] = dmy(row.tanggal)
        for f in d:
            if type(d[f]) is str:
                d[f] = d[f].strip()
        return d

    def get_item_table(self, row=None, **kwargs):
        return

    def before_edit(self, form):
        return form

    def view_edit(self, **kwargs):
        request = self.req
        row = self.query_id().first()
        is_object = kwargs.get("is_object", False)

        if not row:
            return self.id_not_found(**kwargs)
        if not self.bindings:
            self.bindings = self.get_bindings(row)
        form = self.get_form(self.edit_schema, **kwargs)
        table = self.get_item_table(row)
        resources = form.get_widget_resources()
        if request.POST:
            if 'save' in request.POST:
                # log.debug("Save Edit")
                # log.debug(dict(request.POST.items()))
                # log.debug(request.POST)
                controls = request.POST.items()
                log.debug(controls)
                # log.debug(dict(controls))
                # log.debug(list(controls))
                try:
                    controls = form.validate(controls)
                except ValidationFailure as e:
                    log.error(f"Edit Error: {str(e.error)}")
                    # log.debug(f"Edit Data: {e.cstruct}")
                    form.set_appstruct(e.cstruct)
                    if is_object:
                        return dict(form=form,
                                    table=table and table.render() or None,
                                    scripts=self.form_scripts,
                                    css=resources["css"],
                                    js=resources["js"],
                                    **kwargs
                                    )
                    return dict(form=form.render(),
                                table=table and table.render() or None,
                                scripts=self.form_scripts, css=resources["css"],
                                js=resources["js"])
                c = dict(controls)
                self.save_request(c, row)
                return self.after_edit(row=row, **kwargs)

            return self.next_edit(form, row=row)

        values = self.get_values(row)
        form.set_appstruct(values)
        form = self.before_edit(form)
        if is_object:
            return dict(form=form,
                        table=table and table.render() or None,
                        scripts=self.form_scripts,
                        css=resources["css"],
                        js=resources["js"],
                        **kwargs
                        )
        return dict(form=form.render(), table=table and table.render() or None,
                    scripts=self.form_scripts, css=resources["css"],
                    js=resources["js"])

    def before_delete(self, row):
        pass

    def view_delete(self, **kwargs):
        request = self.req
        q = self.query_id()
        row = q.first()
        is_object = kwargs.get("is_object", False)

        if not row:
            return self.id_not_found()
        if not self.bindings:
            self.bindings = self.get_bindings(row)
        if request.POST:
            if 'delete' in request.POST:
                msg = self.delete_msg(row)
                try:
                    self.before_delete(row)
                except Exception as e:
                    self.ses.flash(e, "error")
                    return self.route_list()

                q.delete()
                self.db_session.flush()
                request.session.flash(msg)
            return self.route_list()
        form = self.get_form(
            self.edit_schema, buttons=(btn_delete, btn_cancel))
        table = self.get_item_table(row)
        resources = form.get_widget_resources()
        form.set_appstruct(self.get_values(row))
        if is_object:
            return dict(form=form,
                        readonly=True,
                        table=table and table.render() or None,
                        scripts=self.form_scripts,
                        css=resources["css"],
                        js=resources["js"],
                        **kwargs
                        )
        return dict(form=form.render(readonly=True),
                    table=table and table.render() or None,
                    scripts=self.form_scripts, css=resources["css"],
                    js=resources["js"])

    def query_id(self):
        q = self.db_session.query(self.table).filter_by(
            id=self.req.matchdict['id'])
        if hasattr(self.table, 'compnay_id') and self.req.user.company_id:
            q = q.filter_by(company_id=self.req.user.company_id)
        return q

    def filter_company(self, query):
        if self.req.user.company_id:
            return query.filter(
                self.table.company_id == self.req.user.company_id)
        return query

    def next_add(self, form, **kwargs):
        """
        Digunakan untuk memverifikasi button yang lainnya
        :param form:  Object Form
        :return:
        """
        return self.route_list()


@colander.deferred
def deferred_status(node, kw):
    values = kw.get('daftar_status', [])
    return widget.SelectWidget(values=values)


def email_validator(node, value):
    name, email = parseaddr(value)
    if not email or email.find('@') < 0:
        raise colander.Invalid(node, 'Invalid email format')


class Store(dict):
    def preview_url(self, name):
        return ""


username_re = re.compile('^[a-z0-9_]{6,16}$', re.IGNORECASE)


def user_name_validator(node, value):
    if not username_re.match(value):
        raise colander.Invalid(
            node,
            'Value must be between 6 and 16 characters and can only contain ' +
            'uppercase and lowercase alphanumeric characters or an underscore')


def need_captcha():
    is_captcha = get_params("reg_captcha")
    return is_captcha == '1' or is_captcha == "True" or is_captcha == "true" \
        or is_captcha == True


def need_verify():
    result = get_params("reg_verify")
    return result == '1' or result == "True" or result == "true" or result == True


def get_url_captcha(request):
    captcha = get_captcha(request)
    return os.path.join(get_urls(request.route_url('home')), 'captcha', captcha)