import tldextract

from core_images import settings


def get_allowed_root_domain(request):
    """Returns allowed root domain to set cookie or None."""

    host = request.get_host()
    root_domain = tldextract.extract(host).registered_domain
    if root_domain and host in settings.ALLOWED_HOSTS:
        return root_domain

    return None