e407b78506
* Remove six for PY2 compatability since our dependencies are not, PY2 is legacy. * Removed usages of utils from future/past libraries since they are deprecated. This includes 'from __future__ ...' and 'from past...' statements. * Removed compatibility imports for PY2, switched from six imports to standard library imports. * Removed utils code blocks that handle operations depending on PY2/3 versions. * Removed 'from __future__ ...' lines from templates/code generators * Used PY3 syntaxes in place of PY2 compatible blocks. eg: metaclass
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
import frappe
|
|
|
|
ignore_doctypes = ("DocType", "Print Format", "Role", "Module Def", "Communication",
|
|
"ToDo")
|
|
|
|
def notify_link_count(doctype, name):
|
|
'''updates link count for given document'''
|
|
if hasattr(frappe.local, 'link_count'):
|
|
if (doctype, name) in frappe.local.link_count:
|
|
frappe.local.link_count[(doctype, name)] += 1
|
|
else:
|
|
frappe.local.link_count[(doctype, name)] = 1
|
|
|
|
def flush_local_link_count():
|
|
'''flush from local before ending request'''
|
|
if not getattr(frappe.local, 'link_count', None):
|
|
return
|
|
|
|
link_count = frappe.cache().get_value('_link_count')
|
|
if not link_count:
|
|
link_count = {}
|
|
|
|
for key, value in frappe.local.link_count.items():
|
|
if key in link_count:
|
|
link_count[key] += frappe.local.link_count[key]
|
|
else:
|
|
link_count[key] = frappe.local.link_count[key]
|
|
|
|
frappe.cache().set_value('_link_count', link_count)
|
|
|
|
def update_link_count():
|
|
'''increment link count in the `idx` column for the given document'''
|
|
link_count = frappe.cache().get_value('_link_count')
|
|
|
|
if link_count:
|
|
for key, count in link_count.items():
|
|
if key[0] not in ignore_doctypes:
|
|
try:
|
|
frappe.db.sql('update `tab{0}` set idx = idx + {1} where name=%s'.format(key[0], count),
|
|
key[1], auto_commit=1)
|
|
except Exception as e:
|
|
if not frappe.db.is_table_missing(e): # table not found, single
|
|
raise e
|
|
# reset the count
|
|
frappe.cache().delete_value('_link_count')
|