Dirk Vogt | 57a615d | 2017-05-04 22:29:54 +0200 | [diff] [blame] | 1 | from allauth.socialaccount.adapter import DefaultSocialAccountAdapter |
| 2 | from allauth.account.adapter import DefaultAccountAdapter |
| 3 | from django.core.exceptions import PermissionDenied |
| 4 | from django.contrib.auth.models import Group |
| 5 | |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 6 | |
Dirk Vogt | 57a615d | 2017-05-04 22:29:54 +0200 | [diff] [blame] | 7 | class FairphoneAccountAdapter(DefaultSocialAccountAdapter): |
Dirk Vogt | 57a615d | 2017-05-04 22:29:54 +0200 | [diff] [blame] | 8 | def is_open_for_signup(self, request, sociallogin): |
| 9 | return True |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 10 | |
Dirk Vogt | 57a615d | 2017-05-04 22:29:54 +0200 | [diff] [blame] | 11 | def save_user(self, request, sociallogin, form=None): |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 12 | u = DefaultSocialAccountAdapter.save_user( |
| 13 | self, request, sociallogin, form=None |
| 14 | ) |
| 15 | if u.email.split("@")[1] == "fairphone.com": |
| 16 | g = Group.objects.get(name="FairphoneSoftwareTeam") |
Dirk Vogt | 57a615d | 2017-05-04 22:29:54 +0200 | [diff] [blame] | 17 | g.user_set.add(u) |
| 18 | return u |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 19 | |
| 20 | def populate_user(self, request, sociallogin, data): |
| 21 | u = DefaultSocialAccountAdapter.populate_user( |
| 22 | self, request, sociallogin, data |
| 23 | ) |
| 24 | if not u.email.split("@")[1] == "fairphone.com": |
| 25 | raise PermissionDenied() |
Dirk Vogt | 57a615d | 2017-05-04 22:29:54 +0200 | [diff] [blame] | 26 | return u |
| 27 | |
Mitja Nikolaus | cb50f2c | 2018-08-24 13:54:48 +0200 | [diff] [blame^] | 28 | |
Dirk Vogt | 57a615d | 2017-05-04 22:29:54 +0200 | [diff] [blame] | 29 | class FormAccountAdapter(DefaultAccountAdapter): |
| 30 | def is_open_for_signup(self, request): |
| 31 | return False |