Additional methods for setting up mapper session.

default - use localhost connection
set_session - use a session bootstrapped in a different context
This commit is contained in:
Adam Holmberg
2015-02-10 13:57:36 -06:00
parent b2a5cb9348
commit 46ba8061fa
2 changed files with 23 additions and 2 deletions

View File

@@ -36,7 +36,25 @@ Host = namedtuple('Host', ['name', 'port'])
cluster = None
session = None
lazy_connect_args = None
default_consistency_level = None
default_consistency_level = ConsistencyLevel.ONE
def default():
"""
Configures the global mapper connection to localhost, using the driver defaults
"""
global cluster, session
cluster = Cluster()
session = cluster.connect()
def set_session(s):
"""
Configures the global mapper connection with a preexisting :class:`cassandra.cluster.Session`
"""
global cluster, session
session = s
cluster = s.cluster
def setup(

View File

@@ -3,5 +3,8 @@
.. module:: cassandra.cqlengine.connection
.. autofunction:: setup
.. autofunction:: default
.. autofunction:: set_session
.. autofunction:: setup