blob: 6c25689cdb0fa2353d137a168ce4440c2b260b18 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
from django.utils.deprecation import MiddlewareMixin
from authentication.models import UserProfile
class UserProfileMiddleware(MiddlewareMixin):
def process_request(self, request):
if request.user.is_authenticated:
try:
user_profile = UserProfile.objects.get(user=request.user)
except UserProfile.DoesNotExist:
user_profile = UserProfile(user=request.user)
user_profile.save()
request.user.profile = user_profile
else:
request.user.profile = None
|