seitime-frappe/py/webnotes/session_cache.py
2012-02-15 14:02:26 +05:30

77 lines
2 KiB
Python

"""
Boot session from cache or build
Session bootstraps info needed by common client side activities including
permission, homepage, control panel variables, system defaults etc
"""
import webnotes
@webnotes.whitelist()
def clear():
"""clear all cache"""
clear_cache()
webnotes.response['message'] = "Cache Cleared"
def clear_cache(user=''):
"""clear cache"""
if user:
webnotes.conn.sql("delete from __SessionCache where user=%s", user)
webnotes.conn.sql("update tabSessions set sessiondata=NULL where user=%s", user)
else:
webnotes.conn.sql("delete from __SessionCache")
webnotes.conn.sql("update tabSessions set sessiondata=NULL")
# rebuild a cache for guest
if webnotes.session:
webnotes.session['data'] = {}
def get():
"""get session boot info"""
import webnotes.defs
# get country
country = webnotes.session['data'].get('ipinfo', {}).get('countryName', 'Unknown Country')
# check if cache exists
if not getattr(webnotes.defs,'auto_cache_clear',None):
cache = load(country)
if cache:
return cache
# if not create it
import webnotes.boot
bootinfo = webnotes.boot.get_bootinfo()
add_to_cache(bootinfo, country)
return bootinfo
def load(country):
"""load from cache"""
try:
sd = webnotes.conn.sql("select cache from __SessionCache where user='%s' %s" % (webnotes.session['user'], (country and (" and country='%s'" % country) or '')))
if sd:
return eval(sd[0][0])
else:
return None
except Exception, e:
if e.args[0]==1146:
make_cache_table()
else:
raise e
def add_to_cache(bootinfo, country):
"""add to cache"""
import webnotes.model.utils
if bootinfo.get('docs'):
bootinfo['docs'] = webnotes.model.utils.compress(bootinfo['docs'])
# delete earlier (?)
webnotes.conn.sql("""delete from __SessionCache where user=%s
and country=%s""", (webnotes.session['user'], country))
# make new
webnotes.conn.sql("""insert into `__SessionCache`
(user, country, cache) VALUES (%s, %s, %s)""", \
(webnotes.session['user'], country, str(bootinfo)))