kopia lustrzana https://github.com/jointakahe/takahe
34 wiersze
1.0 KiB
Python
34 wiersze
1.0 KiB
Python
from django.http import Http404
|
|
from django.shortcuts import get_object_or_404
|
|
|
|
from users.models import Domain, Identity
|
|
|
|
|
|
def by_handle_or_404(request, handle, local=True):
|
|
"""
|
|
Retrieves an Identity by its long or short handle.
|
|
Domain-sensitive, so it will understand short handles on alternate domains.
|
|
"""
|
|
if "@" not in handle:
|
|
if "HTTP_HOST" not in request.META:
|
|
raise Http404("No hostname available")
|
|
username = handle
|
|
domain_instance = Domain.get_local_domain(request.META["HTTP_HOST"])
|
|
if domain_instance is None:
|
|
raise Http404("No matching domains found")
|
|
domain = domain_instance.domain
|
|
else:
|
|
username, domain = handle.split("@", 1)
|
|
if local:
|
|
return get_object_or_404(
|
|
Identity.objects.filter(local=True),
|
|
username=username,
|
|
domain_id=domain,
|
|
)
|
|
else:
|
|
return get_object_or_404(
|
|
Identity,
|
|
username=username,
|
|
domain_id=domain,
|
|
)
|