Current File : //proc/self/root/usr/lib/python2.7/site-packages/clcommon/cpapi/plugins/directadmin.py
#!/usr/bin/python
# -*- coding: utf-8 -*-
__cpname__ = 'DirectAdmin'
def detect():
return os.path.isfile('/usr/local/directadmin/directadmin') or \
os.path.isfile('/usr/local/directadmin/custombuild/build')
import os
import glob
import re
from clcommon.clconfpars import load as loadconfig
from clcommon.cpapi.cpapiexceptions import NoDBAccessData, CpApiTypeError
from clcommon.cpapi.plugins.universal import _dblogin_cplogin_pairs
from clcommon import ClPwd
DA_DB_CONF = '/usr/local/directadmin/conf/mysql.conf'
DA_USERS_PATH = '/usr/local/directadmin/data/users'
USER_CONF = 'user.conf'
RESELLERS_LIST = '/usr/local/directadmin/data/admin/reseller.list'
USER_PATTERN = re.compile(r'.+/(.+)/%s' % re.escape(USER_CONF))
USERCONF_PARAM_MAP = {
'dns': 'domain',
'package': 'package',
'mail': 'email',
'locale': 'language',
'cplogin': 'name',
'reseller': 'reseller'
}
def db_access(_conf_path=DA_DB_CONF):
access = dict()
try:
login_data = loadconfig(_conf_path)
access['login'] = login_data['user']
access['pass'] = login_data['passwd']
except IOError, err:
raise NoDBAccessData('Can not open file with data to database access; ' + str(err))
except KeyError:
raise NoDBAccessData('Can not get database access data from file %s' % (_conf_path,))
return access
def cpusers(_path=DA_USERS_PATH):
match_list = [USER_PATTERN.match(path) for path in glob.glob(os.path.join(_path, '*', USER_CONF))]
users_list = [match.group(1) for match in match_list if match]
return tuple(users_list)
def resellers(_resellers_path=RESELLERS_LIST):
stream = open(_resellers_path)
resellers_list = map(str.strip, stream.readlines())
stream.close()
return tuple(resellers_list)
def dblogin_cplogin_pairs(cplogin_lst=None, with_system_users=False):
access = db_access()
data = _dblogin_cplogin_pairs(cplogin_lst=cplogin_lst, access=access)
return data
def cpinfo(cpuser=None, keyls=('cplogin', 'package', 'mail', 'reseller', 'dns', 'locale'),
_path=DA_USERS_PATH):
returned = list()
if isinstance(cpuser, (str, unicode)):
cpusers_list = [cpuser]
elif isinstance(cpuser, (list, tuple)):
cpusers_list = tuple(cpuser)
elif cpuser is None:
match_list = [USER_PATTERN.match(path) for path in glob.iglob(os.path.join(_path, '*', USER_CONF))]
cpusers_list = [match.group(1) for match in match_list if match]
else:
raise CpApiTypeError(funcname='cpinfo', supportettypes='str|unicode|list|tuple',
received_type=type(cpuser).__name__)
# generate user reseller map
user_reseller_map = dict()
reseller_list_stream = open(RESELLERS_LIST)
for reseller_login in reseller_list_stream:
reseller_cpuser_list = open(os.path.join(os.path.join(_path, reseller_login.strip(), 'users.list')))
user_reseller_map.update(dict([(user_name, reseller_login) for user_name in reseller_cpuser_list
if user_name in cpusers_list]))
reseller_cpuser_list.close()
for cpuser in cpusers_list:
cpuser_data = loadconfig(os.path.join(_path, cpuser, 'user.conf'))
cpuser_data['reseller'] = user_reseller_map.get(cpuser) # insert reseller login
cpuser_data_lst = tuple([cpuser_data.get(USERCONF_PARAM_MAP.get(key)) for key in keyls])
if cpuser_data_lst not in returned:
returned.append(cpuser_data_lst)
return returned
def homedirs(_sysusers=None, _path=DA_USERS_PATH):
"""
Detects and returns list of folders contained the home dirs of users of the DirectAdmin
:param str|None _sysusers: for testing
:param str|None _path: for testing
:return: list of folders, which are parent of home dirs of users of the panel
"""
homedirs = []
clpwd = ClPwd()
users_dict = clpwd.get_user_dict()
# for testing only
if isinstance(_sysusers, (list, tuple)):
class pw(object):
def __init__(self, name, dir):
self.pw_name = name
self.pw_dir = dir
users_dict = {}
for (name,dir) in _sysusers:
users_dict[name] = pw(name, dir)
for user_name in users_dict:
conf_file = _path + '/' + user_name + '/user.conf'
if os.path.exists(conf_file):
homedir = os.path.dirname(users_dict[user_name].pw_dir)
if homedir not in homedirs:
homedirs.append(homedir)
return homedirs
Mini Shell