132 lines
		
	
	
		
			4.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			132 lines
		
	
	
		
			4.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| """passbook oauth_client forms"""
 | |
| 
 | |
| from django import forms
 | |
| 
 | |
| from passbook.admin.forms.source import SOURCE_FORM_FIELDS
 | |
| from passbook.flows.models import Flow, FlowDesignation
 | |
| from passbook.sources.oauth.models import OAuthSource
 | |
| from passbook.sources.oauth.types.manager import MANAGER
 | |
| 
 | |
| 
 | |
| class OAuthSourceForm(forms.ModelForm):
 | |
|     """OAuthSource Form"""
 | |
| 
 | |
|     def __init__(self, *args, **kwargs):
 | |
|         super().__init__(*args, **kwargs)
 | |
|         self.fields["authentication_flow"].queryset = Flow.objects.filter(
 | |
|             designation=FlowDesignation.AUTHENTICATION
 | |
|         )
 | |
|         self.fields["enrollment_flow"].queryset = Flow.objects.filter(
 | |
|             designation=FlowDesignation.ENROLLMENT
 | |
|         )
 | |
|         if hasattr(self.Meta, "overrides"):
 | |
|             for overide_field, overide_value in getattr(self.Meta, "overrides").items():
 | |
|                 self.fields[overide_field].initial = overide_value
 | |
|                 self.fields[overide_field].widget.attrs["readonly"] = "readonly"
 | |
| 
 | |
|     class Meta:
 | |
| 
 | |
|         model = OAuthSource
 | |
|         fields = SOURCE_FORM_FIELDS + [
 | |
|             "provider_type",
 | |
|             "request_token_url",
 | |
|             "authorization_url",
 | |
|             "access_token_url",
 | |
|             "profile_url",
 | |
|             "consumer_key",
 | |
|             "consumer_secret",
 | |
|         ]
 | |
|         widgets = {
 | |
|             "name": forms.TextInput(),
 | |
|             "consumer_key": forms.TextInput(),
 | |
|             "consumer_secret": forms.TextInput(),
 | |
|             "provider_type": forms.Select(choices=MANAGER.get_name_tuple()),
 | |
|         }
 | |
| 
 | |
| 
 | |
| class GitHubOAuthSourceForm(OAuthSourceForm):
 | |
|     """OAuth Source form with pre-determined URL for GitHub"""
 | |
| 
 | |
|     class Meta(OAuthSourceForm.Meta):
 | |
| 
 | |
|         overrides = {
 | |
|             "provider_type": "github",
 | |
|             "request_token_url": "",
 | |
|             "authorization_url": "https://github.com/login/oauth/authorize",
 | |
|             "access_token_url": "https://github.com/login/oauth/access_token",
 | |
|             "profile_url": "https://api.github.com/user",
 | |
|         }
 | |
| 
 | |
| 
 | |
| class TwitterOAuthSourceForm(OAuthSourceForm):
 | |
|     """OAuth Source form with pre-determined URL for Twitter"""
 | |
| 
 | |
|     class Meta(OAuthSourceForm.Meta):
 | |
| 
 | |
|         overrides = {
 | |
|             "provider_type": "twitter",
 | |
|             "request_token_url": "https://api.twitter.com/oauth/request_token",
 | |
|             "authorization_url": "https://api.twitter.com/oauth/authenticate",
 | |
|             "access_token_url": "https://api.twitter.com/oauth/access_token",
 | |
|             "profile_url": (
 | |
|                 "https://api.twitter.com/1.1/account/"
 | |
|                 "verify_credentials.json?include_email=true"
 | |
|             ),
 | |
|         }
 | |
| 
 | |
| 
 | |
| class FacebookOAuthSourceForm(OAuthSourceForm):
 | |
|     """OAuth Source form with pre-determined URL for Facebook"""
 | |
| 
 | |
|     class Meta(OAuthSourceForm.Meta):
 | |
| 
 | |
|         overrides = {
 | |
|             "provider_type": "facebook",
 | |
|             "request_token_url": "",
 | |
|             "authorization_url": "https://www.facebook.com/v7.0/dialog/oauth",
 | |
|             "access_token_url": "https://graph.facebook.com/v7.0/oauth/access_token",
 | |
|             "profile_url": "https://graph.facebook.com/v7.0/me?fields=id,name,email",
 | |
|         }
 | |
| 
 | |
| 
 | |
| class DiscordOAuthSourceForm(OAuthSourceForm):
 | |
|     """OAuth Source form with pre-determined URL for Discord"""
 | |
| 
 | |
|     class Meta(OAuthSourceForm.Meta):
 | |
| 
 | |
|         overrides = {
 | |
|             "provider_type": "discord",
 | |
|             "request_token_url": "",
 | |
|             "authorization_url": "https://discord.com/api/oauth2/authorize",
 | |
|             "access_token_url": "https://discord.com/api/oauth2/token",
 | |
|             "profile_url": "https://discord.com/api/users/@me",
 | |
|         }
 | |
| 
 | |
| 
 | |
| class GoogleOAuthSourceForm(OAuthSourceForm):
 | |
|     """OAuth Source form with pre-determined URL for Google"""
 | |
| 
 | |
|     class Meta(OAuthSourceForm.Meta):
 | |
| 
 | |
|         overrides = {
 | |
|             "provider_type": "google",
 | |
|             "request_token_url": "",
 | |
|             "authorization_url": "https://accounts.google.com/o/oauth2/auth",
 | |
|             "access_token_url": "https://accounts.google.com/o/oauth2/token",
 | |
|             "profile_url": "https://www.googleapis.com/oauth2/v1/userinfo",
 | |
|         }
 | |
| 
 | |
| 
 | |
| class AzureADOAuthSourceForm(OAuthSourceForm):
 | |
|     """OAuth Source form with pre-determined URL for AzureAD"""
 | |
| 
 | |
|     class Meta(OAuthSourceForm.Meta):
 | |
| 
 | |
|         overrides = {
 | |
|             "provider_type": "azure-ad",
 | |
|             "request_token_url": "",
 | |
|             "authorization_url": "https://login.microsoftonline.com/common/oauth2/authorize",
 | |
|             "access_token_url": "https://login.microsoftonline.com/common/oauth2/token",
 | |
|             "profile_url": "https://graph.windows.net/myorganization/me?api-version=1.6",
 | |
|         }
 | 
