def get(self, model_class, shelve=False, named_shelve=None, **kwargs):
"""
Create an instance with data and persist it.
:shelve: the current configuration will be stored in the DDF library.
:named_shelve: restore configuration saved in DDF library with a name.
"""
if 'locators' in kwargs:
locators = kwargs.pop('locators')
else:
locators = []
if len(kwargs) > 0 and locators == [] and _get_instance_by_args(model_class, **kwargs) != None:
ret = _get_instance_by_args(model_class, **kwargs)
return ret
elif locators:
# we need to be sure that we havent build something meeting the locators yet.
ret = self._check_against_db(model_class, locators, **kwargs)
if ret:
return ret
if(model_class in get_created_classes() and model_class not in ALLOW_MULTIPLE):
return get_instance_by_class(model_class)
instance = self.new(model_class, locators, shelve=shelve, named_shelve=named_shelve, **kwargs)
if is_model_abstract(model_class):
raise InvalidModelError(get_unique_model_name(model_class))
try:
if self.validate_models:
instance.full_clean()
if model_class in _PRE_SAVE:
try:
_PRE_SAVE[model_class](instance)
except Exception as e:
six.reraise(InvalidReceiverError, InvalidReceiverError(e), sys.exc_info()[2])
instance = self._save_the_instance(instance)
created_instances.add(instance)
if model_class in _POST_SAVE:
try:
_POST_SAVE[model_class](instance)
except Exception as e:
six.reraise(InvalidReceiverError, InvalidReceiverError(e), sys.exc_info()[2])
except Exception as e:
if self.print_errors:
print_field_values(instance)
six.reraise(BadDataError, BadDataError(get_unique_model_name(model_class), e), sys.exc_info()[2])
self.fields_processed = [] # TODO: need more tests for M2M and Copier
self.pending_fields = []
for field in get_many_to_many_fields_from_model(model_class):
if field.name in kwargs.keys(): # TODO: library
manytomany_field = getattr(instance, field.name)
fixture = kwargs[field.name]
try:
self._process_many_to_many_field(field, manytomany_field, fixture, instance)
except InvalidManyToManyConfigurationError as e:
six.reraise(InvalidManyToManyConfigurationError, e, sys.exc_info()[2])
except Exception as e:
six.reraise(InvalidManyToManyConfigurationError, InvalidManyToManyConfigurationError(get_unique_field_name(field), e), sys.exc_info()[2])
return instance
def get(self, model_class, shelve=False, named_shelve=None, **kwargs):
"""
Create an instance with data and persist it.
:shelve: the current configuration will be stored in the DDF library.
:named_shelve: restore configuration saved in DDF library with a name.
"""
instance = self.new(model_class,
shelve=shelve,
named_shelve=named_shelve,
**kwargs)
if is_model_abstract(model_class):
raise InvalidModelError(get_unique_model_name(model_class))
try:
if self.validate_models:
instance.full_clean()
if model_class in _PRE_SAVE:
try:
_PRE_SAVE[model_class](instance)
except Exception as e:
six.reraise(InvalidReceiverError, InvalidReceiverError(e),
sys.exc_info()[2])
self._save_the_instance(instance)
if model_class in _POST_SAVE:
try:
_POST_SAVE[model_class](instance)
except Exception as e:
six.reraise(InvalidReceiverError, InvalidReceiverError(e),
sys.exc_info()[2])
except Exception as e:
if self.print_errors:
print_field_values(instance)
six.reraise(BadDataError,
BadDataError(get_unique_model_name(model_class), e),
sys.exc_info()[2])
self.fields_processed = [] # TODO: need more tests for M2M and Copier
self.pending_fields = []
for field in get_many_to_many_fields_from_model(model_class):
if field.name in kwargs.keys(): # TODO: library
manytomany_field = getattr(instance, field.name)
fixture = kwargs[field.name]
try:
self._process_many_to_many_field(field, manytomany_field,
fixture, instance)
except InvalidManyToManyConfigurationError as e:
six.reraise(InvalidManyToManyConfigurationError, e,
sys.exc_info()[2])
except Exception as e:
six.reraise(
InvalidManyToManyConfigurationError,
InvalidManyToManyConfigurationError(
get_unique_field_name(field), e),
sys.exc_info()[2])
return instance
def get(self, model_class, ddf_lesson=None, **kwargs):
'''
Create an instance with data and persist it.
:ddf_lesson: a custom lesson that will be used to create the model object.
'''
instance = self.new(model_class, ddf_lesson=ddf_lesson, **kwargs)
if is_model_abstract(model_class):
raise InvalidModelError(get_unique_model_name(model_class))
try:
if self.validate_models:
instance.full_clean()
if model_class in _PRE_SAVE:
try:
_PRE_SAVE[model_class](instance)
except Exception as e:
six.reraise(InvalidReceiverError, InvalidReceiverError(e),
sys.exc_info()[2])
self._save_the_instance(instance)
if model_class in _POST_SAVE:
try:
_POST_SAVE[model_class](instance)
except Exception as e:
six.reraise(InvalidReceiverError, InvalidReceiverError(e),
sys.exc_info()[2])
except Exception as e:
if self.print_errors:
print_field_values(instance)
six.reraise(BadDataError,
BadDataError(get_unique_model_name(model_class), e),
sys.exc_info()[2])
self.fields_processed = [] # TODO: need more tests for M2M and Copier
self.pending_fields = []
for field in get_many_to_many_fields_from_model(model_class):
if field.name in kwargs.keys(): # TODO: library
manytomany_field = getattr(instance, field.name)
fixture = kwargs[field.name]
try:
self._process_many_to_many_field(field, manytomany_field,
fixture, instance)
except InvalidManyToManyConfigurationError as e:
six.reraise(InvalidManyToManyConfigurationError, e,
sys.exc_info()[2])
except Exception as e:
six.reraise(
InvalidManyToManyConfigurationError,
InvalidManyToManyConfigurationError(
get_unique_field_name(field), e),
sys.exc_info()[2])
return instance
def get(self, model_class, shelve=False, named_shelve=None, **kwargs):
"""
Create an instance with data and persist it.
:shelve: the current configuration will be stored in the DDF library.
:named_shelve: restore configuration saved in DDF library with a name.
"""
instance = self.new(model_class, shelve=shelve, named_shelve=named_shelve, **kwargs)
if is_model_abstract(model_class):
raise InvalidModelError(get_unique_model_name(model_class))
try:
if self.validate_models:
instance.full_clean()
if model_class in _PRE_SAVE:
try:
_PRE_SAVE[model_class](instance)
except Exception as e:
six.reraise(InvalidReceiverError, InvalidReceiverError(e), sys.exc_info()[2])
self._save_the_instance(instance)
if model_class in _POST_SAVE:
try:
_POST_SAVE[model_class](instance)
except Exception as e:
six.reraise(InvalidReceiverError, InvalidReceiverError(e), sys.exc_info()[2])
except Exception as e:
if self.print_errors:
print_field_values(instance)
six.reraise(BadDataError, BadDataError(get_unique_model_name(model_class), e), sys.exc_info()[2])
self.fields_processed = [] # TODO: need more tests for M2M and Copier
self.pending_fields = []
for field in get_many_to_many_fields_from_model(model_class):
if field.name in kwargs.keys(): # TODO: library
manytomany_field = getattr(instance, field.name)
fixture = kwargs[field.name]
try:
self._process_many_to_many_field(field, manytomany_field, fixture, instance)
except InvalidManyToManyConfigurationError as e:
six.reraise(InvalidManyToManyConfigurationError, e, sys.exc_info()[2])
except Exception as e:
six.reraise(InvalidManyToManyConfigurationError, InvalidManyToManyConfigurationError(get_unique_field_name(field), e), sys.exc_info()[2])
return instance
