rhodecode/model/meta.py

"""SQLAlchemy Metadata and Session object"""
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import scoped_session, sessionmaker
from rhodecode.model import caching_query
from beaker import cache
import os
from os.path import join as jn, dirname as dn, abspath
import time

# Beaker CacheManager.  A home base for cache configurations.
cache_manager = cache.CacheManager()

__all__ = ['Base', 'Session']
#
# SQLAlchemy session manager. Updated by model.init_model()
#
Session = scoped_session(
                sessionmaker(
                    query_cls=caching_query.query_callable(cache_manager)
                )
          )

# The declarative Base
Base = declarative_base()
#For another db...
#Base2 = declarative_base()

#===============================================================================
# CACHE OPTIONS
#===============================================================================
cache_base = jn(dn(dn(dn(abspath(__file__)))), 'data')
cache_dir = jn(cache_base, 'cache')

if not os.path.isdir(cache_base):
    os.mkdir(cache_base)

if not os.path.isdir(cache_dir):
    os.mkdir(cache_dir)
# set start_time to current time
# to re-cache everything
# upon application startup
start_time = time.time()
# configure the "sqlalchemy" cache region.
cache_manager.regions['sql_cache_short'] = {
        'type':'memory',
        'data_dir':cache_dir,
        'expire':10,
        'start_time':start_time
    }
cache_manager.regions['sql_cache_med'] = {
        'type':'memory',
        'data_dir':cache_dir,
        'expire':360,
        'start_time':start_time
    }
cache_manager.regions['sql_cache_long'] = {
        'type':'file',
        'data_dir':cache_dir,
        'expire':3600,
        'start_time':start_time
    }
#to use cache use this in query
#.options(FromCache("sqlalchemy_cache_type", "cachekey"))