relstorage/relstorage/config.py

67 lines
2.3 KiB
Python

##############################################################################
#
# Copyright (c) 2008 Zope Foundation and Contributors.
# All Rights Reserved.
#
# This software is subject to the provisions of the Zope Public License,
# Version 2.1 (ZPL). A copy of the ZPL should accompany this distribution.
# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
# FOR A PARTICULAR PURPOSE.
#
##############################################################################
"""ZConfig directive implementations for binding RelStorage to Zope"""
from __future__ import absolute_import
from relstorage.options import Options
from relstorage.storage import RelStorage
class BaseConfig(object):
def __init__(self, config):
self.config = config
self.name = config.getSectionName()
class RelStorageFactory(BaseConfig):
"""Open a storage configured via ZConfig"""
def open(self):
config = self.config
# Hoist the driver setting to the section we really want it.
config.driver = config.adapter.config.driver
config.adapter.config.driver = None
options = Options.copy_valid_options(config)
adapter = config.adapter.create(options)
return RelStorage(adapter, name=config.name, options=options)
class PostgreSQLAdapterFactory(BaseConfig):
def create(self, options):
from .adapters.postgresql import PostgreSQLAdapter
return PostgreSQLAdapter(
dsn=self.config.dsn,
options=options)
class OracleAdapterFactory(BaseConfig):
def create(self, options):
from .adapters.oracle import OracleAdapter
config = self.config
return OracleAdapter(
user=config.user,
password=config.password,
dsn=config.dsn,
options=options)
class MySQLAdapterFactory(BaseConfig):
def create(self, options):
from .adapters.mysql import MySQLAdapter
params = {}
for key in self.config.getSectionAttributes():
value = getattr(self.config, key)
if value is not None:
params[key] = value
return MySQLAdapter(options=options, **params)