# -*- coding: utf-8 -*-
# domainlib.py - library for miscelaneous domain operations
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2019 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT
from clcommon import cpapi
from clcommon.clexception import FormattedException
from clcommon.cpapi import NotSupported
class DomainException(FormattedException):
pass
class Domain:
def __init__(self):
self._userdomain_cache = None # type: dict[str, str]
def get_user_domain(self, username):
"""
Retrive main domain for supplied user
:param username: User name
:return: Domain name or N/A
"""
if self._userdomain_cache is None:
try:
self._userdomain_cache = dict(cpapi.cpinfo(keyls=('cplogin', 'dns')))
except NotSupported:
self._userdomain_cache = {}
return self._userdomain_cache.get(username, 'N/A')
def get_user_name_by_domain(self, domain_name):
"""
Retrive username by domain
:param domain_name: Domain name
:return: Username
"""
try:
username = cpapi.domain_owner(domain_name)
if username:
return username
except (IOError, cpapi.cpapiexceptions.NotSupported) as e:
raise DomainException(str(e)) from e
raise DomainException({'message': "There is no domain '%(domain)s'",
'context': {'domain': domain_name}})
|