Skip to content

aa.gusti / opensipkd-base

Go to a project
Commit 884eca57 by aa.gustiana@gmail.com
Options

add XLS export functionality and update CSV button logic in DeTable

1 parent b42be916
Inline Side-by-side
Showing 2 changed files with 76 additions and 6 deletions
opensipkd/base/views/base_views.py
...@@ -20,10 +20,10 @@ from opensipkd.tools import dmy, get_settings, get_ext, \ ...@@ -20,10 +20,10 @@ from opensipkd.tools import dmy, get_settings, get_ext, \
date_from_str, get_random_string, Upload, InvalidExtension, mem_tmp_store date_from_str, get_random_string, Upload, InvalidExtension, mem_tmp_store
from opensipkd.tools.buttons import ( from opensipkd.tools.buttons import (
btn_save, btn_cancel, btn_close, btn_delete, btn_add, btn_csv, btn_edit, btn_save, btn_cancel, btn_close, btn_delete, btn_add, btn_csv, btn_edit,
btn_pdf, btn_upload) btn_pdf, btn_upload, btn_xls)
# from opensipkd.tools.captcha import get_captcha # from opensipkd.tools.captcha import get_captcha
from opensipkd.tools.report import csv_response, file_response from opensipkd.tools.report import csv_response, file_response, xls_response
from opensipkd.base import BASE_CLASS from opensipkd.base import BASE_CLASS
from .common import DataTables from .common import DataTables
from ..models import DBSession, Partner, Base from ..models import DBSession, Partner, Base
...@@ -74,9 +74,11 @@ class CSRFSchema(colander.Schema): ...@@ -74,9 +74,11 @@ class CSRFSchema(colander.Schema):
colander.String(), colander.String(),
widget=widget_os.CSRFWidget(), widget=widget_os.CSRFWidget(),
) )
import re
from pyramid.interfaces import IRoutesMapper from pyramid.interfaces import IRoutesMapper
from pyramid.threadlocal import get_current_registry from pyramid.threadlocal import get_current_registry
import io
import xlsxwriter
class BaseView(object): class BaseView(object):
def __init__(self, request): def __init__(self, request):
...@@ -558,6 +560,9 @@ class BaseView(object): ...@@ -558,6 +560,9 @@ class BaseView(object):
elif url_dict['act'] == 'pdf': elif url_dict['act'] == 'pdf':
return self.pdf_response(**kwargs) return self.pdf_response(**kwargs)
elif url_dict['act'] == 'xls':
return self.xls_response(**kwargs)
else: else:
return self.next_act(**kwargs) return self.next_act(**kwargs)
...@@ -678,6 +683,59 @@ class BaseView(object): ...@@ -678,6 +683,59 @@ class BaseView(object):
} }
return csv_response(self.req, value, filename) return csv_response(self.req, value, filename)
def remove_tags(self, text):
# Regular expression pattern to match HTML tags
clean_text = re.sub(r'<[^>]+>', '', text)
return clean_text
# def render_xls(self, header, rows):
# output = io.BytesIO()
# workbook = xlsxwriter.Workbook(output)
# worksheet = workbook.add_worksheet("Sheet 1")
# for col_num, header in enumerate(header):
# worksheet.write_row(0, col_num, header)
# for row_num, row in enumerate(rows, start=1):
# for col_num, cell_value in enumerate(row):
# worksheet.write_row(row_num, col_num, cell_value)
# workbook.close()
# output.seek(0)
# filename = f"{get_random_string(16)}.xlsx"
# response = self.req.response
# response.content_type = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
# response.content_disposition = 'filename=' + filename
# return response.write(output.read())
def xls_data(self, **kwargs):
resp = self.get_list(**kwargs)
data = resp.get("data", [])
if not data:
raise HTTPNotFound("No data to export")
header = list(data[0].keys())
list_schema = self.list_schema()
for i, h in enumerate(header):
for d in list_schema:
if d.name == h and hasattr(d, "title"):
header[i] = d.title
rows = [list(item.values()) for item in data]
for row in rows:
for i, value in enumerate(row):
if isinstance(value, str):
row[i] = self.remove_tags(value)
value = {
'header': header,
'rows': rows,
}
return value
def xls_response(self, **kwargs):
value = self.xls_data()
return xls_response(self.req, value)
def get_bindings(self, row=None): def get_bindings(self, row=None):
return {"row": row} return {"row": row}
......
opensipkd/detable/detable.py
...@@ -160,7 +160,7 @@ class DeTable(field.Field): ...@@ -160,7 +160,7 @@ class DeTable(field.Field):
close_url = self.action.split("/") close_url = self.action.split("/")
close_url = "/".join(close_url[:-1]) close_url = "/".join(close_url[:-1])
close_url.replace(":/", "://") close_url.replace(":/", "://")
params = params and f"?{params}" or "" params = params and f"?{params}" or "?"
dict_buttons = { dict_buttons = {
"close": "{window.location = '" + close_url + "'; return false;}", "close": "{window.location = '" + close_url + "'; return false;}",
"add": "{window.location = o%sUri+'/add%s';}" % (tableid, params), "add": "{window.location = o%sUri+'/add%s';}" % (tableid, params),
...@@ -183,8 +183,20 @@ class DeTable(field.Field): ...@@ -183,8 +183,20 @@ class DeTable(field.Field):
event.stopPropagation(); event.stopPropagation();
}""" % (tableid, tableid, tableid), }""" % (tableid, tableid, tableid),
"csv": "{window.location = o%sUri+'/csv/act%s';}" % ( "csv": """{
tableid, params), params = o%s.ajax.params();
paramString = $.param(params);
console.log(params);
window.location = o%sUri+'/csv/act%s&'+paramString;}
""" % (
tableid, tableid, params),
"xls": """{
params = o%s.ajax.params();
paramString = $.param(params);
console.log(params);
window.location = o%sUri+'/xls/act%s&'+paramString;}
""" % (
tableid, tableid, params),
"pdf": "{window.open(o%sUri+'/pdf/act%s');}" % (tableid, params), "pdf": "{window.open(o%sUri+'/pdf/act%s');}" % (tableid, params),
"upload": "{window.location = o%sUri+'/upload%s';}" % ( "upload": "{window.location = o%sUri+'/upload%s';}" % (
tableid, params), tableid, params),
......
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!