python - 用于通过解析器中的参数从特定模型中删除数据的 BaseCommand

Question

我需要一个 BaseCommand 来删除参数中指定模型的条目。

运行这个:./manage.py delete_data_model app.shop

# delete_data_model.py

from django.core.management.base import BaseCommand, CommandError
from django.db.transaction import atomic


class Command(BaseCommand):
    help = "Deleted data from model"

    def add_arguments(self, parser):
        parser.add_argument('model', required=True, type=str)

    def handle(self, *args, **options):
        self.stdout.write("Begin")
        with atomic(using='default'):
            try:
                path = options['model']
                app, model = path.split('.')[:2]
                from (app) import (model) as Model  # ??? how do this
                Model.objects.all().delete()
            except Exception as e:
                raise CommandError("Error {}".format(e))
        self.stdout.write("Complete")

我希望空表 Shop。

Answer 1

试试这个

from django.core.management.base import BaseCommand, CommandError
from django.db.transaction import atomic
<b>from django.apps import apps</b>


class Command(BaseCommand):
    help = "Deleted data from model"

    def add_arguments(self, parser):
        parser.add_argument('model', required=True, type=str)

    def handle(self, *args, **options):
        self.stdout.write("Begin")
        with atomic(using='default'):
            try:
                path = options['model']
                app, model = path.split('.')[:2]
                <b>Model = apps.get_model(app_label=app, model_name=model)</b>
                Model.objects.all().delete()
            except Exception as e:
                raise CommandError("Error {}".format(e))
        self.stdout.write("Complete")

引用:
1. Django: Get model from string?
2. apps.get_model(app_label, model_name, require_ready=True) -- Django Doc