aboutsummaryrefslogtreecommitdiff
path: root/middleware/userprofilemiddleware.py
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