81 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			81 lines
		
	
	
		
			2.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| """passbook OAuth2 Provider Forms"""
 | |
| 
 | |
| from django import forms
 | |
| from django.utils.translation import gettext as _
 | |
| 
 | |
| from passbook.admin.fields import CodeMirrorWidget
 | |
| from passbook.core.expression import PropertyMappingEvaluator
 | |
| from passbook.crypto.models import CertificateKeyPair
 | |
| from passbook.flows.models import Flow, FlowDesignation
 | |
| from passbook.providers.oauth2.generators import (
 | |
|     generate_client_id,
 | |
|     generate_client_secret,
 | |
| )
 | |
| from passbook.providers.oauth2.models import OAuth2Provider, ScopeMapping
 | |
| 
 | |
| 
 | |
| class OAuth2ProviderForm(forms.ModelForm):
 | |
|     """OAuth2 Provider form"""
 | |
| 
 | |
|     def __init__(self, *args, **kwargs):
 | |
|         super().__init__(*args, **kwargs)
 | |
|         self.fields["authorization_flow"].queryset = Flow.objects.filter(
 | |
|             designation=FlowDesignation.AUTHORIZATION
 | |
|         )
 | |
|         self.fields["client_id"].initial = generate_client_id()
 | |
|         self.fields["client_secret"].initial = generate_client_secret()
 | |
|         self.fields["rsa_key"].queryset = CertificateKeyPair.objects.exclude(
 | |
|             key_data__exact=""
 | |
|         )
 | |
|         self.fields["property_mappings"].queryset = ScopeMapping.objects.all()
 | |
| 
 | |
|     class Meta:
 | |
|         model = OAuth2Provider
 | |
|         fields = [
 | |
|             "name",
 | |
|             "authorization_flow",
 | |
|             "client_type",
 | |
|             "client_id",
 | |
|             "client_secret",
 | |
|             "response_type",
 | |
|             "jwt_alg",
 | |
|             "rsa_key",
 | |
|             "redirect_uris",
 | |
|             "sub_mode",
 | |
|             "property_mappings",
 | |
|         ]
 | |
|         widgets = {
 | |
|             "name": forms.TextInput(),
 | |
|         }
 | |
|         labels = {"property_mappings": _("Scopes")}
 | |
|         help_texts = {
 | |
|             "property_mappings": _(
 | |
|                 (
 | |
|                     "Select which scopes <b>can</b> be used by the client. "
 | |
|                     "The client stil has to specify the scope to access the data."
 | |
|                 )
 | |
|             )
 | |
|         }
 | |
| 
 | |
| 
 | |
| class ScopeMappingForm(forms.ModelForm):
 | |
|     """Form to edit ScopeMappings"""
 | |
| 
 | |
|     def clean_expression(self):
 | |
|         """Test Syntax"""
 | |
|         expression = self.cleaned_data.get("expression")
 | |
|         evaluator = PropertyMappingEvaluator()
 | |
|         evaluator.validate(expression)
 | |
|         return expression
 | |
| 
 | |
|     class Meta:
 | |
| 
 | |
|         model = ScopeMapping
 | |
|         fields = ["name", "scope_name", "description", "expression"]
 | |
|         widgets = {
 | |
|             "name": forms.TextInput(),
 | |
|             "scope_name": forms.TextInput(),
 | |
|             "description": forms.TextInput(),
 | |
|             "expression": CodeMirrorWidget(mode="python"),
 | |
|         }
 | 
