Nevar pievienot vairāk kā 25 tēmas Tēmai ir jāsākas ar burtu vai ciparu, tā var saturēt domu zīmes ('-') un var būt līdz 35 simboliem gara.
 
 
 
 
 
 

1415 rindas
43 KiB

  1. # Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
  2. # License: MIT. See LICENSE
  3. import hashlib
  4. import json
  5. import time
  6. from werkzeug.exceptions import NotFound
  7. import frappe
  8. from frappe import _, msgprint, is_whitelisted
  9. from frappe.utils import flt, cstr, now, get_datetime_str, file_lock, date_diff
  10. from frappe.model.base_document import BaseDocument, get_controller
  11. from frappe.model.naming import set_new_name
  12. from frappe.model.docstatus import DocStatus
  13. from frappe.model import optional_fields, table_fields
  14. from frappe.model.workflow import validate_workflow
  15. from frappe.model.workflow import set_workflow_state_on_action
  16. from frappe.utils.global_search import update_global_search
  17. from frappe.integrations.doctype.webhook import run_webhooks
  18. from frappe.desk.form.document_follow import follow_document
  19. from frappe.core.doctype.server_script.server_script_utils import run_server_script_for_doc_event
  20. from frappe.utils.data import get_absolute_url
  21. # once_only validation
  22. # methods
  23. def get_doc(*args, **kwargs):
  24. """returns a frappe.model.Document object.
  25. :param arg1: Document dict or DocType name.
  26. :param arg2: [optional] document name.
  27. :param for_update: [optional] select document for update.
  28. There are multiple ways to call `get_doc`
  29. # will fetch the latest user object (with child table) from the database
  30. user = get_doc("User", "test@example.com")
  31. # create a new object
  32. user = get_doc({
  33. "doctype":"User"
  34. "email_id": "test@example.com",
  35. "roles: [
  36. {"role": "System Manager"}
  37. ]
  38. })
  39. # create new object with keyword arguments
  40. user = get_doc(doctype='User', email_id='test@example.com')
  41. # select a document for update
  42. user = get_doc("User", "test@example.com", for_update=True)
  43. """
  44. if args:
  45. if isinstance(args[0], BaseDocument):
  46. # already a document
  47. return args[0]
  48. elif isinstance(args[0], str):
  49. doctype = args[0]
  50. elif isinstance(args[0], dict):
  51. # passed a dict
  52. kwargs = args[0]
  53. else:
  54. raise ValueError('First non keyword argument must be a string or dict')
  55. if len(args) < 2 and kwargs:
  56. if 'doctype' in kwargs:
  57. doctype = kwargs['doctype']
  58. else:
  59. raise ValueError('"doctype" is a required key')
  60. controller = get_controller(doctype)
  61. if controller:
  62. return controller(*args, **kwargs)
  63. raise ImportError(doctype)
  64. class Document(BaseDocument):
  65. """All controllers inherit from `Document`."""
  66. def __init__(self, *args, **kwargs):
  67. """Constructor.
  68. :param arg1: DocType name as string or document **dict**
  69. :param arg2: Document name, if `arg1` is DocType name.
  70. If DocType name and document name are passed, the object will load
  71. all values (including child documents) from the database.
  72. """
  73. self.doctype = self.name = None
  74. self._default_new_docs = {}
  75. self.flags = frappe._dict()
  76. if args and args[0] and isinstance(args[0], str):
  77. # first arugment is doctype
  78. if len(args)==1:
  79. # single
  80. self.doctype = self.name = args[0]
  81. else:
  82. self.doctype = args[0]
  83. if isinstance(args[1], dict):
  84. # filter
  85. self.name = frappe.db.get_value(args[0], args[1], "name")
  86. if self.name is None:
  87. frappe.throw(_("{0} {1} not found").format(_(args[0]), args[1]),
  88. frappe.DoesNotExistError)
  89. else:
  90. self.name = args[1]
  91. if 'for_update' in kwargs:
  92. self.flags.for_update = kwargs.get('for_update')
  93. self.load_from_db()
  94. return
  95. if args and args[0] and isinstance(args[0], dict):
  96. # first argument is a dict
  97. kwargs = args[0]
  98. if kwargs:
  99. # init base document
  100. super(Document, self).__init__(kwargs)
  101. self.init_valid_columns()
  102. else:
  103. # incorrect arguments. let's not proceed.
  104. raise ValueError('Illegal arguments')
  105. @staticmethod
  106. def whitelist(fn):
  107. """Decorator: Whitelist method to be called remotely via REST API."""
  108. frappe.whitelist()(fn)
  109. return fn
  110. def reload(self):
  111. """Reload document from database"""
  112. self.load_from_db()
  113. def load_from_db(self):
  114. """Load document and children from database and create properties
  115. from fields"""
  116. if not getattr(self, "_metaclass", False) and self.meta.issingle:
  117. single_doc = frappe.db.get_singles_dict(self.doctype)
  118. if not single_doc:
  119. single_doc = frappe.new_doc(self.doctype).as_dict()
  120. single_doc["name"] = self.doctype
  121. del single_doc["__islocal"]
  122. super(Document, self).__init__(single_doc)
  123. self.init_valid_columns()
  124. self._fix_numeric_types()
  125. else:
  126. d = frappe.db.get_value(self.doctype, self.name, "*", as_dict=1, for_update=self.flags.for_update)
  127. if not d:
  128. frappe.throw(_("{0} {1} not found").format(_(self.doctype), self.name), frappe.DoesNotExistError)
  129. super(Document, self).__init__(d)
  130. if self.name=="DocType" and self.doctype=="DocType":
  131. from frappe.model.meta import DOCTYPE_TABLE_FIELDS
  132. table_fields = DOCTYPE_TABLE_FIELDS
  133. else:
  134. table_fields = self.meta.get_table_fields()
  135. for df in table_fields:
  136. children = frappe.db.get_values(df.options,
  137. {"parent": self.name, "parenttype": self.doctype, "parentfield": df.fieldname},
  138. "*", as_dict=True, order_by="idx asc")
  139. if children:
  140. self.set(df.fieldname, children)
  141. else:
  142. self.set(df.fieldname, [])
  143. # sometimes __setup__ can depend on child values, hence calling again at the end
  144. if hasattr(self, "__setup__"):
  145. self.__setup__()
  146. def get_latest(self):
  147. if not getattr(self, "latest", None):
  148. self.latest = frappe.get_doc(self.doctype, self.name)
  149. return self.latest
  150. def check_permission(self, permtype='read', permlevel=None):
  151. """Raise `frappe.PermissionError` if not permitted"""
  152. if not self.has_permission(permtype):
  153. self.raise_no_permission_to(permlevel or permtype)
  154. def has_permission(self, permtype="read", verbose=False):
  155. """Call `frappe.has_permission` if `self.flags.ignore_permissions`
  156. is not set.
  157. :param permtype: one of `read`, `write`, `submit`, `cancel`, `delete`"""
  158. import frappe.permissions
  159. if self.flags.ignore_permissions:
  160. return True
  161. return frappe.permissions.has_permission(self.doctype, permtype, self, verbose=verbose)
  162. def raise_no_permission_to(self, perm_type):
  163. """Raise `frappe.PermissionError`."""
  164. frappe.flags.error_message = _('Insufficient Permission for {0}').format(self.doctype)
  165. raise frappe.PermissionError
  166. def insert(self, ignore_permissions=None, ignore_links=None, ignore_if_duplicate=False,
  167. ignore_mandatory=None, set_name=None, set_child_names=True):
  168. """Insert the document in the database (as a new document).
  169. This will check for user permissions and execute `before_insert`,
  170. `validate`, `on_update`, `after_insert` methods if they are written.
  171. :param ignore_permissions: Do not check permissions if True."""
  172. if self.flags.in_print:
  173. return
  174. self.flags.notifications_executed = []
  175. if ignore_permissions is not None:
  176. self.flags.ignore_permissions = ignore_permissions
  177. if ignore_links is not None:
  178. self.flags.ignore_links = ignore_links
  179. if ignore_mandatory is not None:
  180. self.flags.ignore_mandatory = ignore_mandatory
  181. self.set("__islocal", True)
  182. self._set_defaults()
  183. self.set_user_and_timestamp()
  184. self.set_docstatus()
  185. self.check_if_latest()
  186. self._validate_links()
  187. self.check_permission("create")
  188. self.run_method("before_insert")
  189. self.set_new_name(set_name=set_name, set_child_names=set_child_names)
  190. self.set_parent_in_children()
  191. self.validate_higher_perm_levels()
  192. self.flags.in_insert = True
  193. self.run_before_save_methods()
  194. self._validate()
  195. self.set_docstatus()
  196. self.flags.in_insert = False
  197. # run validate, on update etc.
  198. # parent
  199. if getattr(self.meta, "issingle", 0):
  200. self.update_single(self.get_valid_dict())
  201. else:
  202. try:
  203. self.db_insert()
  204. except frappe.DuplicateEntryError as e:
  205. if not ignore_if_duplicate:
  206. raise e
  207. # children
  208. for d in self.get_all_children():
  209. d.db_insert()
  210. self.run_method("after_insert")
  211. self.flags.in_insert = True
  212. if self.get("amended_from"):
  213. self.copy_attachments_from_amended_from()
  214. # flag to prevent creation of event update log for create and update both
  215. # during document creation
  216. self.flags.update_log_for_doc_creation = True
  217. self.run_post_save_methods()
  218. self.flags.in_insert = False
  219. # delete __islocal
  220. if hasattr(self, "__islocal"):
  221. delattr(self, "__islocal")
  222. # clear unsaved flag
  223. if hasattr(self, "__unsaved"):
  224. delattr(self, "__unsaved")
  225. if not (frappe.flags.in_migrate or frappe.local.flags.in_install or frappe.flags.in_setup_wizard):
  226. follow_document(self.doctype, self.name, frappe.session.user)
  227. return self
  228. def save(self, *args, **kwargs):
  229. """Wrapper for _save"""
  230. return self._save(*args, **kwargs)
  231. def _save(self, ignore_permissions=None, ignore_version=None):
  232. """Save the current document in the database in the **DocType**'s table or
  233. `tabSingles` (for single types).
  234. This will check for user permissions and execute
  235. `validate` before updating, `on_update` after updating triggers.
  236. :param ignore_permissions: Do not check permissions if True.
  237. :param ignore_version: Do not save version if True."""
  238. if self.flags.in_print:
  239. return
  240. self.flags.notifications_executed = []
  241. if ignore_permissions is not None:
  242. self.flags.ignore_permissions = ignore_permissions
  243. self.flags.ignore_version = frappe.flags.in_test if ignore_version is None else ignore_version
  244. if self.get("__islocal") or not self.get("name"):
  245. return self.insert()
  246. self.check_permission("write", "save")
  247. self.set_user_and_timestamp()
  248. self.set_docstatus()
  249. self.check_if_latest()
  250. self.set_parent_in_children()
  251. self.set_name_in_children()
  252. self.validate_higher_perm_levels()
  253. self._validate_links()
  254. self.run_before_save_methods()
  255. if self._action != "cancel":
  256. self._validate()
  257. if self._action == "update_after_submit":
  258. self.validate_update_after_submit()
  259. self.set_docstatus()
  260. # parent
  261. if self.meta.issingle:
  262. self.update_single(self.get_valid_dict())
  263. else:
  264. self.db_update()
  265. self.update_children()
  266. self.run_post_save_methods()
  267. # clear unsaved flag
  268. if hasattr(self, "__unsaved"):
  269. delattr(self, "__unsaved")
  270. return self
  271. def copy_attachments_from_amended_from(self):
  272. """Copy attachments from `amended_from`"""
  273. from frappe.desk.form.load import get_attachments
  274. #loop through attachments
  275. for attach_item in get_attachments(self.doctype, self.amended_from):
  276. #save attachments to new doc
  277. _file = frappe.get_doc({
  278. "doctype": "File",
  279. "file_url": attach_item.file_url,
  280. "file_name": attach_item.file_name,
  281. "attached_to_name": self.name,
  282. "attached_to_doctype": self.doctype,
  283. "folder": "Home/Attachments"})
  284. _file.save()
  285. def update_children(self):
  286. """update child tables"""
  287. for df in self.meta.get_table_fields():
  288. self.update_child_table(df.fieldname, df)
  289. def update_child_table(self, fieldname, df=None):
  290. """sync child table for given fieldname"""
  291. rows = []
  292. if not df:
  293. df = self.meta.get_field(fieldname)
  294. for d in self.get(df.fieldname):
  295. d.db_update()
  296. rows.append(d.name)
  297. if df.options in (self.flags.ignore_children_type or []):
  298. # do not delete rows for this because of flags
  299. # hack for docperm :(
  300. return
  301. if rows:
  302. # select rows that do not match the ones in the document
  303. deleted_rows = frappe.db.sql("""select name from `tab{0}` where parent=%s
  304. and parenttype=%s and parentfield=%s
  305. and name not in ({1})""".format(df.options, ','.join(['%s'] * len(rows))),
  306. [self.name, self.doctype, fieldname] + rows)
  307. if len(deleted_rows) > 0:
  308. # delete rows that do not match the ones in the document
  309. frappe.db.delete(df.options, {"name": ("in", tuple(row[0] for row in deleted_rows))})
  310. else:
  311. # no rows found, delete all rows
  312. frappe.db.delete(df.options, {
  313. "parent": self.name,
  314. "parenttype": self.doctype,
  315. "parentfield": fieldname
  316. })
  317. def get_doc_before_save(self):
  318. return getattr(self, '_doc_before_save', None)
  319. def has_value_changed(self, fieldname):
  320. '''Returns true if value is changed before and after saving'''
  321. previous = self.get_doc_before_save()
  322. return previous.get(fieldname)!=self.get(fieldname) if previous else True
  323. def set_new_name(self, force=False, set_name=None, set_child_names=True):
  324. """Calls `frappe.naming.set_new_name` for parent and child docs."""
  325. if self.flags.name_set and not force:
  326. return
  327. # If autoname has set as Prompt (name)
  328. if self.get("__newname"):
  329. self.name = self.get("__newname")
  330. self.flags.name_set = True
  331. return
  332. if set_name:
  333. self.name = set_name
  334. else:
  335. set_new_name(self)
  336. if set_child_names:
  337. # set name for children
  338. for d in self.get_all_children():
  339. set_new_name(d)
  340. self.flags.name_set = True
  341. def get_title(self):
  342. """Get the document title based on title_field or `title` or `name`"""
  343. return self.get(self.meta.get_title_field())
  344. def set_title_field(self):
  345. """Set title field based on template"""
  346. def get_values():
  347. values = self.as_dict()
  348. # format values
  349. for key, value in values.items():
  350. if value is None:
  351. values[key] = ""
  352. return values
  353. if self.meta.get("title_field")=="title":
  354. df = self.meta.get_field(self.meta.title_field)
  355. if df.options:
  356. self.set(df.fieldname, df.options.format(**get_values()))
  357. elif self.is_new() and not self.get(df.fieldname) and df.default:
  358. # set default title for new transactions (if default)
  359. self.set(df.fieldname, df.default.format(**get_values()))
  360. def update_single(self, d):
  361. """Updates values for Single type Document in `tabSingles`."""
  362. frappe.db.delete("Singles", {
  363. "doctype": self.doctype
  364. })
  365. for field, value in d.items():
  366. if field != "doctype":
  367. frappe.db.sql("""insert into `tabSingles` (doctype, field, value)
  368. values (%s, %s, %s)""", (self.doctype, field, value))
  369. if self.doctype in frappe.db.value_cache:
  370. del frappe.db.value_cache[self.doctype]
  371. def set_user_and_timestamp(self):
  372. self._original_modified = self.modified
  373. self.modified = now()
  374. self.modified_by = frappe.session.user
  375. # We'd probably want the creation and owner to be set via API
  376. # or Data import at some point, that'd have to be handled here
  377. if self.is_new() and not (frappe.flags.in_patch or frappe.flags.in_migrate):
  378. self.creation = self.modified
  379. self.owner = self.modified_by
  380. for d in self.get_all_children():
  381. d.modified = self.modified
  382. d.modified_by = self.modified_by
  383. if not d.owner:
  384. d.owner = self.owner
  385. if not d.creation:
  386. d.creation = self.creation
  387. frappe.flags.currently_saving.append((self.doctype, self.name))
  388. def set_docstatus(self):
  389. if self.docstatus is None:
  390. self.docstatus = DocStatus.draft()
  391. for d in self.get_all_children():
  392. d.docstatus = self.docstatus
  393. def _validate(self):
  394. self._validate_mandatory()
  395. self._validate_data_fields()
  396. self._validate_selects()
  397. self._validate_non_negative()
  398. self._validate_length()
  399. self._validate_code_fields()
  400. self._extract_images_from_text_editor()
  401. self._sanitize_content()
  402. self._save_passwords()
  403. self.validate_workflow()
  404. children = self.get_all_children()
  405. for d in children:
  406. d._validate_data_fields()
  407. d._validate_selects()
  408. d._validate_non_negative()
  409. d._validate_length()
  410. d._validate_code_fields()
  411. d._extract_images_from_text_editor()
  412. d._sanitize_content()
  413. d._save_passwords()
  414. if self.is_new():
  415. # don't set fields like _assign, _comments for new doc
  416. for fieldname in optional_fields:
  417. self.set(fieldname, None)
  418. else:
  419. self.validate_set_only_once()
  420. def _validate_non_negative(self):
  421. def get_msg(df):
  422. if self.get("parentfield"):
  423. return "{} {} #{}: {} {}".format(frappe.bold(_(self.doctype)),
  424. _("Row"), self.idx, _("Value cannot be negative for"), frappe.bold(_(df.label)))
  425. else:
  426. return _("Value cannot be negative for {0}: {1}").format(_(df.parent), frappe.bold(_(df.label)))
  427. for df in self.meta.get('fields', {'non_negative': ('=', 1),
  428. 'fieldtype': ('in', ['Int', 'Float', 'Currency'])}):
  429. if flt(self.get(df.fieldname)) < 0:
  430. msg = get_msg(df)
  431. frappe.throw(msg, frappe.NonNegativeError, title=_("Negative Value"))
  432. def validate_workflow(self):
  433. """Validate if the workflow transition is valid"""
  434. if frappe.flags.in_install == 'frappe': return
  435. workflow = self.meta.get_workflow()
  436. if workflow:
  437. validate_workflow(self)
  438. if not self._action == 'save':
  439. set_workflow_state_on_action(self, workflow, self._action)
  440. def validate_set_only_once(self):
  441. """Validate that fields are not changed if not in insert"""
  442. set_only_once_fields = self.meta.get_set_only_once_fields()
  443. if set_only_once_fields and self._doc_before_save:
  444. # document exists before saving
  445. for field in set_only_once_fields:
  446. fail = False
  447. value = self.get(field.fieldname)
  448. original_value = self._doc_before_save.get(field.fieldname)
  449. if field.fieldtype in table_fields:
  450. fail = not self.is_child_table_same(field.fieldname)
  451. elif field.fieldtype in ('Date', 'Datetime', 'Time'):
  452. fail = str(value) != str(original_value)
  453. else:
  454. fail = value != original_value
  455. if fail:
  456. frappe.throw(
  457. _("Value cannot be changed for {0}").format(
  458. frappe.bold(self.meta.get_label(field.fieldname))
  459. ),
  460. exc=frappe.CannotChangeConstantError
  461. )
  462. return False
  463. def is_child_table_same(self, fieldname):
  464. """Validate child table is same as original table before saving"""
  465. value = self.get(fieldname)
  466. original_value = self._doc_before_save.get(fieldname)
  467. same = True
  468. if len(original_value) != len(value):
  469. same = False
  470. else:
  471. # check all child entries
  472. for i, d in enumerate(original_value):
  473. new_child = value[i].as_dict(convert_dates_to_str = True)
  474. original_child = d.as_dict(convert_dates_to_str = True)
  475. # all fields must be same other than modified and modified_by
  476. for key in ('modified', 'modified_by', 'creation'):
  477. del new_child[key]
  478. del original_child[key]
  479. if original_child != new_child:
  480. same = False
  481. break
  482. return same
  483. def apply_fieldlevel_read_permissions(self):
  484. """Remove values the user is not allowed to read (called when loading in desk)"""
  485. if frappe.session.user == "Administrator":
  486. return
  487. has_higher_permlevel = False
  488. all_fields = self.meta.fields.copy()
  489. for table_field in self.meta.get_table_fields():
  490. all_fields += frappe.get_meta(table_field.options).fields or []
  491. for df in all_fields:
  492. if df.permlevel > 0:
  493. has_higher_permlevel = True
  494. break
  495. if not has_higher_permlevel:
  496. return
  497. has_access_to = self.get_permlevel_access('read')
  498. for df in self.meta.fields:
  499. if df.permlevel and not df.permlevel in has_access_to:
  500. self.set(df.fieldname, None)
  501. for table_field in self.meta.get_table_fields():
  502. for df in frappe.get_meta(table_field.options).fields or []:
  503. if df.permlevel and not df.permlevel in has_access_to:
  504. for child in self.get(table_field.fieldname) or []:
  505. child.set(df.fieldname, None)
  506. def validate_higher_perm_levels(self):
  507. """If the user does not have permissions at permlevel > 0, then reset the values to original / default"""
  508. if self.flags.ignore_permissions or frappe.flags.in_install:
  509. return
  510. if frappe.session.user == "Administrator":
  511. return
  512. has_access_to = self.get_permlevel_access()
  513. high_permlevel_fields = self.meta.get_high_permlevel_fields()
  514. if high_permlevel_fields:
  515. self.reset_values_if_no_permlevel_access(has_access_to, high_permlevel_fields)
  516. # If new record then don't reset the values for child table
  517. if self.is_new(): return
  518. # check for child tables
  519. for df in self.meta.get_table_fields():
  520. high_permlevel_fields = frappe.get_meta(df.options).get_high_permlevel_fields()
  521. if high_permlevel_fields:
  522. for d in self.get(df.fieldname):
  523. d.reset_values_if_no_permlevel_access(has_access_to, high_permlevel_fields)
  524. def get_permlevel_access(self, permission_type='write'):
  525. if not hasattr(self, "_has_access_to"):
  526. self._has_access_to = {}
  527. self._has_access_to[permission_type] = []
  528. roles = frappe.get_roles()
  529. for perm in self.get_permissions():
  530. if perm.role in roles and perm.get(permission_type):
  531. if perm.permlevel not in self._has_access_to[permission_type]:
  532. self._has_access_to[permission_type].append(perm.permlevel)
  533. return self._has_access_to[permission_type]
  534. def has_permlevel_access_to(self, fieldname, df=None, permission_type='read'):
  535. if not df:
  536. df = self.meta.get_field(fieldname)
  537. return df.permlevel in self.get_permlevel_access(permission_type)
  538. def get_permissions(self):
  539. if self.meta.istable:
  540. # use parent permissions
  541. permissions = frappe.get_meta(self.parenttype).permissions
  542. else:
  543. permissions = self.meta.permissions
  544. return permissions
  545. def _set_defaults(self):
  546. if frappe.flags.in_import:
  547. return
  548. new_doc = frappe.new_doc(self.doctype, as_dict=True)
  549. self.update_if_missing(new_doc)
  550. # children
  551. for df in self.meta.get_table_fields():
  552. new_doc = frappe.new_doc(df.options, as_dict=True)
  553. value = self.get(df.fieldname)
  554. if isinstance(value, list):
  555. for d in value:
  556. d.update_if_missing(new_doc)
  557. def check_if_latest(self):
  558. """Checks if `modified` timestamp provided by document being updated is same as the
  559. `modified` timestamp in the database. If there is a different, the document has been
  560. updated in the database after the current copy was read. Will throw an error if
  561. timestamps don't match.
  562. Will also validate document transitions (Save > Submit > Cancel) calling
  563. `self.check_docstatus_transition`."""
  564. conflict = False
  565. self._action = "save"
  566. if not self.get('__islocal') and not self.meta.get('is_virtual'):
  567. if self.meta.issingle:
  568. modified = frappe.db.sql("""select value from tabSingles
  569. where doctype=%s and field='modified' for update""", self.doctype)
  570. modified = modified and modified[0][0]
  571. if modified and modified != cstr(self._original_modified):
  572. conflict = True
  573. else:
  574. tmp = frappe.db.sql("""select modified, docstatus from `tab{0}`
  575. where name = %s for update""".format(self.doctype), self.name, as_dict=True)
  576. if not tmp:
  577. frappe.throw(_("Record does not exist"))
  578. else:
  579. tmp = tmp[0]
  580. modified = cstr(tmp.modified)
  581. if modified and modified != cstr(self._original_modified):
  582. conflict = True
  583. self.check_docstatus_transition(tmp.docstatus)
  584. if conflict:
  585. frappe.msgprint(_("Error: Document has been modified after you have opened it") \
  586. + (" (%s, %s). " % (modified, self.modified)) \
  587. + _("Please refresh to get the latest document."),
  588. raise_exception=frappe.TimestampMismatchError)
  589. else:
  590. self.check_docstatus_transition(0)
  591. def check_docstatus_transition(self, to_docstatus):
  592. """Ensures valid `docstatus` transition.
  593. Valid transitions are (number in brackets is `docstatus`):
  594. - Save (0) > Save (0)
  595. - Save (0) > Submit (1)
  596. - Submit (1) > Submit (1)
  597. - Submit (1) > Cancel (2)
  598. """
  599. if not self.docstatus:
  600. self.docstatus = DocStatus.draft()
  601. if to_docstatus == DocStatus.draft():
  602. if self.docstatus.is_draft():
  603. self._action = "save"
  604. elif self.docstatus.is_submitted():
  605. self._action = "submit"
  606. self.check_permission("submit")
  607. elif self.docstatus.is_cancelled():
  608. raise frappe.DocstatusTransitionError(_("Cannot change docstatus from 0 (Draft) to 2 (Cancelled)"))
  609. else:
  610. raise frappe.ValidationError(_("Invalid docstatus"), self.docstatus)
  611. elif to_docstatus == DocStatus.submitted():
  612. if self.docstatus.is_submitted():
  613. self._action = "update_after_submit"
  614. self.check_permission("submit")
  615. elif self.docstatus.is_cancelled():
  616. self._action = "cancel"
  617. self.check_permission("cancel")
  618. elif self.docstatus.is_draft():
  619. raise frappe.DocstatusTransitionError(_("Cannot change docstatus from 1 (Submitted) to 0 (Draft)"))
  620. else:
  621. raise frappe.ValidationError(_("Invalid docstatus"), self.docstatus)
  622. elif to_docstatus == DocStatus.cancelled():
  623. raise frappe.ValidationError(_("Cannot edit cancelled document"))
  624. def set_parent_in_children(self):
  625. """Updates `parent` and `parenttype` property in all children."""
  626. for d in self.get_all_children():
  627. d.parent = self.name
  628. d.parenttype = self.doctype
  629. def set_name_in_children(self):
  630. # Set name for any new children
  631. for d in self.get_all_children():
  632. if not d.name:
  633. set_new_name(d)
  634. def validate_update_after_submit(self):
  635. if self.flags.ignore_validate_update_after_submit:
  636. return
  637. self._validate_update_after_submit()
  638. for d in self.get_all_children():
  639. if d.is_new() and self.meta.get_field(d.parentfield).allow_on_submit:
  640. # in case of a new row, don't validate allow on submit, if table is allow on submit
  641. continue
  642. d._validate_update_after_submit()
  643. # TODO check only allowed values are updated
  644. def _validate_mandatory(self):
  645. if self.flags.ignore_mandatory:
  646. return
  647. missing = self._get_missing_mandatory_fields()
  648. for d in self.get_all_children():
  649. missing.extend(d._get_missing_mandatory_fields())
  650. if not missing:
  651. return
  652. for fieldname, msg in missing:
  653. msgprint(msg)
  654. if frappe.flags.print_messages:
  655. print(self.as_json().encode("utf-8"))
  656. raise frappe.MandatoryError('[{doctype}, {name}]: {fields}'.format(
  657. fields=", ".join((each[0] for each in missing)),
  658. doctype=self.doctype,
  659. name=self.name))
  660. def _validate_links(self):
  661. if self.flags.ignore_links or self._action == "cancel":
  662. return
  663. invalid_links, cancelled_links = self.get_invalid_links()
  664. for d in self.get_all_children():
  665. result = d.get_invalid_links(is_submittable=self.meta.is_submittable)
  666. invalid_links.extend(result[0])
  667. cancelled_links.extend(result[1])
  668. if invalid_links:
  669. msg = ", ".join((each[2] for each in invalid_links))
  670. frappe.throw(_("Could not find {0}").format(msg),
  671. frappe.LinkValidationError)
  672. if cancelled_links:
  673. msg = ", ".join((each[2] for each in cancelled_links))
  674. frappe.throw(_("Cannot link cancelled document: {0}").format(msg),
  675. frappe.CancelledLinkError)
  676. def get_all_children(self, parenttype=None):
  677. """Returns all children documents from **Table** type field in a list."""
  678. ret = []
  679. for df in self.meta.get("fields", {"fieldtype": ['in', table_fields]}):
  680. if parenttype:
  681. if df.options==parenttype:
  682. return self.get(df.fieldname)
  683. value = self.get(df.fieldname)
  684. if isinstance(value, list):
  685. ret.extend(value)
  686. return ret
  687. def run_method(self, method, *args, **kwargs):
  688. """run standard triggers, plus those in hooks"""
  689. if "flags" in kwargs:
  690. del kwargs["flags"]
  691. if hasattr(self, method) and hasattr(getattr(self, method), "__call__"):
  692. fn = lambda self, *args, **kwargs: getattr(self, method)(*args, **kwargs)
  693. else:
  694. # hack! to run hooks even if method does not exist
  695. fn = lambda self, *args, **kwargs: None
  696. fn.__name__ = str(method)
  697. out = Document.hook(fn)(self, *args, **kwargs)
  698. self.run_notifications(method)
  699. run_webhooks(self, method)
  700. run_server_script_for_doc_event(self, method)
  701. return out
  702. def run_trigger(self, method, *args, **kwargs):
  703. return self.run_method(method, *args, **kwargs)
  704. def run_notifications(self, method):
  705. """Run notifications for this method"""
  706. if (frappe.flags.in_import and frappe.flags.mute_emails) or frappe.flags.in_patch or frappe.flags.in_install:
  707. return
  708. if self.flags.notifications_executed is None:
  709. self.flags.notifications_executed = []
  710. from frappe.email.doctype.notification.notification import evaluate_alert
  711. if self.flags.notifications is None:
  712. alerts = frappe.cache().hget('notifications', self.doctype)
  713. if alerts is None:
  714. alerts = frappe.get_all('Notification', fields=['name', 'event', 'method'],
  715. filters={'enabled': 1, 'document_type': self.doctype})
  716. frappe.cache().hset('notifications', self.doctype, alerts)
  717. self.flags.notifications = alerts
  718. if not self.flags.notifications:
  719. return
  720. def _evaluate_alert(alert):
  721. if not alert.name in self.flags.notifications_executed:
  722. evaluate_alert(self, alert.name, alert.event)
  723. self.flags.notifications_executed.append(alert.name)
  724. event_map = {
  725. "on_update": "Save",
  726. "after_insert": "New",
  727. "on_submit": "Submit",
  728. "on_cancel": "Cancel"
  729. }
  730. if not self.flags.in_insert:
  731. # value change is not applicable in insert
  732. event_map['on_change'] = 'Value Change'
  733. for alert in self.flags.notifications:
  734. event = event_map.get(method, None)
  735. if event and alert.event == event:
  736. _evaluate_alert(alert)
  737. elif alert.event=='Method' and method == alert.method:
  738. _evaluate_alert(alert)
  739. @whitelist.__func__
  740. def _submit(self):
  741. """Submit the document. Sets `docstatus` = 1, then saves."""
  742. self.docstatus = DocStatus.submitted()
  743. return self.save()
  744. @whitelist.__func__
  745. def _cancel(self):
  746. """Cancel the document. Sets `docstatus` = 2, then saves.
  747. """
  748. self.docstatus = DocStatus.cancelled()
  749. return self.save()
  750. @whitelist.__func__
  751. def submit(self):
  752. """Submit the document. Sets `docstatus` = 1, then saves."""
  753. return self._submit()
  754. @whitelist.__func__
  755. def cancel(self):
  756. """Cancel the document. Sets `docstatus` = 2, then saves."""
  757. return self._cancel()
  758. def delete(self, ignore_permissions=False):
  759. """Delete document."""
  760. frappe.delete_doc(self.doctype, self.name, ignore_permissions = ignore_permissions, flags=self.flags)
  761. def run_before_save_methods(self):
  762. """Run standard methods before `INSERT` or `UPDATE`. Standard Methods are:
  763. - `validate`, `before_save` for **Save**.
  764. - `validate`, `before_submit` for **Submit**.
  765. - `before_cancel` for **Cancel**
  766. - `before_update_after_submit` for **Update after Submit**
  767. Will also update title_field if set"""
  768. self.load_doc_before_save()
  769. self.reset_seen()
  770. # before_validate method should be executed before ignoring validations
  771. if self._action in ("save", "submit"):
  772. self.run_method("before_validate")
  773. if self.flags.ignore_validate:
  774. return
  775. if self._action=="save":
  776. self.run_method("validate")
  777. self.run_method("before_save")
  778. elif self._action=="submit":
  779. self.run_method("validate")
  780. self.run_method("before_submit")
  781. elif self._action=="cancel":
  782. self.run_method("before_cancel")
  783. elif self._action=="update_after_submit":
  784. self.run_method("before_update_after_submit")
  785. self.set_title_field()
  786. def load_doc_before_save(self):
  787. """Save load document from db before saving"""
  788. self._doc_before_save = None
  789. if not self.is_new():
  790. try:
  791. self._doc_before_save = frappe.get_doc(self.doctype, self.name)
  792. except frappe.DoesNotExistError:
  793. self._doc_before_save = None
  794. frappe.clear_last_message()
  795. def run_post_save_methods(self):
  796. """Run standard methods after `INSERT` or `UPDATE`. Standard Methods are:
  797. - `on_update` for **Save**.
  798. - `on_update`, `on_submit` for **Submit**.
  799. - `on_cancel` for **Cancel**
  800. - `update_after_submit` for **Update after Submit**"""
  801. doc_before_save = self.get_doc_before_save()
  802. if self._action=="save":
  803. self.run_method("on_update")
  804. elif self._action=="submit":
  805. self.run_method("on_update")
  806. self.run_method("on_submit")
  807. elif self._action=="cancel":
  808. self.run_method("on_cancel")
  809. self.check_no_back_links_exist()
  810. elif self._action=="update_after_submit":
  811. self.run_method("on_update_after_submit")
  812. self.clear_cache()
  813. self.notify_update()
  814. update_global_search(self)
  815. self.save_version()
  816. self.run_method('on_change')
  817. if (self.doctype, self.name) in frappe.flags.currently_saving:
  818. frappe.flags.currently_saving.remove((self.doctype, self.name))
  819. self.latest = None
  820. def clear_cache(self):
  821. frappe.clear_document_cache(self.doctype, self.name)
  822. def reset_seen(self):
  823. """Clear _seen property and set current user as seen"""
  824. if getattr(self.meta, 'track_seen', False):
  825. frappe.db.set_value(self.doctype, self.name, "_seen", json.dumps([frappe.session.user]), update_modified=False)
  826. def notify_update(self):
  827. """Publish realtime that the current document is modified"""
  828. if frappe.flags.in_patch: return
  829. frappe.publish_realtime("doc_update", {"modified": self.modified, "doctype": self.doctype, "name": self.name},
  830. doctype=self.doctype, docname=self.name, after_commit=True)
  831. if not self.meta.get("read_only") and not self.meta.get("issingle") and \
  832. not self.meta.get("istable"):
  833. data = {
  834. "doctype": self.doctype,
  835. "name": self.name,
  836. "user": frappe.session.user
  837. }
  838. frappe.publish_realtime("list_update", data, after_commit=True)
  839. def db_set(self, fieldname, value=None, update_modified=True, notify=False, commit=False):
  840. """Set a value in the document object, update the timestamp and update the database.
  841. WARNING: This method does not trigger controller validations and should
  842. be used very carefully.
  843. :param fieldname: fieldname of the property to be updated, or a {"field":"value"} dictionary
  844. :param value: value of the property to be updated
  845. :param update_modified: default True. updates the `modified` and `modified_by` properties
  846. :param notify: default False. run doc.notify_updated() to send updates via socketio
  847. :param commit: default False. run frappe.db.commit()
  848. """
  849. if isinstance(fieldname, dict):
  850. self.update(fieldname)
  851. else:
  852. self.set(fieldname, value)
  853. if update_modified and (self.doctype, self.name) not in frappe.flags.currently_saving:
  854. # don't update modified timestamp if called from post save methods
  855. # like on_update or on_submit
  856. self.set("modified", now())
  857. self.set("modified_by", frappe.session.user)
  858. # load but do not reload doc_before_save because before_change or on_change might expect it
  859. if not self.get_doc_before_save():
  860. self.load_doc_before_save()
  861. # to trigger notification on value change
  862. self.run_method('before_change')
  863. frappe.db.set_value(self.doctype, self.name, fieldname, value,
  864. self.modified, self.modified_by, update_modified=update_modified)
  865. self.run_method('on_change')
  866. if notify:
  867. self.notify_update()
  868. self.clear_cache()
  869. if commit:
  870. frappe.db.commit()
  871. def db_get(self, fieldname):
  872. """get database value for this fieldname"""
  873. return frappe.db.get_value(self.doctype, self.name, fieldname)
  874. def check_no_back_links_exist(self):
  875. """Check if document links to any active document before Cancel."""
  876. from frappe.model.delete_doc import check_if_doc_is_linked, check_if_doc_is_dynamically_linked
  877. if not self.flags.ignore_links:
  878. check_if_doc_is_linked(self, method="Cancel")
  879. check_if_doc_is_dynamically_linked(self, method="Cancel")
  880. def save_version(self):
  881. """Save version info"""
  882. # don't track version under following conditions
  883. if (not getattr(self.meta, 'track_changes', False)
  884. or self.doctype == 'Version'
  885. or self.flags.ignore_version
  886. or frappe.flags.in_install
  887. or (not self._doc_before_save and frappe.flags.in_patch)):
  888. return
  889. version = frappe.new_doc('Version')
  890. if not self._doc_before_save:
  891. version.for_insert(self)
  892. version.insert(ignore_permissions=True)
  893. elif version.set_diff(self._doc_before_save, self):
  894. version.insert(ignore_permissions=True)
  895. if not frappe.flags.in_migrate:
  896. # follow since you made a change?
  897. follow_document(self.doctype, self.name, frappe.session.user)
  898. @staticmethod
  899. def hook(f):
  900. """Decorator: Make method `hookable` (i.e. extensible by another app).
  901. Note: If each hooked method returns a value (dict), then all returns are
  902. collated in one dict and returned. Ideally, don't return values in hookable
  903. methods, set properties in the document."""
  904. def add_to_return_value(self, new_return_value):
  905. if new_return_value is None:
  906. self._return_value = self.get("_return_value")
  907. return
  908. if isinstance(new_return_value, dict):
  909. if not self.get("_return_value"):
  910. self._return_value = {}
  911. self._return_value.update(new_return_value)
  912. else:
  913. self._return_value = new_return_value
  914. def compose(fn, *hooks):
  915. def runner(self, method, *args, **kwargs):
  916. add_to_return_value(self, fn(self, *args, **kwargs))
  917. for f in hooks:
  918. add_to_return_value(self, f(self, method, *args, **kwargs))
  919. return self._return_value
  920. return runner
  921. def composer(self, *args, **kwargs):
  922. hooks = []
  923. method = f.__name__
  924. doc_events = frappe.get_doc_hooks()
  925. for handler in doc_events.get(self.doctype, {}).get(method, []) \
  926. + doc_events.get("*", {}).get(method, []):
  927. hooks.append(frappe.get_attr(handler))
  928. composed = compose(f, *hooks)
  929. return composed(self, method, *args, **kwargs)
  930. return composer
  931. def is_whitelisted(self, method_name):
  932. method = getattr(self, method_name, None)
  933. if not method:
  934. raise NotFound("Method {0} not found".format(method_name))
  935. is_whitelisted(getattr(method, '__func__', method))
  936. def validate_value(self, fieldname, condition, val2, doc=None, raise_exception=None):
  937. """Check that value of fieldname should be 'condition' val2
  938. else throw Exception."""
  939. error_condition_map = {
  940. "in": _("one of"),
  941. "not in": _("none of"),
  942. "^": _("beginning with"),
  943. }
  944. if not doc:
  945. doc = self
  946. val1 = doc.get_value(fieldname)
  947. df = doc.meta.get_field(fieldname)
  948. val2 = doc.cast(val2, df)
  949. if not frappe.compare(val1, condition, val2):
  950. label = doc.meta.get_label(fieldname)
  951. condition_str = error_condition_map.get(condition, condition)
  952. if doc.get("parentfield"):
  953. msg = _("Incorrect value in row {0}: {1} must be {2} {3}").format(doc.idx, label, condition_str, val2)
  954. else:
  955. msg = _("Incorrect value: {0} must be {1} {2}").format(label, condition_str, val2)
  956. # raise passed exception or True
  957. msgprint(msg, raise_exception=raise_exception or True)
  958. def validate_table_has_rows(self, parentfield, raise_exception=None):
  959. """Raise exception if Table field is empty."""
  960. if not (isinstance(self.get(parentfield), list) and len(self.get(parentfield)) > 0):
  961. label = self.meta.get_label(parentfield)
  962. frappe.throw(_("Table {0} cannot be empty").format(label), raise_exception or frappe.EmptyTableError)
  963. def round_floats_in(self, doc, fieldnames=None):
  964. """Round floats for all `Currency`, `Float`, `Percent` fields for the given doc.
  965. :param doc: Document whose numeric properties are to be rounded.
  966. :param fieldnames: [Optional] List of fields to be rounded."""
  967. if not fieldnames:
  968. fieldnames = (df.fieldname for df in
  969. doc.meta.get("fields", {"fieldtype": ["in", ["Currency", "Float", "Percent"]]}))
  970. for fieldname in fieldnames:
  971. doc.set(fieldname, flt(doc.get(fieldname), self.precision(fieldname, doc.get("parentfield"))))
  972. def get_url(self):
  973. """Returns Desk URL for this document."""
  974. return get_absolute_url(self.doctype, self.name)
  975. def add_comment(self, comment_type='Comment', text=None, comment_email=None, link_doctype=None, link_name=None, comment_by=None):
  976. """Add a comment to this document.
  977. :param comment_type: e.g. `Comment`. See Communication for more info."""
  978. out = frappe.get_doc({
  979. "doctype":"Comment",
  980. 'comment_type': comment_type,
  981. "comment_email": comment_email or frappe.session.user,
  982. "comment_by": comment_by,
  983. "reference_doctype": self.doctype,
  984. "reference_name": self.name,
  985. "content": text or comment_type,
  986. "link_doctype": link_doctype,
  987. "link_name": link_name
  988. }).insert(ignore_permissions=True)
  989. return out
  990. def add_seen(self, user=None):
  991. """add the given/current user to list of users who have seen this document (_seen)"""
  992. if not user:
  993. user = frappe.session.user
  994. if self.meta.track_seen:
  995. _seen = self.get('_seen') or []
  996. _seen = frappe.parse_json(_seen)
  997. if user not in _seen:
  998. _seen.append(user)
  999. frappe.db.set_value(self.doctype, self.name, '_seen', json.dumps(_seen), update_modified=False)
  1000. frappe.local.flags.commit = True
  1001. def add_viewed(self, user=None):
  1002. """add log to communication when a user views a document"""
  1003. if not user:
  1004. user = frappe.session.user
  1005. if hasattr(self.meta, 'track_views') and self.meta.track_views:
  1006. frappe.get_doc({
  1007. "doctype": "View Log",
  1008. "viewed_by": frappe.session.user,
  1009. "reference_doctype": self.doctype,
  1010. "reference_name": self.name,
  1011. }).insert(ignore_permissions=True)
  1012. frappe.local.flags.commit = True
  1013. def get_signature(self):
  1014. """Returns signature (hash) for private URL."""
  1015. return hashlib.sha224(get_datetime_str(self.creation).encode()).hexdigest()
  1016. def get_liked_by(self):
  1017. liked_by = getattr(self, "_liked_by", None)
  1018. if liked_by:
  1019. return json.loads(liked_by)
  1020. else:
  1021. return []
  1022. def set_onload(self, key, value):
  1023. if not self.get("__onload"):
  1024. self.set("__onload", frappe._dict())
  1025. self.get("__onload")[key] = value
  1026. def get_onload(self, key=None):
  1027. if not key:
  1028. return self.get("__onload", frappe._dict())
  1029. return self.get('__onload')[key]
  1030. def queue_action(self, action, **kwargs):
  1031. """Run an action in background. If the action has an inner function,
  1032. like _submit for submit, it will call that instead"""
  1033. # call _submit instead of submit, so you can override submit to call
  1034. # run_delayed based on some action
  1035. # See: Stock Reconciliation
  1036. from frappe.utils.background_jobs import enqueue
  1037. if hasattr(self, '_' + action):
  1038. action = '_' + action
  1039. if file_lock.lock_exists(self.get_signature()):
  1040. frappe.throw(_('This document is currently queued for execution. Please try again'),
  1041. title=_('Document Queued'))
  1042. self.lock()
  1043. enqueue('frappe.model.document.execute_action', doctype=self.doctype, name=self.name,
  1044. action=action, **kwargs)
  1045. def lock(self, timeout=None):
  1046. """Creates a lock file for the given document. If timeout is set,
  1047. it will retry every 1 second for acquiring the lock again
  1048. :param timeout: Timeout in seconds, default 0"""
  1049. signature = self.get_signature()
  1050. if file_lock.lock_exists(signature):
  1051. lock_exists = True
  1052. if timeout:
  1053. for i in range(timeout):
  1054. time.sleep(1)
  1055. if not file_lock.lock_exists(signature):
  1056. lock_exists = False
  1057. break
  1058. if lock_exists:
  1059. raise frappe.DocumentLockedError
  1060. file_lock.create_lock(signature)
  1061. def unlock(self):
  1062. """Delete the lock file for this document"""
  1063. file_lock.delete_lock(self.get_signature())
  1064. # validation helpers
  1065. def validate_from_to_dates(self, from_date_field, to_date_field):
  1066. """
  1067. Generic validation to verify date sequence
  1068. """
  1069. if date_diff(self.get(to_date_field), self.get(from_date_field)) < 0:
  1070. frappe.throw(_('{0} must be after {1}').format(
  1071. frappe.bold(self.meta.get_label(to_date_field)),
  1072. frappe.bold(self.meta.get_label(from_date_field)),
  1073. ), frappe.exceptions.InvalidDates)
  1074. def get_assigned_users(self):
  1075. assigned_users = frappe.get_all('ToDo',
  1076. fields=['allocated_to'],
  1077. filters={
  1078. 'reference_type': self.doctype,
  1079. 'reference_name': self.name,
  1080. 'status': ('!=', 'Cancelled'),
  1081. }, pluck='allocated_to')
  1082. users = set(assigned_users)
  1083. return users
  1084. def add_tag(self, tag):
  1085. """Add a Tag to this document"""
  1086. from frappe.desk.doctype.tag.tag import DocTags
  1087. DocTags(self.doctype).add(self.name, tag)
  1088. def get_tags(self):
  1089. """Return a list of Tags attached to this document"""
  1090. from frappe.desk.doctype.tag.tag import DocTags
  1091. return DocTags(self.doctype).get_tags(self.name).split(",")[1:]
  1092. def __repr__(self):
  1093. name = self.name or "unsaved"
  1094. doctype = self.__class__.__name__
  1095. docstatus = f" docstatus={self.docstatus}" if self.docstatus else ""
  1096. repr_str = f"<{doctype}: {name}{docstatus}"
  1097. if not hasattr(self, "parent"):
  1098. return repr_str + ">"
  1099. return f"{repr_str} parent={self.parent}>"
  1100. def __str__(self):
  1101. name = self.name or "unsaved"
  1102. doctype = self.__class__.__name__
  1103. return f"{doctype}({name})"
  1104. def execute_action(doctype, name, action, **kwargs):
  1105. """Execute an action on a document (called by background worker)"""
  1106. doc = frappe.get_doc(doctype, name)
  1107. doc.unlock()
  1108. try:
  1109. getattr(doc, action)(**kwargs)
  1110. except Exception:
  1111. frappe.db.rollback()
  1112. # add a comment (?)
  1113. if frappe.local.message_log:
  1114. msg = json.loads(frappe.local.message_log[-1]).get('message')
  1115. else:
  1116. msg = '<pre><code>' + frappe.get_traceback() + '</pre></code>'
  1117. doc.add_comment('Comment', _('Action Failed') + '<br><br>' + msg)
  1118. doc.notify_update()