from allauth.socialaccount.adapter import DefaultSocialAccountAdapter
from allauth.account.adapter import DefaultAccountAdapter
from django.core.exceptions import PermissionDenied
from django.contrib.auth.models import Group

class FairphoneAccountAdapter(DefaultSocialAccountAdapter):
    
    def is_open_for_signup(self, request, sociallogin):
        return True
    
    def save_user(self, request, sociallogin, form=None):
        u =DefaultSocialAccountAdapter.save_user(self, request, sociallogin, form=None)
        if u.email.split('@')[1] == "fairphone.com":
            g = Group.objects.get(name='FairphoneSoftwareTeam') 
            g.user_set.add(u)
        return u
            
    def populate_user(self,
                      request,
                      sociallogin,
                      data):
        u = DefaultSocialAccountAdapter.populate_user(self,request,sociallogin,data)
        if not u.email.split('@')[1] == "fairphone.com":
             raise PermissionDenied()
        return u

class FormAccountAdapter(DefaultAccountAdapter):
    def is_open_for_signup(self, request):
        return False