mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-04 05:53:43 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			74 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			74 lines
		
	
	
		
			2.7 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from django.core.exceptions import ValidationError
 | 
						|
from django.http import HttpRequest, HttpResponse
 | 
						|
from django.utils.translation import gettext as _
 | 
						|
 | 
						|
from zerver.actions.realm_domains import (
 | 
						|
    do_add_realm_domain,
 | 
						|
    do_change_realm_domain,
 | 
						|
    do_remove_realm_domain,
 | 
						|
)
 | 
						|
from zerver.decorator import require_realm_admin
 | 
						|
from zerver.lib.domains import validate_domain
 | 
						|
from zerver.lib.exceptions import JsonableError
 | 
						|
from zerver.lib.request import REQ, has_request_variables
 | 
						|
from zerver.lib.response import json_success
 | 
						|
from zerver.lib.validator import check_bool
 | 
						|
from zerver.models import RealmDomain, UserProfile, get_realm_domains
 | 
						|
 | 
						|
 | 
						|
def list_realm_domains(request: HttpRequest, user_profile: UserProfile) -> HttpResponse:
 | 
						|
    domains = get_realm_domains(user_profile.realm)
 | 
						|
    return json_success(request, data={"domains": domains})
 | 
						|
 | 
						|
 | 
						|
@require_realm_admin
 | 
						|
@has_request_variables
 | 
						|
def create_realm_domain(
 | 
						|
    request: HttpRequest,
 | 
						|
    user_profile: UserProfile,
 | 
						|
    domain: str = REQ(),
 | 
						|
    allow_subdomains: bool = REQ(json_validator=check_bool),
 | 
						|
) -> HttpResponse:
 | 
						|
    domain = domain.strip().lower()
 | 
						|
    try:
 | 
						|
        validate_domain(domain)
 | 
						|
    except ValidationError as e:
 | 
						|
        raise JsonableError(_("Invalid domain: {}").format(e.messages[0]))
 | 
						|
    if RealmDomain.objects.filter(realm=user_profile.realm, domain=domain).exists():
 | 
						|
        raise JsonableError(
 | 
						|
            _("The domain {domain} is already a part of your organization.").format(domain=domain)
 | 
						|
        )
 | 
						|
    realm_domain = do_add_realm_domain(
 | 
						|
        user_profile.realm, domain, allow_subdomains, acting_user=user_profile
 | 
						|
    )
 | 
						|
    return json_success(request, data={"new_domain": [realm_domain.id, realm_domain.domain]})
 | 
						|
 | 
						|
 | 
						|
@require_realm_admin
 | 
						|
@has_request_variables
 | 
						|
def patch_realm_domain(
 | 
						|
    request: HttpRequest,
 | 
						|
    user_profile: UserProfile,
 | 
						|
    domain: str,
 | 
						|
    allow_subdomains: bool = REQ(json_validator=check_bool),
 | 
						|
) -> HttpResponse:
 | 
						|
    try:
 | 
						|
        realm_domain = RealmDomain.objects.get(realm=user_profile.realm, domain=domain)
 | 
						|
        do_change_realm_domain(realm_domain, allow_subdomains, acting_user=user_profile)
 | 
						|
    except RealmDomain.DoesNotExist:
 | 
						|
        raise JsonableError(_("No entry found for domain {domain}.").format(domain=domain))
 | 
						|
    return json_success(request)
 | 
						|
 | 
						|
 | 
						|
@require_realm_admin
 | 
						|
@has_request_variables
 | 
						|
def delete_realm_domain(
 | 
						|
    request: HttpRequest, user_profile: UserProfile, domain: str
 | 
						|
) -> HttpResponse:
 | 
						|
    try:
 | 
						|
        realm_domain = RealmDomain.objects.get(realm=user_profile.realm, domain=domain)
 | 
						|
        do_remove_realm_domain(realm_domain, acting_user=user_profile)
 | 
						|
    except RealmDomain.DoesNotExist:
 | 
						|
        raise JsonableError(_("No entry found for domain {domain}.").format(domain=domain))
 | 
						|
    return json_success(request)
 |