31 lines
878 B
Python
31 lines
878 B
Python
# Copyright (c) 2013, Web Notes Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
from __future__ import unicode_literals
|
|
|
|
import memcache, frappe
|
|
|
|
class MClient(memcache.Client):
|
|
"""memcache client that will automatically prefix conf.db_name"""
|
|
def n(self, key):
|
|
return (frappe.conf.db_name + ":" + key.replace(" ", "_")).encode('utf-8')
|
|
|
|
def set_value(self, key, val):
|
|
frappe.local.cache[key] = val
|
|
self.set(self.n(key), val)
|
|
|
|
def get_value(self, key, builder=None):
|
|
val = frappe.local.cache.get(key)
|
|
if not val:
|
|
val = self.get(self.n(key))
|
|
if not val and builder:
|
|
val = builder()
|
|
self.set_value(key, val)
|
|
return val
|
|
|
|
def delete_value(self, keys):
|
|
if not isinstance(keys, (list, tuple)):
|
|
keys = (keys,)
|
|
for key in keys:
|
|
self.delete(self.n(key))
|
|
if key in frappe.local.cache:
|
|
del frappe.local.cache[key]
|