在 Django 中编写自己的管理命令很简单;并且从 Django 的命令继承loaddata
使它变得微不足道:
exclude_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
# exclude a matched file path, directory or name (filename without extension)
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