2011-01-06 66 views
7

我試着寫我的模特一些自定義的方法,但我發現了以下錯誤內相關車型的一個問題:過濾模型定義

Attribute Error: 'ForeignRelatedObjectsDescriptor' object has no attribute all|filter

這發生在我運行此代碼:

 
chore = Chore(name='Laundry') 
chore.schedule_set.create(week_day='monday', time_due='17:30:00') 
chore.scheduled() 

有沒有人對如何使這項工作還是什麼我可能會丟失什麼建議嗎?我檢查過Django文檔,但它們似乎只涵蓋了模型的最基本用途。

models.py:

 
from django.db import models 
from datetime import date, timedelta 

class ChoreManager(models.Manager): 
    def by_day(self, week_day): 
     if week_day == 'today': 
      week_day = date.today().strftime("%A") 

     chores = self.filter(week_day=week_day) 

     if chores.count() > 0: 
      return chores 
     else: 
      return False 

    def today(self): 
     return self.by_day(week_day='today') 

class Chore(models.Model): 
    chores = ChoreManager() 
    name = models.CharField(max_length=50) 
    notes = models.TextField(null=True) 

    def scheduled(self, week_day=None): 
     if week_day is None: 
      schedule_count = Chore.schedule_set.all().count() 
     else: 
      if week_day == 'today': 
       week_day = date.today().strftime("%A") 

      schedule_count = Chore.schedule_set.filter(week_day=week_day).count() 

     if schedule_count > 0: 
      return True 
     else: 
      return False 

    def times_by_day(self, week_day): 
     if self.scheduled() == True: 
      if week_day == 'today': 
       week_day = date.today().strftime("%A") 

      return Chore.schedule_set.filter(week_day=week_day).values('time_due') 
     else: 
      return False 

class Schedule(models.Model): 
    chore = models.ForeignKey('Chore') 
    week_day = models.CharField(max_length=9) 
    time_due = models.TimeField() 

    def mark_complete(self): 
     completed_event = Schedule.completedevent_set.create() 
     completed_event.save() 

    def completed_this_week(self): 
     today = date.today() 
     weekstart = today - timedelta(days=today.weekday()) 
     weekend = weekstart + timedelta(days=7, hours=23, minutes=59, seconds=59) 

     if Schedule.completedevent_set.filter(datetime_completed__gte=weekstart, datetime_completed__lte=weekend).count() > 0: 
      return True 
     else: 
      return False 

class CompletedEvent(models.Model): 
    schedule = models.ForeignKey('Schedule') 
    datetime_completed = models.DateTimeField(auto_now_add=True) 

回答

11

變化:

schedule_count = Chore.schedule_set.all().count() 

到:

schedule_count = self.schedule_set.all().count() 
在所有出現

..

+2

謝謝。當Django模型與Django模型實例之間的區別開始變得模糊時,這個答案在接近漫長一天結束時就是黃金。 :) – 2012-12-14 21:09:02