openstack/deb-python-cassandra-driver
Code Issues Proposed changes

cqle: Fix minor issues with polymorphic models

Browse Source
This commit is contained in:
jpuerta
2015-06-02 16:29:12 +01:00
parent 8bfcfb8824
commit be0d0037ea
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
cassandra/cqlengine/models.py
View File

@@ -433,12 +433,12 @@ class BaseModel(object):
poly_base._discover_polymorphic_submodels() poly_base._discover_polymorphic_submodels()
klass = poly_base._get_model_by_discriminator_value(disc_key) klass = poly_base._get_model_by_discriminator_value(disc_key)
if klass is None: if klass is None:
raise PolyMorphicModelException( raise PolymorphicModelException(
'unrecognized discriminator column {0} for class {1}'.format(poly_key, poly_base.__name__) 'unrecognized discriminator column {0} for class {1}'.format(disc_key, poly_base.__name__)
) )
if not issubclass(klass, cls): if not issubclass(klass, cls):
raise PolyMorphicModelException( raise PolymorphicModelException(
'{0} is not a subclass of {1}'.format(klass.__name__, cls.__name__) '{0} is not a subclass of {1}'.format(klass.__name__, cls.__name__)
) )