Add type hint

This wasn't found due to no typing information, so lets add some
This commit is contained in:
zandercymatics 2024-09-10 13:24:24 -06:00
parent 0eda760947
commit 4dfa2f7fe5
No known key found for this signature in database
GPG key ID: FF4636ABEC9682B7

View file

@ -1,7 +1,7 @@
import logging import logging
from django.http import JsonResponse from django.http import JsonResponse
from django.core.paginator import Paginator from django.core.paginator import Paginator
from registrar.models import UserDomainRole, Domain, DomainInformation from registrar.models import UserDomainRole, Domain, DomainInformation, User
from django.contrib.auth.decorators import login_required from django.contrib.auth.decorators import login_required
from django.urls import reverse from django.urls import reverse
from django.db.models import Q from django.db.models import Q
@ -50,7 +50,8 @@ def get_domain_ids_from_request(request):
""" """
portfolio = request.GET.get("portfolio") portfolio = request.GET.get("portfolio")
if portfolio: if portfolio:
if request.user.is_org_user(request) and request.user.has_view_all_domains_portfolio_permission(portfolio): current_user: User = request.user
if current_user.is_org_user(request) and current_user.has_view_all_domains_portfolio_permission(portfolio):
domain_infos = DomainInformation.objects.filter(portfolio=portfolio) domain_infos = DomainInformation.objects.filter(portfolio=portfolio)
return domain_infos.values_list("domain_id", flat=True) return domain_infos.values_list("domain_id", flat=True)
else: else: