forked from GithubBackups/healthchecks
23 lines
647 B
Python
23 lines
647 B
Python
from hc.accounts.models import Profile
|
|
|
|
|
|
class TeamAccessMiddleware(object):
|
|
def process_request(self, request):
|
|
if not request.user.is_authenticated():
|
|
return
|
|
|
|
teams_q = Profile.objects.filter(member__user_id=request.user.id)
|
|
teams_q = teams_q.select_related("user")
|
|
request.teams = list(teams_q)
|
|
|
|
try:
|
|
profile = request.user.profile
|
|
except Profile.DoesNotExist:
|
|
profile = Profile(user=request.user)
|
|
profile.save()
|
|
|
|
if profile.current_team:
|
|
request.team = profile.current_team
|
|
else:
|
|
request.team = profile
|