takahe/users/shortcuts.py

48 lines
1.6 KiB
Python
Raw Normal View History

from django.http import Http404
2022-11-05 20:17:27 +00:00
from users.models import Domain, Identity
2022-11-05 20:17:27 +00:00
def by_handle_or_404(request, handle, local=True, fetch=False) -> Identity:
2022-11-05 20:17:27 +00:00
"""
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 "host" not in request.headers:
raise Http404("No hostname available")
username = handle
domain_instance = Domain.get_domain(request.headers["host"])
if domain_instance is None:
raise Http404("No matching domains found")
domain = domain_instance.domain
else:
username, domain = handle.split("@", 1)
# Resolve the domain to the display domain
domain_instance = Domain.get_domain(domain)
if domain_instance is None:
domain_instance = Domain.get_remote_domain(domain)
domain = domain_instance.domain
identity = Identity.by_username_and_domain(
username,
domain_instance,
local=local,
fetch=fetch,
)
if identity is None:
raise Http404(f"No identity for handle {handle}")
2022-12-17 02:42:48 +00:00
if identity.blocked:
raise Http404("Blocked user")
return identity
def by_handle_for_user_or_404(request, handle):
"""
Retrieves an identity the local user can control via their handle, or
raises a 404.
"""
identity = by_handle_or_404(request, handle, local=True, fetch=False)
if not identity.users.filter(id=request.user.id).exists():
raise Http404("Current user does not own identity")
return identity