2012-02-24 106 views
1

我有一個模型通過OneToOneField引用另一個模型,因此當您使用Django的內置delete_selected管理員操作時,關聯模型的數據不會被刪除。我想編寫一個自定義管理操作來刪除相關模型中的數據。從管理員中刪除關聯的OneToOneField模型中的SQL數據

這裏是我的模型:

class Party(models.Model): 
    TYPE_CHOICES=(
     ('P','Person'), 
     ('O','Organization') 
    ) 
    partyType = models.CharField(max_length=1, choices=TYPE_CHOICES) 
    name = models.CharField(max_length=100) 
    comment = models.CharField(max_length=500,blank=True) 
    accessIdCrossRef=models.IntegerField(blank=True, null=True) 
    mailingLists = models.ManyToManyField(MailingList) 
    inMainList=models.BooleanField(default=False) 
    inSubList=models.BooleanField(default=False) 

    class Meta: 
     db_table='party' 
     ordering=['name',] 

    def __unicode__(self): 
     return self.name 

class Person(models.Model): 
    party = models.OneToOneField(Party, editable=False) 
    firstName=models.CharField(max_length=60) 
    lastName=models.CharField(max_length=60) 
    ... 


    def save(self): 
     if None == self.party : 
      print 'Creating party for person' 
      p = Party() 
      p.partyType = 'P' 
      p.save() 
      self.party = p 

     # Get address to set party name used in list 
     city="" 
     state="" 
     postalCode="" 
     try: 
      partyAddress = PartyPostalAddress.objects.get(party=self.party) 
      address = partyAddress.postalAddress 
      city=address.city 
      state=address.state 
      postalCode=address.postalCode 
     except PartyPostalAddress.DoesNotExist: 
      pass 

     self.party.name = '%s, %s - %s, %s %s' %(self.lastName, self.firstName, city, state, postalCode) 
     self.party.save() 
     super(Person,self).save() 

我的假設是在我的模型是這樣寫:

def delete(self): 
    self.party.delete() 
    self.delete() 

而像這樣的管理措施:

class PersonAdmin(admin.ModelAdmin): 
    list_display = ('lastName','firstName') 
    search_fields = ('firstName', 'lastName') 
     actions=['really_delete_selected'] 

    def get_actions(self, request): 
    actions = super(PersonAdmin, self).get_actions(request) 
    del actions['delete_selected'] 
    return actions 

    def really_delete_selected(self, request, queryset): 
    for obj in queryset: 
     obj.delete() 

    if queryset.count() == 1: 
     message_bit = "1 person was" 
    else: 
     message_bit = "%s people were" % queryset.count() 
    self.message_user(request, "%s successfully deleted." % message_bit) 
    really_delete_selected.short_description = "Delete selected entries" 

那刪除person.party和大部分人,但因爲人的party OneToOneFi而拋出錯誤現在是空的。具體的錯誤是:

「/ admin/common/person/ AssertionError由於它的id屬性設置爲None,所以無法刪除對象。」

任何想法? This,thisthis問題是相關的,但其中只有一個使用了OneToOneField,他錯誤地做了這個。

回答

1

清理乾淨了!

我的模型:

class Party(models.Model): 
    name = models.CharField(max_length=100) 
    ... 

class Person(models.Model): 
    party = models.OneToOneField(Party, editable=False) 
    firstName=models.CharField(max_length=60) 
    lastName=models.CharField(max_length=60) 

    def delete(self): 
     d = self.party.id 
     Party.objects.get(id__exact=d).delete() 

我的管理員:

class PersonAdmin(admin.ModelAdmin): 
    actions=['really_delete_selected'] 

    def get_actions(self, request): 
     actions = super(PersonAdmin, self).get_actions(request) 
     del actions['delete_selected'] 
     return actions 

    def really_delete_selected(self, request, queryset): 
     for obj in queryset: 
      obj.delete() 

     if queryset.count() == 1: 
      message_bit = "1 person was" 
     else: 
      message_bit = "%s people were" % queryset.count() 

     self.message_user(request, "%s successfully deleted." % message_bit) 
     really_delete_selected.short_description = "Delete selected entries" 
     ... 
1

我感覺它應該像切換兩個刪除序列一樣簡單(除非您已經嘗試過)。由於該人與聚會相關聯,因此一旦刪除了人,就無法訪​​問聚會。因此你應該這樣做

person.party.delete() 
person.delete() 
+0

謝謝,@Priyeshj。接得好。但是,仍然沒有使用該編輯。 – reK1NDLE 2012-02-27 15:46:32

相關問題