我的网站有一个警报系统。发生某种操作时,系统会在以下模型中记录警报:
class Alert(models.Model):
title = models.CharField(max_length=60)
read = models.BooleanField #if this is a new alert of not
for_user = models.ForeignKey(User) #which user will see it
link = models.CharField(max_length=100)
许多功能需要检查用户拥有的警报数量(主要是显示网站警报选项卡旁边的新警报数)。因此,我创建了这个函数:
@login_required()
def get_alertnum(user):
alert_objects = Alert.objects.filter(read = False, for_user=user)
num = 0
for n in alert_objects:
num += 1
return num
此功能可以访问:
@login_required()
def posting_draft(request):
user = request.user
user_drafts = Draft.objects.filter(user = user)
drafts = dict()
for d in user_drafts:
drafts[d.title] = d.id
alertnum = get_alertnum(user)
return render_to_response('posting_draft.html', {'STATIC_URL':STATIC_URL, 'draft_l' : drafts, 'selected':"dr", alertnum: alertnum})
但是我收到以下错误:
Environment:
Request Method: GET
Request URL: http://127.0.0.1:8000/posting/drafts
Django Version: 1.4
Python Version: 2.7.3
Installed Applications:
('django.contrib.auth',
'django.contrib.contenttypes',
'django.contrib.sessions',
'django.contrib.sites',
'django.contrib.messages',
'django.contrib.staticfiles',
'Knights',
'django.contrib.admin')
Installed Middleware:
('django.middleware.common.CommonMiddleware',
'django.contrib.sessions.middleware.SessionMiddleware',
'django.contrib.auth.middleware.AuthenticationMiddleware',
'django.contrib.messages.middleware.MessageMiddleware')
Traceback:
File "/Library/Python/2.7/site-packages/Django-1.4-py2.7.egg/django/core/handlers/base.py" in get_response
111. response = callback(request, *callback_args, **callback_kwargs)
File "/Library/Python/2.7/site-packages/Django-1.4-py2.7.egg/django/contrib/auth/decorators.py" in _wrapped_view
20. return view_func(request, *args, **kwargs)
File "/Users/Mike/Desktop/Main/Django-Development/BBN/Knights/views.py" in posting_draft
245. alertnum = get_alertnum(user)
File "/Library/Python/2.7/site-packages/Django-1.4-py2.7.egg/django/contrib/auth/decorators.py" in _wrapped_view
19. if test_func(request.user):
File "/Library/Python/2.7/site-packages/Django-1.4-py2.7.egg/django/utils/functional.py" in inner
185. return func(self._wrapped, *args)
Exception Type: AttributeError at /posting/drafts
Exception Value: 'User' object has no attribute 'user'
答案 0 :(得分:13)
您需要从get_alertnum()函数中删除@login_required()装饰器。装饰器假定第一个参数是请求对象,并且正在尝试访问用户属性。
您还可以通过以下方式简化和加快功能:
def get_alertnum(user):
return Alert.objects.filter(read=False, for_user=user).count()
以下是计数方法的说明。
https://docs.djangoproject.com/en/dev/ref/models/querysets/#count
答案 1 :(得分:10)
在课堂上使用@login_required
装饰器时,也会弹出此错误。对于课堂装饰,请使用@method_decorator(login_required)
。请参阅Class based views docs.
答案 2 :(得分:2)
@login_required
装饰器仅适用于第一个参数是请求的函数。您的堆栈跟踪是因为它尝试使用user
对象,就像它是request
对象一样,并且它不起作用。 (正如其他答案所指出的,user
个对象没有.user
属性)
或许get_alertnum()
可以先检查user.is_authenticated()
,如果用户未经过身份验证,则返回0
。
例如:
def get_alertnum(user):
if not user.is_authenticated():
return 0
else:
return Alerts.objects.filter(read=False, for_user=user).count()