Commit 89d77c3b by Őry Máté

dashboard: create default profile on login

ref #43
parent e1db1dc1
...@@ -2,6 +2,7 @@ from logging import getLogger ...@@ -2,6 +2,7 @@ from logging import getLogger
from django.conf import settings from django.conf import settings
from django.contrib.auth.models import User from django.contrib.auth.models import User
from django.contrib.auth.signals import user_logged_in
from django.db.models import ( from django.db.models import (
Model, ForeignKey, OneToOneField, CharField, IntegerField Model, ForeignKey, OneToOneField, CharField, IntegerField
) )
...@@ -29,6 +30,12 @@ class Profile(Model): ...@@ -29,6 +30,12 @@ class Profile(Model):
instance_limit = IntegerField(default=5) instance_limit = IntegerField(default=5)
def create_profile(sender, user, request, **kwargs):
profile, created = Profile.objects.get_or_create(user=user)
return created
user_logged_in.connect(create_profile)
if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'): if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'):
logger.debug("Register save_org_id to djangosaml2 pre_user_save") logger.debug("Register save_org_id to djangosaml2 pre_user_save")
from djangosaml2.signals import pre_user_save from djangosaml2.signals import pre_user_save
...@@ -54,6 +61,7 @@ if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'): ...@@ -54,6 +61,7 @@ if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'):
value, sender.username) value, sender.username)
return False return False
pre_user_save.connect(save_org_id, weak=False) pre_user_save.connect(save_org_id)
else: else:
logger.debug("Do not register save_org_id to djangosaml2 pre_user_save") logger.debug("Do not register save_org_id to djangosaml2 pre_user_save")
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment