Source code for pyad.addomain

from .adcontainer import ADContainer
from .adobject import ADObject


class ADDomain(ADContainer):
    def get_default_upn(self):
        """Returns the default userPrincipalName for the domain."""

        self._ldap_adsi_obj.GetInfoEx(
            [
                "canonicalName",
            ],
            0,
        )
        return self._ldap_adsi_obj.get("canonicalName").rstrip("/")


def __get_domain(self):
    """Returns the domain to which the object belongs."""

    if self._domain_pyad_obj is None:
        domain_path = "dc=" + self.dn.lower().split("dc=", 1)[1]
        self._domain_pyad_obj = ADDomain.from_dn(
            domain_path, options=self._make_options()
        )
    return self._domain_pyad_obj


ADObject.get_domain = __get_domain

ADObject._py_ad_object_mappings["domain"] = ADDomain