117 lines
3.2 KiB
Python
117 lines
3.2 KiB
Python
# Copyright (c) 2018, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import frappe, json
|
|
import frappe.defaults
|
|
from frappe.desk.notifications import (delete_notification_count_for,
|
|
clear_notifications)
|
|
|
|
common_default_keys = ["__default", "__global"]
|
|
|
|
global_cache_keys = ("app_hooks", "installed_apps",
|
|
"app_modules", "module_app", "system_settings",
|
|
'scheduler_events', 'time_zone', 'webhooks', 'active_domains',
|
|
'active_modules', 'assignment_rule')
|
|
|
|
user_cache_keys = ("bootinfo", "user_recent", "roles", "user_doc", "lang",
|
|
"defaults", "user_permissions", "home_page", "linked_with",
|
|
"desktop_icons", 'portal_menu_items')
|
|
|
|
doctype_cache_keys = ("meta", "form_meta", "table_columns", "last_modified",
|
|
"linked_doctypes", 'notifications', 'workflow' ,'energy_point_rule_map')
|
|
|
|
|
|
def clear_user_cache(user=None):
|
|
cache = frappe.cache()
|
|
|
|
# this will automatically reload the global cache
|
|
# so it is important to clear this first
|
|
clear_notifications(user)
|
|
|
|
if user:
|
|
for name in user_cache_keys:
|
|
cache.hdel(name, user)
|
|
cache.delete_keys("user:" + user)
|
|
clear_defaults_cache(user)
|
|
else:
|
|
for name in user_cache_keys:
|
|
cache.delete_key(name)
|
|
clear_defaults_cache()
|
|
clear_global_cache()
|
|
|
|
def clear_global_cache():
|
|
from frappe.website.render import clear_cache as clear_website_cache
|
|
|
|
clear_doctype_cache()
|
|
clear_website_cache()
|
|
frappe.cache().delete_value(global_cache_keys)
|
|
frappe.setup_module_map()
|
|
|
|
def clear_defaults_cache(user=None):
|
|
if user:
|
|
for p in ([user] + common_default_keys):
|
|
frappe.cache().hdel("defaults", p)
|
|
elif frappe.flags.in_install!="frappe":
|
|
frappe.cache().delete_key("defaults")
|
|
|
|
def clear_document_cache():
|
|
frappe.local.document_cache = {}
|
|
frappe.cache().delete_key("document_cache")
|
|
|
|
def clear_doctype_cache(doctype=None):
|
|
cache = frappe.cache()
|
|
|
|
if getattr(frappe.local, 'meta_cache') and (doctype in frappe.local.meta_cache):
|
|
del frappe.local.meta_cache[doctype]
|
|
|
|
for key in ('is_table', 'doctype_modules'):
|
|
cache.delete_value(key)
|
|
|
|
def clear_single(dt):
|
|
for name in doctype_cache_keys:
|
|
cache.hdel(name, dt)
|
|
|
|
if doctype:
|
|
clear_single(doctype)
|
|
|
|
# clear all parent doctypes
|
|
|
|
for dt in frappe.db.get_all('DocField', 'parent',
|
|
dict(fieldtype=['in', frappe.model.table_fields], options=doctype)):
|
|
clear_single(dt.parent)
|
|
|
|
# clear all notifications
|
|
delete_notification_count_for(doctype)
|
|
|
|
else:
|
|
# clear all
|
|
for name in doctype_cache_keys:
|
|
cache.delete_value(name)
|
|
|
|
# Clear all document's cache. To clear documents of a specific DocType document_cache should be restructured
|
|
clear_document_cache()
|
|
|
|
def get_doctype_map(doctype, name, filters, order_by=None):
|
|
cache = frappe.cache()
|
|
cache_key = frappe.scrub(doctype) + '_map'
|
|
doctype_map = cache.hget(cache_key, name)
|
|
|
|
if doctype_map:
|
|
# cached, return
|
|
items = json.loads(doctype_map)
|
|
else:
|
|
# non cached, build cache
|
|
try:
|
|
items = frappe.get_all(doctype, filters=filters, order_by = order_by)
|
|
cache.hset(cache_key, doctype, json.dumps(items))
|
|
except frappe.db.TableMissingError:
|
|
# executed from inside patch, ignore
|
|
items = []
|
|
|
|
return items
|
|
|
|
def clear_doctype_map(doctype, name):
|
|
cache_key = frappe.scrub(doctype) + '_map'
|
|
frappe.cache().hdel(cache_key, name)
|