编写Django的自定义管理命令
非常简单; 并继承Django的
loaddata
命令使其变得微不足道:
excluding_loaddata.py
from optparse import make_option
from django.core.management.commands.loaddata import Command as LoadDataCommand
class Command(LoadDataCommand):
option_list = LoadDataCommand.option_list + (
make_option('-e', '--exclude', action='append',
help='Exclude given fixture/s from being loaded'),
)
def handle(self, *fixture_labels, **options):
self.exclude = options.get('exclude')
return super(Command, self).handle(*fixture_labels, **options)
def find_fixtures(self, *args, **kwargs):
updated_fixtures = []
fixture_files = super(Command, self).find_fixtures(*args, **kwargs)
for fixture_file in fixture_files:
file, directory, name = fixture_file
if file in self.exclude or directory in self.exclude or name in self.exclude:
if self.verbosity >= 1:
self.stdout.write('Fixture skipped (excluded: %s, matches %s)' %
(self.exclude, [file, directory, name]))
else:
updated_fixtures.append(fixture_file)
return updated_fixtures
用法:
$ python manage.py excluding_loaddata app/fixtures/* -e not_this_fixture
loaddata
有任何排除或忽略选项。也许你可以将所有想要包含的fixture合并成一个更大的fixture,然后通过名称导入它? - Fiverloaddata
以获取所需的fixture,并使用call_command
运行该命令。 - tutuDajuju