encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-01-23 15:54:16 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update throttling.py

Browse Source 
Use pk pseudo attribute for identifying the user (in case the user model is not the default and has a different column name for the unique id)
This commit is contained in:
Kaptian 2015-02-05 13:27:26 -08:00
parent 3b00824560
commit fffde8a63b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
rest_framework/throttling.py
View File

@ -191,7 +191,7 @@ class UserRateThrottle(SimpleRateThrottle):
def get_cache_key(self, request, view): def get_cache_key(self, request, view):
if request.user.is_authenticated(): if request.user.is_authenticated():
ident = request.user.id ident = request.user.pk
else: else:
ident = self.get_ident(request) ident = self.get_ident(request)
@ -239,7 +239,7 @@ class ScopedRateThrottle(SimpleRateThrottle):
with the '.throttle_scope` property of the view. with the '.throttle_scope` property of the view.
""" """
if request.user.is_authenticated(): if request.user.is_authenticated():
ident = request.user.id ident = request.user.pk
else: else:
ident = self.get_ident(request) ident = self.get_ident(request)