def __init__(self, options):
self.model = get_model(getattr(options, 'model'))
self.publish_fields = getattr(options, 'publish_fields', None)
if not self.publish_fields:
self.publish_fields = self.get_fields(self.model)
if isinstance(self.publish_fields, str):
self.publish_fields = (self.publish_fields, )
self.update_fields = getattr(options, 'update_fields', ())
if isinstance(self.update_fields, str):
self.update_fields = (self.update_fields, )
self.id_field = getattr(options, 'id_field', 'pk')
self.base_channel = getattr(options, 'base_channel', self.model._meta.model_name)
def __init__(self, options):
self.model = get_model(getattr(options, 'model'))
self.publish_fields = getattr(options, 'publish_fields', None)
if not self.publish_fields:
self.publish_fields = self.get_fields(self.model)
if isinstance(self.publish_fields, str):
self.publish_fields = (self.publish_fields, )
self.update_fields = getattr(options, 'update_fields', ())
if isinstance(self.update_fields, str):
self.update_fields = (self.update_fields, )
self.id_field = getattr(options, 'id_field', 'pk')
self.base_channel = getattr(options, 'base_channel', self.model._meta.model_name)
def get_base_channel(cls):
if hasattr(cls.Meta, 'base_channel'):
return '{}|'.format(getattr(cls.Meta, 'base_channel'))
return '{}|'.format(get_model(cls.Meta.model)._meta.model_name)
def get_base_channel(cls):
if hasattr(cls.Meta, 'base_channel'):
return '{}|'.format(getattr(cls.Meta, 'base_channel'))
return '{}|'.format(get_model(cls.Meta.model)._meta.model_name)
def test_get_model(self):
parent_model = 'tests.ParentModel'
model = model_tools.get_model(parent_model)
self.assertEqual(model, ParentModel)
