django ModelCommand

model_command.py

import re

from django.core.management.base import LabelCommand, CommandError
from django.db.models.loading import get_model

class ModelCommand(LabelCommand):
    args = 'model+'

    def handle_label(self, label, **options):
        if not re.search(r'^\w+\.\w+$', label):
            raise CommandError(u"%s doesn't look like model name" % label)

        model = get_model(*label.split('.'))
        if not model:
            raise CommandError(u"Can't find model %s" % label)

        self.handle_model(model)