saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-16 01:30:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Moving setting the LOAD_BALANCING_POLICY_MAP dictionary into the try except block that determines if the cassandra_cql module should be made available.

Browse source
This commit is contained in:
Gareth J. Greenaway 2022-10-13 16:02:15 -07:00
parent 1404abe773
commit a3e56f4d60
No known key found for this signature in database
GPG key ID: 10B62F8A7CAD7A41
2 changed files with 18 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
changelog/62886.fixed Normal file
View file

@ -0,0 +1 @@
Moving setting the LOAD_BALANCING_POLICY_MAP dictionary into the try except block that determines if the cassandra_cql module should be made available.

34
salt/modules/cassandra_cql.py
View file

@ -136,27 +136,27 @@ try:
# pylint: enable=import-error,no-name-in-module
HAS_DRIVER = True
LOAD_BALANCING_POLICY_MAP = {
"HostDistance": HostDistance,
"LoadBalancingPolicy": LoadBalancingPolicy,
"RoundRobinPolicy": RoundRobinPolicy,
"DCAwareRoundRobinPolicy": DCAwareRoundRobinPolicy,
"WhiteListRoundRobinPolicy": WhiteListRoundRobinPolicy,
"TokenAwarePolicy": TokenAwarePolicy,
"HostFilterPolicy": HostFilterPolicy,
"SimpleConvictionPolicy": SimpleConvictionPolicy,
"ExponentialReconnectionPolicy": ExponentialReconnectionPolicy,
"RetryPolicy": RetryPolicy,
"IdentityTranslator": IdentityTranslator,
"NoSpeculativeExecutionPlan": NoSpeculativeExecutionPlan,
"NoSpeculativeExecutionPolicy": NoSpeculativeExecutionPolicy,
}
except ImportError:
pass
LOAD_BALANCING_POLICY_MAP = {
"HostDistance": HostDistance,
"LoadBalancingPolicy": LoadBalancingPolicy,
"RoundRobinPolicy": RoundRobinPolicy,
"DCAwareRoundRobinPolicy": DCAwareRoundRobinPolicy,
"WhiteListRoundRobinPolicy": WhiteListRoundRobinPolicy,
"TokenAwarePolicy": TokenAwarePolicy,
"HostFilterPolicy": HostFilterPolicy,
"SimpleConvictionPolicy": SimpleConvictionPolicy,
"ExponentialReconnectionPolicy": ExponentialReconnectionPolicy,
"RetryPolicy": RetryPolicy,
"IdentityTranslator": IdentityTranslator,
"NoSpeculativeExecutionPlan": NoSpeculativeExecutionPlan,
"NoSpeculativeExecutionPolicy": NoSpeculativeExecutionPolicy,
}
def __virtual__():
"""
Return virtual name of the module only if the python driver can be loaded.