seitime-frappe/frappe/client.py

569 lines
16 KiB
Python

# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
# License: MIT. See LICENSE
import json
import os
from typing import TYPE_CHECKING, Any
import frappe
import frappe.model
import frappe.utils
from frappe import _
from frappe.desk.reportview import validate_args
from frappe.desk.search import PAGE_LENGTH_FOR_LINK_VALIDATION, search_widget
from frappe.utils import attach_expanded_links, get_safe_filters
from frappe.utils.caching import http_cache
if TYPE_CHECKING:
from frappe.model.document import Document
"""
Handle RESTful requests that are mapped to the `/api/resource` route.
Requests via FrappeClient are also handled here.
"""
@frappe.whitelist()
def get_list(
doctype: str,
fields: str | list[str | dict[str, Any]] | None = None,
filters: str | list | dict[str, Any] | None = None,
group_by: str | list[str] | None = None,
order_by: str | list[str] | None = None,
limit_start: int | str | None = None,
limit_page_length: int | str = 20,
parent: str | None = None,
debug: bool | int = False,
as_dict: bool | int = True,
or_filters: str | list[list] | dict[str, Any] | None = None,
expand: str | list[str] | None = None,
):
"""Return a list of records by filters, fields, ordering and limit.
:param doctype: DocType of the data to be queried
:param fields: fields to be returned. Default is `name`
:param filters: filter list by this dict
:param order_by: Order by this fieldname
:param limit_start: Start at this index
:param limit_page_length: Number of records to be returned (default 20)"""
args = frappe._dict(
doctype=doctype,
parent_doctype=parent,
fields=fields,
filters=filters,
or_filters=or_filters,
group_by=group_by,
order_by=order_by,
limit_start=limit_start,
limit_page_length=limit_page_length,
debug=debug,
as_list=not as_dict,
)
validate_args(args)
_list = frappe.get_list(**args)
if not expand:
return _list
if fields and not fields[0] == "*":
expand = [f for f in expand if f in fields]
attach_expanded_links(doctype, _list, expand)
return _list
@frappe.whitelist()
def get_count(
doctype: str,
filters: str | list | dict[str, Any] | None = None,
debug: int | bool = False,
cache: int | bool = False,
):
from frappe.desk.reportview import get_count
frappe.form_dict.doctype = doctype
frappe.form_dict.filters = get_safe_filters(filters)
frappe.form_dict.debug = debug
return get_count()
@frappe.whitelist()
def get(
doctype: str,
name: str | int | None = None,
filters: str | list | dict[str, Any] | None = None,
parent: str | None = None,
):
"""Return a document by name or filters.
:param doctype: DocType of the document to be returned
:param name: return document of this `name`
:param filters: If name is not set, filter by these values and return the first match"""
if name:
doc = frappe.get_doc(doctype, name)
elif filters or filters == {}:
doc = frappe.get_doc(doctype, frappe.parse_json(filters))
else:
doc = frappe.get_doc(doctype) # single
doc.check_permission()
doc.apply_fieldlevel_read_permissions()
return doc.as_dict(no_nulls=True)
@frappe.whitelist()
def get_value(
doctype: str,
fieldname: str | list[str] | dict[str, Any],
filters: str | list | dict[str, Any] | None = None,
as_dict: int | bool = True,
debug: int | bool = False,
parent: str | None = None,
):
"""Return a value from a document.
:param doctype: DocType to be queried
:param fieldname: Field to be returned (default `name`)
:param filters: dict or string for identifying the record"""
if not frappe.has_permission(doctype, parent_doctype=parent):
frappe.throw(_("No permission for {0}").format(_(doctype)), frappe.PermissionError)
filters = get_safe_filters(filters)
if isinstance(filters, str):
filters = {"name": filters}
try:
fields = frappe.parse_json(fieldname)
except (TypeError, ValueError):
# name passed, not json
fields = [fieldname]
# check whether the used filters were really parseable and usable
# and did not just result in an empty string or dict
if not filters:
filters = None
if frappe.get_meta(doctype).issingle:
value = frappe.db.get_values_from_single(fields, filters, doctype, as_dict=as_dict, debug=debug)
else:
value = get_list(
doctype,
filters=filters,
fields=fields,
debug=debug,
limit_page_length=1,
parent=parent,
as_dict=as_dict,
)
if as_dict:
return value[0] if value else {}
if not value:
return
return value[0] if len(fields) > 1 else value[0][0]
@frappe.whitelist()
def get_single_value(doctype: str, field: str):
if not frappe.has_permission(doctype):
frappe.throw(_("No permission for {0}").format(_(doctype)), frappe.PermissionError)
return frappe.db.get_single_value(doctype, field)
@frappe.whitelist(methods=["POST", "PUT"])
def set_value(doctype: str, name: str | int, fieldname: str | dict[str, Any], value: Any | None = None):
"""Set a value using get_doc, group of values
:param doctype: DocType of the document
:param name: name of the document
:param fieldname: fieldname string or JSON / dict with key value pair
:param value: value if fieldname is JSON / dict"""
if fieldname in (frappe.model.default_fields + frappe.model.child_table_fields):
frappe.throw(_("Cannot edit standard fields"))
if not value:
values = fieldname
if isinstance(fieldname, str):
try:
values = json.loads(fieldname)
except ValueError:
values = {fieldname: ""}
else:
values = {fieldname: value}
# check for child table doctype
if not frappe.get_meta(doctype).istable:
doc = frappe.get_doc(doctype, name)
doc.update(values)
else:
doc = frappe.db.get_value(doctype, name, ["parenttype", "parent"], as_dict=True)
doc = frappe.get_doc(doc.parenttype, doc.parent)
child = doc.getone({"doctype": doctype, "name": name})
child.update(values)
doc.save()
return doc.as_dict()
@frappe.whitelist(methods=["POST", "PUT"])
def insert(doc: str | dict[str, Any] | None = None):
"""Insert a document
:param doc: JSON or dict object to be inserted"""
if isinstance(doc, str):
doc = json.loads(doc)
return insert_doc(doc).as_dict()
@frappe.whitelist(methods=["POST", "PUT"])
def insert_many(docs: str | list[dict[str, Any]] | None = None):
"""Insert multiple documents
:param docs: JSON or list of dict objects to be inserted in one request"""
if isinstance(docs, str):
docs = json.loads(docs)
if len(docs) > 200:
frappe.throw(_("Only 200 inserts allowed in one request"))
return [insert_doc(doc).name for doc in docs]
@frappe.whitelist(methods=["POST", "PUT"])
def save(doc: str | dict[str, Any]):
"""Update (save) an existing document
:param doc: JSON or dict object with the properties of the document to be updated"""
if isinstance(doc, str):
doc = json.loads(doc)
doc = frappe.get_doc(doc)
doc.save()
return doc.as_dict()
@frappe.whitelist(methods=["POST", "PUT"])
def rename_doc(doctype: str, old_name: str | int, new_name: str | int, merge: bool = False):
"""Rename document
:param doctype: DocType of the document to be renamed
:param old_name: Current `name` of the document to be renamed
:param new_name: New `name` to be set"""
new_name = frappe.rename_doc(doctype, old_name, new_name, merge=merge)
return new_name
@frappe.whitelist(methods=["POST", "PUT"])
def submit(doc: str | dict[str, Any]):
"""Submit a document
:param doc: JSON or dict object to be submitted remotely"""
if isinstance(doc, str):
doc = json.loads(doc)
doc = frappe.get_doc(doc)
doc.submit()
return doc.as_dict()
@frappe.whitelist(methods=["POST", "PUT"])
def cancel(doctype: str, name: str | int):
"""Cancel a document
:param doctype: DocType of the document to be cancelled
:param name: name of the document to be cancelled"""
wrapper = frappe.get_doc(doctype, name)
wrapper.cancel()
return wrapper.as_dict()
@frappe.whitelist(methods=["DELETE", "POST"])
def delete(doctype: str, name: str | int):
"""Delete a remote document
:param doctype: DocType of the document to be deleted
:param name: name of the document to be deleted"""
delete_doc(doctype, name)
@frappe.whitelist(methods=["POST", "PUT"])
def bulk_update(docs: str):
"""Bulk update documents
:param docs: JSON list of documents to be updated remotely. Each document must have `docname` property"""
docs = json.loads(docs)
failed_docs = []
for doc in docs:
doc.pop("flags", None)
try:
existing_doc = frappe.get_doc(doc["doctype"], doc["docname"])
existing_doc.update(doc)
existing_doc.save()
except Exception:
failed_docs.append({"doc": doc, "exc": frappe.utils.get_traceback()})
return {"failed_docs": failed_docs}
@frappe.whitelist()
def has_permission(doctype: str, docname: str | int, perm_type: str = "read"):
"""Return a JSON with data whether the document has the requested permission.
:param doctype: DocType of the document to be checked
:param docname: `name` of the document to be checked
:param perm_type: one of `read`, `write`, `create`, `submit`, `cancel`, `report`. Default is `read`"""
# perm_type can be one of read, write, create, submit, cancel, report
return {"has_permission": frappe.has_permission(doctype, perm_type.lower(), docname)}
@frappe.whitelist()
def get_doc_permissions(doctype: str, docname: str | int):
"""Return an evaluated document permissions dict like `{"read":1, "write":1}`.
:param doctype: DocType of the document to be evaluated
:param docname: `name` of the document to be evaluated
"""
doc = frappe.get_lazy_doc(doctype, docname)
return {"permissions": frappe.permissions.get_doc_permissions(doc)}
@frappe.whitelist()
def get_password(doctype: str, name: str | int, fieldname: str):
"""Return a password type property. Only applicable for System Managers
:param doctype: DocType of the document that holds the password
:param name: `name` of the document that holds the password
:param fieldname: `fieldname` of the password property
"""
frappe.only_for("System Manager")
return frappe.get_lazy_doc(doctype, name).get_password(fieldname)
from frappe.deprecation_dumpster import get_js as _get_js
get_js = frappe.whitelist()(_get_js)
@frappe.whitelist(allow_guest=True)
def get_time_zone():
"""Return the default time zone."""
return {"time_zone": frappe.defaults.get_defaults().get("time_zone")}
@frappe.whitelist(methods=["POST", "PUT"])
def attach_file(
filename: str | None = None,
filedata: str | None = None,
doctype: str | None = None,
docname: str | int | None = None,
folder: str | None = None,
decode_base64: int | bool = False,
is_private: int | bool | None = None,
docfield: str | None = None,
):
"""Attach a file to Document
:param filename: filename e.g. test-file.txt
:param filedata: base64 encode filedata which must be urlencoded
:param doctype: Reference DocType to attach file to
:param docname: Reference DocName to attach file to
:param folder: Folder to add File into
:param decode_base64: decode filedata from base64 encode, default is False
:param is_private: Attach file as private file (1 or 0)
:param docfield: file to attach to (optional)"""
doc = frappe.get_lazy_doc(doctype, docname, check_permission=True)
file = frappe.get_doc(
{
"doctype": "File",
"file_name": filename,
"attached_to_doctype": doctype,
"attached_to_name": docname,
"attached_to_field": docfield,
"folder": folder,
"is_private": is_private,
"content": filedata,
"decode": decode_base64,
}
).save()
if docfield and doctype:
doc.set(docfield, file.file_url)
doc.save()
return file
@frappe.whitelist()
@http_cache(max_age=10 * 60)
def is_document_amended(doctype: str, docname: str | int):
if frappe.permissions.has_permission(doctype):
try:
return frappe.db.exists(doctype, {"amended_from": docname})
except frappe.db.InternalError:
pass
return False
@frappe.whitelist(methods=["GET", "POST"])
def validate_link_and_fetch(
doctype: str,
docname: str | int,
fields_to_fetch: list[str] | str | None = None,
# search_widget parameters
query: str | None = None,
filters: dict | list | str | None = None,
**search_args,
):
if not docname:
frappe.throw(_("Document Name must not be empty"))
meta = frappe.get_meta(doctype)
fields_to_fetch = frappe.parse_json(fields_to_fetch)
# only cache is no fields to fetch and request is GET
can_cache = not fields_to_fetch and frappe.request.method == "GET"
# Use search_widget to validate - ensures filters/custom queries are respected
# in addition to standard permission checks
# we match the exact docname for non-custom queries and rely on txt for custom queries
search_args.update(
as_dict=False,
# when relying on txt (custom queries), we want to match "A" with "A" only and not "A1", "BA" etc.
# so we set page_length to a conservative value within which exact match is expected to appear
page_length=PAGE_LENGTH_FOR_LINK_VALIDATION,
# translated doctypes are expected to be searchable with translated values, even for custom queries
# for non-custom queries, docname is always matched exactly so we don't translate it
txt=_(docname) if (query and meta.translated_doctype) else docname,
for_link_validation=True,
)
search_result = frappe.call(
search_widget,
doctype=doctype,
query=query,
filters=filters,
**search_args,
)
if not search_result:
return {} # Either the record does not exist or was excluded by link_filters
values = None
is_virtual_dt = bool(meta.get("is_virtual"))
if is_virtual_dt:
try:
doc = frappe.get_doc(doctype, docname)
doc.check_permission("select")
values = {"name": doc.name}
except frappe.DoesNotExistError:
frappe.clear_last_message()
else:
# get value in the right case and type (str | int)
# for matching with search result
columns_to_fetch = ["name"]
if frappe.is_table(doctype):
columns_to_fetch.append("parenttype") # for child table permission check
values = frappe.db.get_value(doctype, docname, columns_to_fetch, as_dict=True)
if not values:
return {} # does not exist
name_to_compare = values["name"]
# this will be used to fetch fields later
parent_doctype = values.pop("parenttype", None)
# try to match name in search result
# if search_result is large, assume valid link (result may not appear in some custom queries)
if len(search_result) < PAGE_LENGTH_FOR_LINK_VALIDATION and not any(
item[0] == name_to_compare for item in search_result
):
return {} # no permission or filtered out
# don't cache or fetch for virtual doctypes
if is_virtual_dt:
return values
if not fields_to_fetch:
if can_cache:
frappe.local.response_headers.set(
"Cache-Control", "private,max-age=1800,stale-while-revalidate=7200"
)
return values
try:
values.update(get_value(doctype, fields_to_fetch, docname, parent=parent_doctype))
except frappe.PermissionError:
frappe.clear_last_message()
frappe.msgprint(
_("You need {0} permission to fetch values from {1} {2}").format(
frappe.bold(_("Read")), frappe.bold(doctype), frappe.bold(docname)
),
title=_("Cannot Fetch Values"),
indicator="orange",
)
return values
def insert_doc(doc) -> "Document":
"""Insert document and return parent document object with appended child document if `doc` is child document else return the inserted document object.
:param doc: doc to insert (dict)"""
doc = frappe._dict(doc)
if frappe.is_table(doc.doctype):
if not (doc.parenttype and doc.parent and doc.parentfield):
frappe.throw(_("Parenttype, Parent and Parentfield are required to insert a child record"))
# inserting a child record
parent = frappe.get_doc(doc.parenttype, doc.parent)
parent.append(doc.parentfield, doc)
parent.save()
return parent
return frappe.get_doc(doc).insert()
def delete_doc(doctype, name):
"""Deletes document
if doctype is a child table, then deletes the child record using the parent doc
so that the parent doc's `on_update` is called
"""
if frappe.is_table(doctype):
values = frappe.db.get_value(doctype, name, ["parenttype", "parent", "parentfield"])
if not values:
raise frappe.DoesNotExistError(doctype=doctype)
parenttype, parent, parentfield = values
parent = frappe.get_doc(parenttype, parent)
if not parent.has_permission("write"):
raise frappe.DoesNotExistError(doctype=doctype)
for row in parent.get(parentfield):
if row.name == name:
parent.remove(row)
parent.save()
break
else:
frappe.delete_doc(doctype, name, ignore_missing=False)