cqle: always set default keyspace from conn.set_session

PYTHON-486
This commit is contained in:
Adam Holmberg
2016-03-02 14:10:21 -06:00
parent d3cdb168e3
commit bcd42baf31

View File

@@ -81,9 +81,9 @@ def set_session(s):
session = s
cluster = s.cluster
# Set default keyspace from given session's keyspace if not already set
from cassandra.cqlengine import models
if not models.DEFAULT_KEYSPACE and session.keyspace:
# Set default keyspace from given session's keyspace
if session.keyspace:
from cassandra.cqlengine import models
models.DEFAULT_KEYSPACE = session.keyspace
_setup_session(session)