diff --git a/pycassa/columnfamilymap.py b/pycassa/columnfamilymap.py index 41914b9c..ebc034a6 100644 --- a/pycassa/columnfamilymap.py +++ b/pycassa/columnfamilymap.py @@ -14,6 +14,7 @@ from pycassa.types import CassandraType from pycassa.columnfamily import ColumnFamily import pycassa.util as util +import inspect __all__ = ['ColumnFamilyMap'] @@ -51,7 +52,7 @@ def __init__(self, cls, pool, column_family, raw_columns=False, **kwargs): self.dict_class = util.OrderedDict self.defaults = {} self.fields = [] - for name, val_type in self.cls.__dict__.iteritems(): + for name, val_type in inspect.getmembers(self.cls): if name != 'key' and isinstance(val_type, CassandraType): self.fields.append(name) self.column_validators[name] = val_type