def get_data_fixture(default='sequential'): # It must be 'sequential', 'static_sequential', 'global_sequential', 'random' or 'path.to.CustomDataFixtureClass' try: INTERNAL_DATA_FIXTURES = { 'sequential': SequentialDataFixture(), 'static_sequential': StaticSequentialDataFixture(), 'global_sequential': GlobalSequentialDataFixture(), 'random': RandomDataFixture() } if hasattr(settings, 'DDF_DEFAULT_DATA_FIXTURE'): if settings.DDF_DEFAULT_DATA_FIXTURE in INTERNAL_DATA_FIXTURES.keys( ): return INTERNAL_DATA_FIXTURES[ settings.DDF_DEFAULT_DATA_FIXTURE] else: # path.to.CustomDataFixtureClass mod_name, obj_name = get_mod_func( settings.DDF_DEFAULT_DATA_FIXTURE) module = import_module(mod_name) custom_data_fixture = getattr(module, obj_name) return custom_data_fixture() else: return INTERNAL_DATA_FIXTURES[default] except: six.reraise( DDFImproperlyConfigured, DDFImproperlyConfigured( "DDF_DEFAULT_DATA_FIXTURE (%s) must be 'sequential', 'static_sequential', 'global_sequential', 'random' or 'path.to.CustomDataFixtureClass'." % settings.DDF_DEFAULT_DATA_FIXTURE), sys.exc_info()[2])
def setUp(self): self.fixture = StaticSequentialDataFixture()
settings, config_name) else default except DDFImproperlyConfigured: six.reraise( DDFImproperlyConfigured, DDFImproperlyConfigured( "%s (%s) must be True or False." % (config_name, getattr(settings, config_name))), sys.exc_info()[2]) # DDF_DEFAULT_DATA_FIXTURE default = 'sequential' # It must be 'sequential', 'static_sequential', 'global_sequential', 'random' or 'path.to.CustomDataFixtureClass' try: INTERNAL_DATA_FIXTURES = { 'sequential': SequentialDataFixture(), 'static_sequential': StaticSequentialDataFixture(), 'global_sequential': GlobalSequentialDataFixture(), 'random': RandomDataFixture() } if hasattr(settings, 'DDF_DEFAULT_DATA_FIXTURE'): if settings.DDF_DEFAULT_DATA_FIXTURE in INTERNAL_DATA_FIXTURES.keys(): DDF_DEFAULT_DATA_FIXTURE = INTERNAL_DATA_FIXTURES[ settings.DDF_DEFAULT_DATA_FIXTURE] else: # path.to.CustomDataFixtureClass mod_name, obj_name = get_mod_func( settings.DDF_DEFAULT_DATA_FIXTURE) module = import_module(mod_name) custom_data_fixture = getattr(module, obj_name) DDF_DEFAULT_DATA_FIXTURE = custom_data_fixture() else: