|
|
# -*- coding: utf-8 -*-
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
# (at your option) any later version.
|
|
|
#
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
# GNU General Public License for more details.
|
|
|
#
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
"""
|
|
|
SQLAlchemy Metadata and Session object
|
|
|
"""
|
|
|
from beaker import cache
|
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
|
from sqlalchemy.orm import scoped_session, sessionmaker
|
|
|
|
|
|
from kallithea.lib import caching_query
|
|
|
|
|
|
|
|
|
# Beaker CacheManager. A home base for cache configurations.
|
|
|
cache_manager = cache.CacheManager()
|
|
|
|
|
|
__all__ = ['Base', 'Session']
|
|
|
|
|
|
#
|
|
|
# SQLAlchemy session manager.
|
|
|
#
|
|
|
session_factory = sessionmaker(
|
|
|
query_cls=caching_query.query_callable(cache_manager),
|
|
|
expire_on_commit=True)
|
|
|
Session = scoped_session(session_factory)
|
|
|
|
|
|
# The base class for declarative schemas in db.py
|
|
|
# Engine is injected when model.__init__.init_model() sets meta.Base.metadata.bind
|
|
|
Base = declarative_base()
|
|
|
|
|
|
# to use cache use this in query:
|
|
|
# .options(FromCache("sqlalchemy_cache_type", "cachekey"))
|
|
|
|
|
|
|
|
|
# Define naming conventions for foreign keys, primary keys, indexes,
|
|
|
# check constraints, and unique constraints, respectively.
|
|
|
Base.metadata.naming_convention = {
|
|
|
'fk': 'fk_%(table_name)s_%(column_0_name)s',
|
|
|
'pk': 'pk_%(table_name)s',
|
|
|
'ix': 'ix_%(column_0_label)s',
|
|
|
'ck': 'ck_%(table_name)s_%(column_0_name)s',
|
|
|
'uq': 'uq_%(table_name)s_%(column_0_name)s',
|
|
|
}
|
|
|
# For custom CheckConstraints (not those autogenerated e.g. for Boolean
|
|
|
# types), a name should be given explicitly, since "column_0" is here a
|
|
|
# rather vague notion. A custom name is also necesarry if the generated
|
|
|
# name is very long, since MySQL limits identifiers to 64 characters.
|
|
|
|