492 lines
14 KiB
Python
492 lines
14 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import frappe
|
|
import os, json, datetime
|
|
|
|
from frappe import _
|
|
from frappe.modules import scrub, get_module_path
|
|
from frappe.utils import flt, cint, get_html_format, cstr, get_url_to_form
|
|
from frappe.model.utils import render_include
|
|
from frappe.translate import send_translations
|
|
import frappe.desk.reportview
|
|
from frappe.utils.csvutils import read_csv_content_from_attached_file
|
|
from frappe.permissions import get_role_permissions
|
|
from six import string_types, iteritems
|
|
from datetime import timedelta
|
|
|
|
def get_report_doc(report_name):
|
|
doc = frappe.get_doc("Report", report_name)
|
|
if not doc.is_permitted():
|
|
frappe.throw(_("You don't have access to Report: {0}").format(report_name), frappe.PermissionError)
|
|
|
|
if not frappe.has_permission(doc.ref_doctype, "report"):
|
|
frappe.throw(_("You don't have permission to get a report on: {0}").format(doc.ref_doctype),
|
|
frappe.PermissionError)
|
|
|
|
if doc.disabled:
|
|
frappe.throw(_("Report {0} is disabled").format(report_name))
|
|
|
|
return doc
|
|
|
|
|
|
def generate_report_result(report, filters=None, user=None):
|
|
status = None
|
|
if not user:
|
|
user = frappe.session.user
|
|
if not filters:
|
|
filters = []
|
|
|
|
if filters and isinstance(filters, string_types):
|
|
filters = json.loads(filters)
|
|
columns, result, message, chart, data_to_be_printed = [], [], None, None, None
|
|
if report.report_type == "Query Report":
|
|
if not report.query:
|
|
status = "error"
|
|
frappe.msgprint(_("Must specify a Query to run"), raise_exception=True)
|
|
|
|
if not report.query.lower().startswith("select"):
|
|
status = "error"
|
|
frappe.msgprint(_("Query must be a SELECT"), raise_exception=True)
|
|
|
|
result = [list(t) for t in frappe.db.sql(report.query, filters)]
|
|
columns = [cstr(c[0]) for c in frappe.db.get_description()]
|
|
else:
|
|
module = report.module or frappe.db.get_value("DocType", report.ref_doctype, "module")
|
|
if report.is_standard == "Yes":
|
|
method_name = get_report_module_dotted_path(module, report.name) + ".execute"
|
|
threshold = 10
|
|
res = []
|
|
|
|
start_time = datetime.datetime.now()
|
|
# The JOB
|
|
res = frappe.get_attr(method_name)(frappe._dict(filters))
|
|
|
|
end_time = datetime.datetime.now()
|
|
|
|
if (end_time - start_time).seconds > threshold and not report.prepared_report:
|
|
report.db_set('prepared_report', 1)
|
|
|
|
columns, result = res[0], res[1]
|
|
if len(res) > 2:
|
|
message = res[2]
|
|
if len(res) > 3:
|
|
chart = res[3]
|
|
if len(res) > 4:
|
|
data_to_be_printed = res[4]
|
|
|
|
if result:
|
|
result = get_filtered_data(report.ref_doctype, columns, result, user)
|
|
|
|
if cint(report.add_total_row) and result:
|
|
result = add_total_row(result, columns)
|
|
|
|
return {
|
|
"result": result,
|
|
"columns": columns,
|
|
"message": message,
|
|
"chart": chart,
|
|
"data_to_be_printed": data_to_be_printed,
|
|
"status": status
|
|
}
|
|
|
|
@frappe.whitelist()
|
|
def background_enqueue_run(report_name, filters=None, user=None):
|
|
"""run reports in background"""
|
|
if not user:
|
|
user = frappe.session.user
|
|
report = get_report_doc(report_name)
|
|
track_instance = \
|
|
frappe.get_doc({
|
|
"doctype": "Prepared Report",
|
|
"report_name": report_name,
|
|
"filters": json.dumps(filters),
|
|
"ref_report_doctype": report_name,
|
|
"report_type": report.report_type,
|
|
"query": report.query,
|
|
"module": report.module,
|
|
})
|
|
track_instance.insert(ignore_permissions=True)
|
|
frappe.db.commit()
|
|
return {
|
|
"redirect_url": get_url_to_form("Prepared Report", track_instance.name)
|
|
}
|
|
|
|
|
|
@frappe.whitelist()
|
|
def get_script(report_name):
|
|
report = get_report_doc(report_name)
|
|
|
|
module = report.module or frappe.db.get_value("DocType", report.ref_doctype, "module")
|
|
module_path = get_module_path(module)
|
|
report_folder = os.path.join(module_path, "report", scrub(report.name))
|
|
script_path = os.path.join(report_folder, scrub(report.name) + ".js")
|
|
print_path = os.path.join(report_folder, scrub(report.name) + ".html")
|
|
|
|
script = None
|
|
if os.path.exists(script_path):
|
|
with open(script_path, "r") as f:
|
|
script = f.read()
|
|
|
|
html_format = get_html_format(print_path)
|
|
|
|
if not script and report.javascript:
|
|
script = report.javascript
|
|
|
|
if not script:
|
|
script = "frappe.query_reports['%s']={}" % report_name
|
|
|
|
# load translations
|
|
if frappe.lang != "en":
|
|
send_translations(frappe.get_lang_dict("report", report_name))
|
|
|
|
return {
|
|
"script": render_include(script),
|
|
"html_format": html_format
|
|
}
|
|
|
|
|
|
@frappe.whitelist()
|
|
@frappe.read_only()
|
|
def run(report_name, filters=None, user=None):
|
|
|
|
report = get_report_doc(report_name)
|
|
if not user:
|
|
user = frappe.session.user
|
|
if not frappe.has_permission(report.ref_doctype, "report"):
|
|
frappe.msgprint(_("Must have report permission to access this report."),
|
|
raise_exception=True)
|
|
|
|
result = None
|
|
|
|
if report.prepared_report:
|
|
if filters:
|
|
if isinstance(filters, string_types):
|
|
filters = json.loads(filters)
|
|
|
|
dn = filters.get("prepared_report_name")
|
|
else:
|
|
dn = ""
|
|
result = get_prepared_report_result(report, filters, dn)
|
|
else:
|
|
result = generate_report_result(report, filters, user)
|
|
|
|
result["add_total_row"] = report.add_total_row
|
|
|
|
return result
|
|
|
|
|
|
def get_prepared_report_result(report, filters, dn=""):
|
|
latest_report_data = {}
|
|
doc_list = frappe.get_all("Prepared Report", filters={"status": "Completed", "report_name": report.name})
|
|
doc = None
|
|
if len(doc_list):
|
|
if dn:
|
|
# Get specified dn
|
|
doc = frappe.get_doc("Prepared Report", dn)
|
|
else:
|
|
# Get latest
|
|
doc = frappe.get_doc("Prepared Report", doc_list[0])
|
|
|
|
data = read_csv_content_from_attached_file(doc)
|
|
if data:
|
|
latest_report_data = {
|
|
"columns": data[0],
|
|
"result": data[1:]
|
|
}
|
|
|
|
latest_report_data.update({
|
|
"prepared_report": True,
|
|
"doc": doc
|
|
})
|
|
|
|
return latest_report_data
|
|
|
|
@frappe.whitelist()
|
|
def export_query():
|
|
"""export from query reports"""
|
|
|
|
data = frappe._dict(frappe.local.form_dict)
|
|
|
|
del data["cmd"]
|
|
if "csrf_token" in data:
|
|
del data["csrf_token"]
|
|
|
|
if isinstance(data.get("filters"), string_types):
|
|
filters = json.loads(data["filters"])
|
|
if isinstance(data.get("report_name"), string_types):
|
|
report_name = data["report_name"]
|
|
if isinstance(data.get("file_format_type"), string_types):
|
|
file_format_type = data["file_format_type"]
|
|
if isinstance(data.get("visible_idx"), string_types):
|
|
visible_idx = json.loads(data.get("visible_idx"))
|
|
else:
|
|
visible_idx = None
|
|
|
|
if file_format_type == "Excel":
|
|
|
|
data = run(report_name, filters)
|
|
data = frappe._dict(data)
|
|
columns = get_columns_dict(data.columns)
|
|
|
|
result = [[]]
|
|
|
|
# add column headings
|
|
for idx in range(len(data.columns)):
|
|
result[0].append(columns[idx]["label"])
|
|
|
|
# build table from dict
|
|
if isinstance(data.result[0], dict):
|
|
for i,row in enumerate(data.result):
|
|
# only rows which are visible in the report
|
|
if row and (i in visible_idx):
|
|
row_list = []
|
|
for idx in range(len(data.columns)):
|
|
row_list.append(row.get(columns[idx]["fieldname"],""))
|
|
result.append(row_list)
|
|
elif not row:
|
|
result.append([])
|
|
else:
|
|
result = result + [d for i,d in enumerate(data.result) if (i in visible_idx)]
|
|
|
|
from frappe.utils.xlsxutils import make_xlsx
|
|
xlsx_file = make_xlsx(result, "Query Report")
|
|
|
|
frappe.response['filename'] = report_name + '.xlsx'
|
|
frappe.response['filecontent'] = xlsx_file.getvalue()
|
|
frappe.response['type'] = 'binary'
|
|
|
|
|
|
def get_report_module_dotted_path(module, report_name):
|
|
return frappe.local.module_app[scrub(module)] + "." + scrub(module) \
|
|
+ ".report." + scrub(report_name) + "." + scrub(report_name)
|
|
|
|
def add_total_row(result, columns, meta = None):
|
|
total_row = [""]*len(columns)
|
|
has_percent = []
|
|
for i, col in enumerate(columns):
|
|
fieldtype, options, fieldname = None, None, None
|
|
if isinstance(col, string_types):
|
|
if meta:
|
|
# get fieldtype from the meta
|
|
field = meta.get_field(col)
|
|
if field:
|
|
fieldtype = meta.get_field(col).fieldtype
|
|
fieldname = meta.get_field(col).fieldname
|
|
else:
|
|
col = col.split(":")
|
|
if len(col) > 1:
|
|
if col[1]:
|
|
fieldtype = col[1]
|
|
if "/" in fieldtype:
|
|
fieldtype, options = fieldtype.split("/")
|
|
else:
|
|
fieldtype = "Data"
|
|
else:
|
|
fieldtype = col.get("fieldtype")
|
|
fieldname = col.get("fieldname")
|
|
options = col.get("options")
|
|
|
|
for row in result:
|
|
cell = row.get(fieldname) if isinstance(row, dict) else row[i]
|
|
if fieldtype in ["Currency", "Int", "Float", "Percent"] and flt(cell):
|
|
total_row[i] = flt(total_row[i]) + flt(cell)
|
|
|
|
if fieldtype == "Percent" and i not in has_percent:
|
|
has_percent.append(i)
|
|
|
|
if fieldtype == "Time" and cell:
|
|
if not total_row[i]:
|
|
total_row[i]=timedelta(hours=0,minutes=0,seconds=0)
|
|
total_row[i] = total_row[i] + cell
|
|
|
|
|
|
if fieldtype=="Link" and options == "Currency":
|
|
total_row[i] = result[0][i]
|
|
|
|
for i in has_percent:
|
|
total_row[i] = flt(total_row[i]) / len(result)
|
|
|
|
first_col_fieldtype = None
|
|
if isinstance(columns[0], string_types):
|
|
first_col = columns[0].split(":")
|
|
if len(first_col) > 1:
|
|
first_col_fieldtype = first_col[1].split("/")[0]
|
|
else:
|
|
first_col_fieldtype = columns[0].get("fieldtype")
|
|
|
|
if first_col_fieldtype not in ["Currency", "Int", "Float", "Percent", "Date"]:
|
|
if first_col_fieldtype == "Link":
|
|
total_row[0] = "'" + _("Total") + "'"
|
|
else:
|
|
total_row[0] = _("Total")
|
|
|
|
result.append(total_row)
|
|
return result
|
|
|
|
|
|
def get_filtered_data(ref_doctype, columns, data, user):
|
|
result = []
|
|
linked_doctypes = get_linked_doctypes(columns, data)
|
|
match_filters_per_doctype = get_user_match_filters(linked_doctypes, user=user)
|
|
shared = frappe.share.get_shared(ref_doctype, user)
|
|
columns_dict = get_columns_dict(columns)
|
|
|
|
role_permissions = get_role_permissions(frappe.get_meta(ref_doctype), user)
|
|
if_owner = role_permissions.get("if_owner", {}).get("report")
|
|
|
|
if match_filters_per_doctype:
|
|
for row in data:
|
|
# Why linked_doctypes.get(ref_doctype)? because if column is empty, linked_doctypes[ref_doctype] is removed
|
|
if linked_doctypes.get(ref_doctype) and shared and row[linked_doctypes[ref_doctype]] in shared:
|
|
result.append(row)
|
|
|
|
elif has_match(row, linked_doctypes, match_filters_per_doctype, ref_doctype, if_owner, columns_dict, user):
|
|
result.append(row)
|
|
else:
|
|
result = list(data)
|
|
|
|
return result
|
|
|
|
|
|
def has_match(row, linked_doctypes, doctype_match_filters, ref_doctype, if_owner, columns_dict, user):
|
|
"""Returns True if after evaluating permissions for each linked doctype
|
|
- There is an owner match for the ref_doctype
|
|
- `and` There is a user permission match for all linked doctypes
|
|
|
|
Returns True if the row is empty
|
|
|
|
Note:
|
|
Each doctype could have multiple conflicting user permission doctypes.
|
|
Hence even if one of the sets allows a match, it is true.
|
|
This behavior is equivalent to the trickling of user permissions of linked doctypes to the ref doctype.
|
|
"""
|
|
resultant_match = True
|
|
|
|
if not row:
|
|
# allow empty rows :)
|
|
return resultant_match
|
|
|
|
for doctype, filter_list in doctype_match_filters.items():
|
|
matched_for_doctype = False
|
|
|
|
if doctype==ref_doctype and if_owner:
|
|
idx = linked_doctypes.get("User")
|
|
if (idx is not None
|
|
and row[idx]==user
|
|
and columns_dict[idx]==columns_dict.get("owner")):
|
|
# owner match is true
|
|
matched_for_doctype = True
|
|
|
|
if not matched_for_doctype:
|
|
for match_filters in filter_list:
|
|
match = True
|
|
for dt, idx in linked_doctypes.items():
|
|
# case handled above
|
|
if dt=="User" and columns_dict[idx]==columns_dict.get("owner"):
|
|
continue
|
|
|
|
cell_value = None
|
|
if isinstance(row, dict):
|
|
cell_value = row.get(idx)
|
|
elif isinstance(row, list):
|
|
cell_value = row[idx]
|
|
|
|
if dt in match_filters and cell_value not in match_filters.get(dt) and frappe.db.exists(dt, cell_value):
|
|
match = False
|
|
break
|
|
|
|
# each doctype could have multiple conflicting user permission doctypes, hence using OR
|
|
# so that even if one of the sets allows a match, it is true
|
|
matched_for_doctype = matched_for_doctype or match
|
|
|
|
if matched_for_doctype:
|
|
break
|
|
|
|
# each doctype's user permissions should match the row! hence using AND
|
|
resultant_match = resultant_match and matched_for_doctype
|
|
|
|
if not resultant_match:
|
|
break
|
|
|
|
return resultant_match
|
|
|
|
def get_linked_doctypes(columns, data):
|
|
linked_doctypes = {}
|
|
|
|
columns_dict = get_columns_dict(columns)
|
|
|
|
for idx, col in enumerate(columns):
|
|
df = columns_dict[idx]
|
|
if df.get("fieldtype")=="Link":
|
|
if isinstance(col, string_types):
|
|
linked_doctypes[df["options"]] = idx
|
|
else:
|
|
# dict
|
|
linked_doctypes[df["options"]] = df["fieldname"]
|
|
|
|
# remove doctype if column is empty
|
|
columns_with_value = []
|
|
for row in data:
|
|
if row:
|
|
if len(row) != len(columns_with_value):
|
|
if isinstance(row, (list, tuple)):
|
|
row = enumerate(row)
|
|
elif isinstance(row, dict):
|
|
row = row.items()
|
|
|
|
for col, val in row:
|
|
if val and col not in columns_with_value:
|
|
columns_with_value.append(col)
|
|
|
|
items = list(iteritems(linked_doctypes))
|
|
|
|
for doctype, key in items:
|
|
if key not in columns_with_value:
|
|
del linked_doctypes[doctype]
|
|
|
|
return linked_doctypes
|
|
|
|
def get_columns_dict(columns):
|
|
"""Returns a dict with column docfield values as dict
|
|
The keys for the dict are both idx and fieldname,
|
|
so either index or fieldname can be used to search for a column's docfield properties
|
|
"""
|
|
columns_dict = frappe._dict()
|
|
for idx, col in enumerate(columns):
|
|
col_dict = frappe._dict()
|
|
|
|
# string
|
|
if isinstance(col, string_types):
|
|
col = col.split(":")
|
|
if len(col) > 1:
|
|
if "/" in col[1]:
|
|
col_dict["fieldtype"], col_dict["options"] = col[1].split("/")
|
|
else:
|
|
col_dict["fieldtype"] = col[1]
|
|
|
|
col_dict["label"] = col[0]
|
|
col_dict["fieldname"] = frappe.scrub(col[0])
|
|
|
|
# dict
|
|
else:
|
|
col_dict.update(col)
|
|
if "fieldname" not in col_dict:
|
|
col_dict["fieldname"] = frappe.scrub(col_dict["label"])
|
|
|
|
columns_dict[idx] = col_dict
|
|
columns_dict[col_dict["fieldname"]] = col_dict
|
|
|
|
return columns_dict
|
|
|
|
def get_user_match_filters(doctypes, user):
|
|
match_filters = {}
|
|
|
|
for dt in doctypes:
|
|
filter_list = frappe.desk.reportview.build_match_conditions(dt, user, False)
|
|
if filter_list:
|
|
match_filters[dt] = filter_list
|
|
|
|
return match_filters
|