0

このmodels.pyを考えてみましょう:

from django.db import models
from django.db.models import signals


class CannotRemoveLastAdmin(Exception):
    def __init__(self, env):
        super(CannotRemoveLastAdmin, self).__init__(
            u'Cannot remove the last admin of environment "%s"' % env)


class Environment(models.Model):
    name = models.CharField(max_length=100, unique=True)
    users = models.ManyToManyField('auth.user', through='UserEnvironment')

    def __unicode__(self):
        return self.name


class UserEnvironment(models.Model):
    environment = models.ForeignKey('Environment')
    user = models.ForeignKey('auth.user')
    is_admin = models.BooleanField()
    default = models.BooleanField()
    creation_datetime = models.DateTimeField(auto_now_add=True)


def admin_required_delete(sender, instance, **kwargs):
    if not instance.is_admin:
        return

    admins = UserEnvironment.objects.filter(environment=instance.environment,
        is_admin=True).exclude(pk=instance.pk).count()

    if not admins:
        raise CannotRemoveLastAdmin(instance)
signals.pre_delete.connect(admin_required_delete, sender=UserEnvironment)

UserEnvironment pre_save シグナルが送信されているため、環境を削除できません。

<<< jpic@zen!7953 G:delete_with_m2m_through jpic_so_env
>>> ./manage.py shell_plus
From 'admin' autoload: LogEntry
From 'auth' autoload: Permission, Group, User
From 'contenttypes' autoload: ContentType
From 'sessions' autoload: Session
From 'sites' autoload: Site
From 'account' autoload: Account, SignupCode, SignupCodeResult, EmailAddress, EmailConfirmation, AccountDeletion
From 'profiles' autoload: Profile
From 'test_app' autoload: Environment, UserEnvironment
Python 2.7.3 (default, Apr 24 2012, 00:06:13) 
[GCC 4.7.0 20120414 (prerelease)] on linux2
Type "help", "copyright", "credits" or "license" for more information.
(InteractiveConsole)
>>> Environment.objects.all().delete()
Traceback (most recent call last):
  File "<console>", line 1, in <module>
  File "/tmp/jpic_so_env/lib/python2.7/site-packages/django/db/models/query.py", line 514, in delete
    collector.delete()
  File "/tmp/jpic_so_env/lib/python2.7/site-packages/django/db/models/deletion.py", line 61, in decorated
    func(self, *args, **kwargs)
  File "/tmp/jpic_so_env/lib/python2.7/site-packages/django/db/models/deletion.py", line 239, in delete
    sender=model, instance=obj, using=self.using
  File "/tmp/jpic_so_env/lib/python2.7/site-packages/django/dispatch/dispatcher.py", line 172, in send
    response = receiver(signal=self, sender=sender, **named)
  File "/tmp/so_test_project/test_app/models.py", line 35, in admin_required_delete
    raise CannotRemoveLastAdmin(instance)
CannotRemoveLastAdmin: Cannot remove the last admin of environment "UserEnvironment object"
>>> 

環境が削除されている場合を除いて、この例外、CannotRemoveLastAdmin を維持したいと思います。

貼り付けることで、この問題を安全に再現できます。

cd /tmp
rm -rf jpic_so_env jpic_so_test_project
virtualenv jpic_so_env
source jpic_so_env/bin/activate
git clone https://github.com/jpic/django_test.git -b delete_with_m2m_through jpic_so_test_project
cd so_test_project
git checkout delete_with_m2m_through
pip install django
pip install -r requirements.txt
./manage.py shell_plus

そして実行します:

Environment.objects.all().delete()

setUp()これは、問題としてテストを実行するためにのみ必要なEnvironment.objects.all().delete()ので、モデルを次のような余分なフィールドで混乱させたくありませんmarked_for_delete...

ハッキングをお楽しみください。ご協力いただきありがとうございます。

4

1 に答える 1

0

モデルにこのようなポリシーを適用する場合は、環境にマークを付ける必要があります。そうしないと、環境を削除できなくなるためです。

mark_for_deleteオプションが本当に気に入らない場合は、回避策として環境名フィールドを(特別な値を設定して)悪用する可能性があります。これは私にはもっとハックなようです。

于 2012-11-28T13:01:19.253 に答える