mirror of
https://github.com/YunoHost/yunohost.git
synced 2024-09-03 20:06:10 +02:00
Merge pull request #1089 from YunoHost/enh-csv
[enh] Import users with a CSV
This commit is contained in:
commit
94b1bdebb3
6 changed files with 505 additions and 89 deletions
|
@ -59,7 +59,7 @@ user:
|
||||||
api: GET /users
|
api: GET /users
|
||||||
arguments:
|
arguments:
|
||||||
--fields:
|
--fields:
|
||||||
help: fields to fetch
|
help: fields to fetch (username, fullname, mail, mail-alias, mail-forward, mailbox-quota, groups, shell, home-path)
|
||||||
nargs: "+"
|
nargs: "+"
|
||||||
|
|
||||||
### user_create()
|
### user_create()
|
||||||
|
@ -199,6 +199,28 @@ user:
|
||||||
username:
|
username:
|
||||||
help: Username or email to get information
|
help: Username or email to get information
|
||||||
|
|
||||||
|
### user_export()
|
||||||
|
export:
|
||||||
|
action_help: Export users into CSV
|
||||||
|
api: GET /users/export
|
||||||
|
|
||||||
|
### user_import()
|
||||||
|
import:
|
||||||
|
action_help: Import several users from CSV
|
||||||
|
api: POST /users/import
|
||||||
|
arguments:
|
||||||
|
csvfile:
|
||||||
|
help: "CSV file with columns username, firstname, lastname, password, mail, mailbox-quota, mail-alias, mail-forward, groups (separated by coma)"
|
||||||
|
type: open
|
||||||
|
-u:
|
||||||
|
full: --update
|
||||||
|
help: Update all existing users contained in the CSV file (by default existing users are ignored)
|
||||||
|
action: store_true
|
||||||
|
-d:
|
||||||
|
full: --delete
|
||||||
|
help: Delete all existing users that are not contained in the CSV file (by default existing users are kept)
|
||||||
|
action: store_true
|
||||||
|
|
||||||
subcategories:
|
subcategories:
|
||||||
group:
|
group:
|
||||||
subcategory_help: Manage user groups
|
subcategory_help: Manage user groups
|
||||||
|
|
|
@ -33,6 +33,7 @@ olcAuthzPolicy: none
|
||||||
olcConcurrency: 0
|
olcConcurrency: 0
|
||||||
olcConnMaxPending: 100
|
olcConnMaxPending: 100
|
||||||
olcConnMaxPendingAuth: 1000
|
olcConnMaxPendingAuth: 1000
|
||||||
|
olcSizeLimit: 50000
|
||||||
olcIdleTimeout: 0
|
olcIdleTimeout: 0
|
||||||
olcIndexSubstrIfMaxLen: 4
|
olcIndexSubstrIfMaxLen: 4
|
||||||
olcIndexSubstrIfMinLen: 2
|
olcIndexSubstrIfMinLen: 2
|
||||||
|
@ -188,7 +189,7 @@ olcDbIndex: memberUid eq
|
||||||
olcDbIndex: uniqueMember eq
|
olcDbIndex: uniqueMember eq
|
||||||
olcDbIndex: virtualdomain eq
|
olcDbIndex: virtualdomain eq
|
||||||
olcDbIndex: permission eq
|
olcDbIndex: permission eq
|
||||||
olcDbMaxSize: 10485760
|
olcDbMaxSize: 104857600
|
||||||
structuralObjectClass: olcMdbConfig
|
structuralObjectClass: olcMdbConfig
|
||||||
|
|
||||||
#
|
#
|
||||||
|
|
|
@ -416,6 +416,7 @@
|
||||||
"log_regen_conf": "Regenerate system configurations '{}'",
|
"log_regen_conf": "Regenerate system configurations '{}'",
|
||||||
"log_user_create": "Add '{}' user",
|
"log_user_create": "Add '{}' user",
|
||||||
"log_user_delete": "Delete '{}' user",
|
"log_user_delete": "Delete '{}' user",
|
||||||
|
"log_user_import": "Import users",
|
||||||
"log_user_group_create": "Create '{}' group",
|
"log_user_group_create": "Create '{}' group",
|
||||||
"log_user_group_delete": "Delete '{}' group",
|
"log_user_group_delete": "Delete '{}' group",
|
||||||
"log_user_group_update": "Update '{}' group",
|
"log_user_group_update": "Update '{}' group",
|
||||||
|
@ -641,10 +642,17 @@
|
||||||
"user_creation_failed": "Could not create user {user}: {error}",
|
"user_creation_failed": "Could not create user {user}: {error}",
|
||||||
"user_deleted": "User deleted",
|
"user_deleted": "User deleted",
|
||||||
"user_deletion_failed": "Could not delete user {user}: {error}",
|
"user_deletion_failed": "Could not delete user {user}: {error}",
|
||||||
"user_home_creation_failed": "Could not create 'home' folder for user",
|
"user_home_creation_failed": "Could not create home folder '{home}' for user",
|
||||||
"user_unknown": "Unknown user: {user}",
|
"user_unknown": "Unknown user: {user}",
|
||||||
"user_update_failed": "Could not update user {user}: {error}",
|
"user_update_failed": "Could not update user {user}: {error}",
|
||||||
"user_updated": "User info changed",
|
"user_updated": "User info changed",
|
||||||
|
"user_import_bad_line": "Incorrect line {line}: {details}",
|
||||||
|
"user_import_bad_file": "Your CSV file is not correctly formatted it will be ignored to avoid potential data loss",
|
||||||
|
"user_import_missing_columns": "The following columns are missing: {columns}",
|
||||||
|
"user_import_partial_failed": "The users import operation partially failed",
|
||||||
|
"user_import_failed": "The users import operation completely failed",
|
||||||
|
"user_import_nothing_to_do": "No user needs to be imported",
|
||||||
|
"user_import_success": "Users successfully imported",
|
||||||
"yunohost_already_installed": "YunoHost is already installed",
|
"yunohost_already_installed": "YunoHost is already installed",
|
||||||
"yunohost_configured": "YunoHost is now configured",
|
"yunohost_configured": "YunoHost is now configured",
|
||||||
"yunohost_installing": "Installing YunoHost...",
|
"yunohost_installing": "Installing YunoHost...",
|
||||||
|
|
|
@ -32,6 +32,7 @@ import psutil
|
||||||
|
|
||||||
from datetime import datetime, timedelta
|
from datetime import datetime, timedelta
|
||||||
from logging import FileHandler, getLogger, Formatter
|
from logging import FileHandler, getLogger, Formatter
|
||||||
|
from io import IOBase
|
||||||
|
|
||||||
from moulinette import m18n, Moulinette
|
from moulinette import m18n, Moulinette
|
||||||
from moulinette.core import MoulinetteError
|
from moulinette.core import MoulinetteError
|
||||||
|
@ -370,6 +371,18 @@ def is_unit_operation(
|
||||||
for field in exclude:
|
for field in exclude:
|
||||||
if field in context:
|
if field in context:
|
||||||
context.pop(field, None)
|
context.pop(field, None)
|
||||||
|
|
||||||
|
# Context is made from args given to main function by argparse
|
||||||
|
# This context will be added in extra parameters in yml file, so this context should
|
||||||
|
# be serializable and short enough (it will be displayed in webadmin)
|
||||||
|
# Argparse can provide some File or Stream, so here we display the filename or
|
||||||
|
# the IOBase, if we have no name.
|
||||||
|
for field, value in context.items():
|
||||||
|
if isinstance(value, IOBase):
|
||||||
|
try:
|
||||||
|
context[field] = value.name
|
||||||
|
except:
|
||||||
|
context[field] = 'IOBase'
|
||||||
operation_logger = OperationLogger(op_key, related_to, args=context)
|
operation_logger = OperationLogger(op_key, related_to, args=context)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
|
|
@ -8,6 +8,10 @@ from yunohost.user import (
|
||||||
user_create,
|
user_create,
|
||||||
user_delete,
|
user_delete,
|
||||||
user_update,
|
user_update,
|
||||||
|
user_import,
|
||||||
|
user_export,
|
||||||
|
FIELDS_FOR_IMPORT,
|
||||||
|
FIRST_ALIASES,
|
||||||
user_group_list,
|
user_group_list,
|
||||||
user_group_create,
|
user_group_create,
|
||||||
user_group_delete,
|
user_group_delete,
|
||||||
|
@ -22,7 +26,7 @@ maindomain = ""
|
||||||
|
|
||||||
def clean_user_groups():
|
def clean_user_groups():
|
||||||
for u in user_list()["users"]:
|
for u in user_list()["users"]:
|
||||||
user_delete(u)
|
user_delete(u, purge=True)
|
||||||
|
|
||||||
for g in user_group_list()["groups"]:
|
for g in user_group_list()["groups"]:
|
||||||
if g not in ["all_users", "visitors"]:
|
if g not in ["all_users", "visitors"]:
|
||||||
|
@ -110,6 +114,65 @@ def test_del_user(mocker):
|
||||||
assert "alice" not in group_res["all_users"]["members"]
|
assert "alice" not in group_res["all_users"]["members"]
|
||||||
|
|
||||||
|
|
||||||
|
def test_import_user(mocker):
|
||||||
|
import csv
|
||||||
|
from io import StringIO
|
||||||
|
fieldnames = [u'username', u'firstname', u'lastname', u'password',
|
||||||
|
u'mailbox-quota', u'mail', u'mail-alias', u'mail-forward',
|
||||||
|
u'groups']
|
||||||
|
with StringIO() as csv_io:
|
||||||
|
writer = csv.DictWriter(csv_io, fieldnames, delimiter=';',
|
||||||
|
quotechar='"')
|
||||||
|
writer.writeheader()
|
||||||
|
writer.writerow({
|
||||||
|
'username': "albert",
|
||||||
|
'firstname': "Albert",
|
||||||
|
'lastname': "Good",
|
||||||
|
'password': "",
|
||||||
|
'mailbox-quota': "1G",
|
||||||
|
'mail': "albert@" + maindomain,
|
||||||
|
'mail-alias': "albert2@" + maindomain,
|
||||||
|
'mail-forward': "albert@example.com",
|
||||||
|
'groups': "dev",
|
||||||
|
})
|
||||||
|
writer.writerow({
|
||||||
|
'username': "alice",
|
||||||
|
'firstname': "Alice",
|
||||||
|
'lastname': "White",
|
||||||
|
'password': "",
|
||||||
|
'mailbox-quota': "1G",
|
||||||
|
'mail': "alice@" + maindomain,
|
||||||
|
'mail-alias': "alice1@" + maindomain + ",alice2@" + maindomain,
|
||||||
|
'mail-forward': "",
|
||||||
|
'groups': "apps",
|
||||||
|
})
|
||||||
|
csv_io.seek(0)
|
||||||
|
with message(mocker, "user_import_success"):
|
||||||
|
user_import(csv_io, update=True, delete=True)
|
||||||
|
|
||||||
|
group_res = user_group_list()['groups']
|
||||||
|
user_res = user_list(list(FIELDS_FOR_IMPORT.keys()))['users']
|
||||||
|
assert "albert" in user_res
|
||||||
|
assert "alice" in user_res
|
||||||
|
assert "bob" not in user_res
|
||||||
|
assert len(user_res['alice']['mail-alias']) == 2
|
||||||
|
assert "albert" in group_res['dev']['members']
|
||||||
|
assert "alice" in group_res['apps']['members']
|
||||||
|
assert "alice" not in group_res['dev']['members']
|
||||||
|
|
||||||
|
|
||||||
|
def test_export_user(mocker):
|
||||||
|
result = user_export()
|
||||||
|
aliases = ','.join([alias + maindomain for alias in FIRST_ALIASES])
|
||||||
|
should_be = (
|
||||||
|
"username;firstname;lastname;password;mail;mail-alias;mail-forward;mailbox-quota;groups\r\n"
|
||||||
|
f"alice;Alice;White;;alice@{maindomain};{aliases};;0;dev\r\n"
|
||||||
|
f"bob;Bob;Snow;;bob@{maindomain};;;0;apps\r\n"
|
||||||
|
f"jack;Jack;Black;;jack@{maindomain};;;0;"
|
||||||
|
)
|
||||||
|
assert result == should_be
|
||||||
|
|
||||||
|
|
||||||
def test_create_group(mocker):
|
def test_create_group(mocker):
|
||||||
|
|
||||||
with message(mocker, "group_created", group="adminsys"):
|
with message(mocker, "group_created", group="adminsys"):
|
||||||
|
|
|
@ -43,32 +43,67 @@ from yunohost.log import is_unit_operation
|
||||||
|
|
||||||
logger = getActionLogger("yunohost.user")
|
logger = getActionLogger("yunohost.user")
|
||||||
|
|
||||||
|
FIELDS_FOR_IMPORT = {
|
||||||
|
'username': r'^[a-z0-9_]+$',
|
||||||
|
'firstname': r'^([^\W\d_]{1,30}[ ,.\'-]{0,3})+$',
|
||||||
|
'lastname': r'^([^\W\d_]{1,30}[ ,.\'-]{0,3})+$',
|
||||||
|
'password': r'^|(.{3,})$',
|
||||||
|
'mail': r'^([\w.-]+@([^\W_A-Z]+([-]*[^\W_A-Z]+)*\.)+((xn--)?[^\W_]{2,}))$',
|
||||||
|
'mail-alias': r'^|([\w.-]+@([^\W_A-Z]+([-]*[^\W_A-Z]+)*\.)+((xn--)?[^\W_]{2,}),?)+$',
|
||||||
|
'mail-forward': r'^|([\w\+.-]+@([^\W_A-Z]+([-]*[^\W_A-Z]+)*\.)+((xn--)?[^\W_]{2,}),?)+$',
|
||||||
|
'mailbox-quota': r'^(\d+[bkMGT])|0|$',
|
||||||
|
'groups': r'^|([a-z0-9_]+(,?[a-z0-9_]+)*)$'
|
||||||
|
}
|
||||||
|
|
||||||
|
FIRST_ALIASES = ['root@', 'admin@', 'webmaster@', 'postmaster@', 'abuse@']
|
||||||
|
|
||||||
|
|
||||||
def user_list(fields=None):
|
def user_list(fields=None):
|
||||||
|
|
||||||
from yunohost.utils.ldap import _get_ldap_interface
|
from yunohost.utils.ldap import _get_ldap_interface
|
||||||
|
|
||||||
user_attrs = {
|
ldap_attrs = {
|
||||||
"uid": "username",
|
'username': 'uid',
|
||||||
"cn": "fullname",
|
'password': '', # We can't request password in ldap
|
||||||
"mail": "mail",
|
'fullname': 'cn',
|
||||||
"maildrop": "mail-forward",
|
'firstname': 'givenName',
|
||||||
"homeDirectory": "home_path",
|
'lastname': 'sn',
|
||||||
"mailuserquota": "mailbox-quota",
|
'mail': 'mail',
|
||||||
|
'mail-alias': 'mail',
|
||||||
|
'mail-forward': 'maildrop',
|
||||||
|
'mailbox-quota': 'mailuserquota',
|
||||||
|
'groups': 'memberOf',
|
||||||
|
'shell': 'loginShell',
|
||||||
|
'home-path': 'homeDirectory'
|
||||||
}
|
}
|
||||||
|
|
||||||
attrs = ["uid"]
|
def display_default(values, _):
|
||||||
|
return values[0] if len(values) == 1 else values
|
||||||
|
|
||||||
|
display = {
|
||||||
|
'password': lambda values, user: '',
|
||||||
|
'mail': lambda values, user: display_default(values[:1], user),
|
||||||
|
'mail-alias': lambda values, _: values[1:],
|
||||||
|
'mail-forward': lambda values, user: [forward for forward in values if forward != user['uid'][0]],
|
||||||
|
'groups': lambda values, user: [
|
||||||
|
group[3:].split(',')[0]
|
||||||
|
for group in values
|
||||||
|
if not group.startswith('cn=all_users,') and
|
||||||
|
not group.startswith('cn=' + user['uid'][0] + ',')],
|
||||||
|
'shell': lambda values, _: len(values) > 0 and values[0].strip() == "/bin/false"
|
||||||
|
}
|
||||||
|
|
||||||
|
attrs = set(['uid'])
|
||||||
users = {}
|
users = {}
|
||||||
|
|
||||||
if fields:
|
if not fields:
|
||||||
keys = user_attrs.keys()
|
fields = ['username', 'fullname', 'mail', 'mailbox-quota']
|
||||||
for attr in fields:
|
|
||||||
if attr in keys:
|
for field in fields:
|
||||||
attrs.append(attr)
|
if field in ldap_attrs:
|
||||||
else:
|
attrs.add(ldap_attrs[field])
|
||||||
raise YunohostError("field_invalid", attr)
|
else:
|
||||||
else:
|
raise YunohostError('field_invalid', field)
|
||||||
attrs = ["uid", "cn", "mail", "mailuserquota"]
|
|
||||||
|
|
||||||
ldap = _get_ldap_interface()
|
ldap = _get_ldap_interface()
|
||||||
result = ldap.search(
|
result = ldap.search(
|
||||||
|
@ -79,12 +114,13 @@ def user_list(fields=None):
|
||||||
|
|
||||||
for user in result:
|
for user in result:
|
||||||
entry = {}
|
entry = {}
|
||||||
for attr, values in user.items():
|
for field in fields:
|
||||||
if values:
|
values = []
|
||||||
entry[user_attrs[attr]] = values[0]
|
if ldap_attrs[field] in user:
|
||||||
|
values = user[ldap_attrs[field]]
|
||||||
|
entry[field] = display.get(field, display_default)(values, user)
|
||||||
|
|
||||||
uid = entry[user_attrs["uid"]]
|
users[user['uid'][0]] = entry
|
||||||
users[uid] = entry
|
|
||||||
|
|
||||||
return {"users": users}
|
return {"users": users}
|
||||||
|
|
||||||
|
@ -99,6 +135,7 @@ def user_create(
|
||||||
password,
|
password,
|
||||||
mailbox_quota="0",
|
mailbox_quota="0",
|
||||||
mail=None,
|
mail=None,
|
||||||
|
from_import=False
|
||||||
):
|
):
|
||||||
|
|
||||||
from yunohost.domain import domain_list, _get_maindomain
|
from yunohost.domain import domain_list, _get_maindomain
|
||||||
|
@ -156,18 +193,13 @@ def user_create(
|
||||||
raise YunohostValidationError("system_username_exists")
|
raise YunohostValidationError("system_username_exists")
|
||||||
|
|
||||||
main_domain = _get_maindomain()
|
main_domain = _get_maindomain()
|
||||||
aliases = [
|
aliases = [alias + main_domain for alias in FIRST_ALIASES]
|
||||||
"root@" + main_domain,
|
|
||||||
"admin@" + main_domain,
|
|
||||||
"webmaster@" + main_domain,
|
|
||||||
"postmaster@" + main_domain,
|
|
||||||
"abuse@" + main_domain,
|
|
||||||
]
|
|
||||||
|
|
||||||
if mail in aliases:
|
if mail in aliases:
|
||||||
raise YunohostValidationError("mail_unavailable")
|
raise YunohostValidationError("mail_unavailable")
|
||||||
|
|
||||||
operation_logger.start()
|
if not from_import:
|
||||||
|
operation_logger.start()
|
||||||
|
|
||||||
# Get random UID/GID
|
# Get random UID/GID
|
||||||
all_uid = {str(x.pw_uid) for x in pwd.getpwall()}
|
all_uid = {str(x.pw_uid) for x in pwd.getpwall()}
|
||||||
|
@ -221,8 +253,10 @@ def user_create(
|
||||||
# Attempt to create user home folder
|
# Attempt to create user home folder
|
||||||
subprocess.check_call(["mkhomedir_helper", username])
|
subprocess.check_call(["mkhomedir_helper", username])
|
||||||
except subprocess.CalledProcessError:
|
except subprocess.CalledProcessError:
|
||||||
if not os.path.isdir("/home/{0}".format(username)):
|
home = f'/home/{username}'
|
||||||
logger.warning(m18n.n("user_home_creation_failed"), exc_info=1)
|
if not os.path.isdir(home):
|
||||||
|
logger.warning(m18n.n('user_home_creation_failed', home=home),
|
||||||
|
exc_info=1)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
subprocess.check_call(
|
subprocess.check_call(
|
||||||
|
@ -247,13 +281,14 @@ def user_create(
|
||||||
hook_callback("post_user_create", args=[username, mail], env=env_dict)
|
hook_callback("post_user_create", args=[username, mail], env=env_dict)
|
||||||
|
|
||||||
# TODO: Send a welcome mail to user
|
# TODO: Send a welcome mail to user
|
||||||
logger.success(m18n.n("user_created"))
|
if not from_import:
|
||||||
|
logger.success(m18n.n('user_created'))
|
||||||
|
|
||||||
return {"fullname": fullname, "username": username, "mail": mail}
|
return {"fullname": fullname, "username": username, "mail": mail}
|
||||||
|
|
||||||
|
|
||||||
@is_unit_operation([("username", "user")])
|
@is_unit_operation([('username', 'user')])
|
||||||
def user_delete(operation_logger, username, purge=False):
|
def user_delete(operation_logger, username, purge=False, from_import=False):
|
||||||
"""
|
"""
|
||||||
Delete user
|
Delete user
|
||||||
|
|
||||||
|
@ -268,7 +303,8 @@ def user_delete(operation_logger, username, purge=False):
|
||||||
if username not in user_list()["users"]:
|
if username not in user_list()["users"]:
|
||||||
raise YunohostValidationError("user_unknown", user=username)
|
raise YunohostValidationError("user_unknown", user=username)
|
||||||
|
|
||||||
operation_logger.start()
|
if not from_import:
|
||||||
|
operation_logger.start()
|
||||||
|
|
||||||
user_group_update("all_users", remove=username, force=True, sync_perm=False)
|
user_group_update("all_users", remove=username, force=True, sync_perm=False)
|
||||||
for group, infos in user_group_list()["groups"].items():
|
for group, infos in user_group_list()["groups"].items():
|
||||||
|
@ -295,13 +331,13 @@ def user_delete(operation_logger, username, purge=False):
|
||||||
subprocess.call(["nscd", "-i", "passwd"])
|
subprocess.call(["nscd", "-i", "passwd"])
|
||||||
|
|
||||||
if purge:
|
if purge:
|
||||||
subprocess.call(["rm", "-rf", "/home/{0}".format(username)])
|
subprocess.call(['rm', '-rf', '/home/{0}'.format(username)])
|
||||||
subprocess.call(["rm", "-rf", "/var/mail/{0}".format(username)])
|
subprocess.call(['rm', '-rf', '/var/mail/{0}'.format(username)])
|
||||||
|
|
||||||
hook_callback("post_user_delete", args=[username, purge])
|
hook_callback('post_user_delete', args=[username, purge])
|
||||||
|
|
||||||
logger.success(m18n.n("user_deleted"))
|
|
||||||
|
|
||||||
|
if not from_import:
|
||||||
|
logger.success(m18n.n('user_deleted'))
|
||||||
|
|
||||||
@is_unit_operation([("username", "user")], exclude=["change_password"])
|
@is_unit_operation([("username", "user")], exclude=["change_password"])
|
||||||
def user_update(
|
def user_update(
|
||||||
|
@ -316,6 +352,7 @@ def user_update(
|
||||||
add_mailalias=None,
|
add_mailalias=None,
|
||||||
remove_mailalias=None,
|
remove_mailalias=None,
|
||||||
mailbox_quota=None,
|
mailbox_quota=None,
|
||||||
|
from_import=False
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Update user informations
|
Update user informations
|
||||||
|
@ -375,7 +412,7 @@ def user_update(
|
||||||
]
|
]
|
||||||
|
|
||||||
# change_password is None if user_update is not called to change the password
|
# change_password is None if user_update is not called to change the password
|
||||||
if change_password is not None:
|
if change_password is not None and change_password != '':
|
||||||
# when in the cli interface if the option to change the password is called
|
# when in the cli interface if the option to change the password is called
|
||||||
# without a specified value, change_password will be set to the const 0.
|
# without a specified value, change_password will be set to the const 0.
|
||||||
# In this case we prompt for the new password.
|
# In this case we prompt for the new password.
|
||||||
|
@ -389,39 +426,41 @@ def user_update(
|
||||||
|
|
||||||
if mail:
|
if mail:
|
||||||
main_domain = _get_maindomain()
|
main_domain = _get_maindomain()
|
||||||
aliases = [
|
aliases = [alias + main_domain for alias in FIRST_ALIASES]
|
||||||
"root@" + main_domain,
|
|
||||||
"admin@" + main_domain,
|
# If the requested mail address is already as main address or as an alias by this user
|
||||||
"webmaster@" + main_domain,
|
if mail in user['mail']:
|
||||||
"postmaster@" + main_domain,
|
user['mail'].remove(mail)
|
||||||
]
|
# Othewise, check that this mail address is not already used by this user
|
||||||
try:
|
else:
|
||||||
ldap.validate_uniqueness({"mail": mail})
|
try:
|
||||||
except Exception as e:
|
ldap.validate_uniqueness({'mail': mail})
|
||||||
raise YunohostValidationError("user_update_failed", user=username, error=e)
|
except Exception as e:
|
||||||
if mail[mail.find("@") + 1 :] not in domains:
|
raise YunohostError('user_update_failed', user=username, error=e)
|
||||||
raise YunohostValidationError(
|
if mail[mail.find('@') + 1:] not in domains:
|
||||||
"mail_domain_unknown", domain=mail[mail.find("@") + 1 :]
|
raise YunohostError('mail_domain_unknown', domain=mail[mail.find('@') + 1:])
|
||||||
)
|
|
||||||
if mail in aliases:
|
if mail in aliases:
|
||||||
raise YunohostValidationError("mail_unavailable")
|
raise YunohostValidationError("mail_unavailable")
|
||||||
|
|
||||||
del user["mail"][0]
|
new_attr_dict['mail'] = [mail] + user['mail'][1:]
|
||||||
new_attr_dict["mail"] = [mail] + user["mail"]
|
|
||||||
|
|
||||||
if add_mailalias:
|
if add_mailalias:
|
||||||
if not isinstance(add_mailalias, list):
|
if not isinstance(add_mailalias, list):
|
||||||
add_mailalias = [add_mailalias]
|
add_mailalias = [add_mailalias]
|
||||||
for mail in add_mailalias:
|
for mail in add_mailalias:
|
||||||
try:
|
# (c.f. similar stuff as before)
|
||||||
ldap.validate_uniqueness({"mail": mail})
|
if mail in user["mail"]:
|
||||||
except Exception as e:
|
user["mail"].remove(mail)
|
||||||
raise YunohostValidationError(
|
else:
|
||||||
"user_update_failed", user=username, error=e
|
try:
|
||||||
)
|
ldap.validate_uniqueness({"mail": mail})
|
||||||
if mail[mail.find("@") + 1 :] not in domains:
|
except Exception as e:
|
||||||
raise YunohostValidationError(
|
raise YunohostError(
|
||||||
"mail_domain_unknown", domain=mail[mail.find("@") + 1 :]
|
"user_update_failed", user=username, error=e
|
||||||
|
)
|
||||||
|
if mail[mail.find("@") + 1:] not in domains:
|
||||||
|
raise YunohostError(
|
||||||
|
"mail_domain_unknown", domain=mail[mail.find("@") + 1:]
|
||||||
)
|
)
|
||||||
user["mail"].append(mail)
|
user["mail"].append(mail)
|
||||||
new_attr_dict["mail"] = user["mail"]
|
new_attr_dict["mail"] = user["mail"]
|
||||||
|
@ -465,7 +504,8 @@ def user_update(
|
||||||
new_attr_dict["mailuserquota"] = [mailbox_quota]
|
new_attr_dict["mailuserquota"] = [mailbox_quota]
|
||||||
env_dict["YNH_USER_MAILQUOTA"] = mailbox_quota
|
env_dict["YNH_USER_MAILQUOTA"] = mailbox_quota
|
||||||
|
|
||||||
operation_logger.start()
|
if not from_import:
|
||||||
|
operation_logger.start()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
ldap.update("uid=%s,ou=users" % username, new_attr_dict)
|
ldap.update("uid=%s,ou=users" % username, new_attr_dict)
|
||||||
|
@ -475,9 +515,10 @@ def user_update(
|
||||||
# Trigger post_user_update hooks
|
# Trigger post_user_update hooks
|
||||||
hook_callback("post_user_update", env=env_dict)
|
hook_callback("post_user_update", env=env_dict)
|
||||||
|
|
||||||
logger.success(m18n.n("user_updated"))
|
if not from_import:
|
||||||
app_ssowatconf()
|
app_ssowatconf()
|
||||||
return user_info(username)
|
logger.success(m18n.n('user_updated'))
|
||||||
|
return user_info(username)
|
||||||
|
|
||||||
|
|
||||||
def user_info(username):
|
def user_info(username):
|
||||||
|
@ -507,11 +548,13 @@ def user_info(username):
|
||||||
raise YunohostValidationError("user_unknown", user=username)
|
raise YunohostValidationError("user_unknown", user=username)
|
||||||
|
|
||||||
result_dict = {
|
result_dict = {
|
||||||
"username": user["uid"][0],
|
'username': user['uid'][0],
|
||||||
"fullname": user["cn"][0],
|
'fullname': user['cn'][0],
|
||||||
"firstname": user["givenName"][0],
|
'firstname': user['givenName'][0],
|
||||||
"lastname": user["sn"][0],
|
'lastname': user['sn'][0],
|
||||||
"mail": user["mail"][0],
|
'mail': user['mail'][0],
|
||||||
|
'mail-aliases': [],
|
||||||
|
'mail-forward': []
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(user["mail"]) > 1:
|
if len(user["mail"]) > 1:
|
||||||
|
@ -566,6 +609,263 @@ def user_info(username):
|
||||||
return result_dict
|
return result_dict
|
||||||
|
|
||||||
|
|
||||||
|
def user_export():
|
||||||
|
"""
|
||||||
|
Export users into CSV
|
||||||
|
|
||||||
|
Keyword argument:
|
||||||
|
csv -- CSV file with columns username;firstname;lastname;password;mailbox-quota;mail;mail-alias;mail-forward;groups
|
||||||
|
|
||||||
|
"""
|
||||||
|
import csv # CSV are needed only in this function
|
||||||
|
from io import StringIO
|
||||||
|
with StringIO() as csv_io:
|
||||||
|
writer = csv.DictWriter(csv_io, list(FIELDS_FOR_IMPORT.keys()),
|
||||||
|
delimiter=';', quotechar='"')
|
||||||
|
writer.writeheader()
|
||||||
|
users = user_list(list(FIELDS_FOR_IMPORT.keys()))['users']
|
||||||
|
for username, user in users.items():
|
||||||
|
user['mail-alias'] = ','.join(user['mail-alias'])
|
||||||
|
user['mail-forward'] = ','.join(user['mail-forward'])
|
||||||
|
user['groups'] = ','.join(user['groups'])
|
||||||
|
writer.writerow(user)
|
||||||
|
|
||||||
|
body = csv_io.getvalue().rstrip()
|
||||||
|
if msettings.get('interface') == 'api':
|
||||||
|
# We return a raw bottle HTTPresponse (instead of serializable data like
|
||||||
|
# list/dict, ...), which is gonna be picked and used directly by moulinette
|
||||||
|
from bottle import HTTPResponse
|
||||||
|
response = HTTPResponse(body=body,
|
||||||
|
headers={
|
||||||
|
"Content-Disposition": "attachment; filename=users.csv",
|
||||||
|
"Content-Type": "text/csv",
|
||||||
|
})
|
||||||
|
return response
|
||||||
|
else:
|
||||||
|
return body
|
||||||
|
|
||||||
|
|
||||||
|
@is_unit_operation()
|
||||||
|
def user_import(operation_logger, csvfile, update=False, delete=False):
|
||||||
|
"""
|
||||||
|
Import users from CSV
|
||||||
|
|
||||||
|
Keyword argument:
|
||||||
|
csvfile -- CSV file with columns username;firstname;lastname;password;mailbox_quota;mail;alias;forward;groups
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
import csv # CSV are needed only in this function
|
||||||
|
from moulinette.utils.text import random_ascii
|
||||||
|
from yunohost.permission import permission_sync_to_user
|
||||||
|
from yunohost.app import app_ssowatconf
|
||||||
|
from yunohost.domain import domain_list
|
||||||
|
|
||||||
|
# Pre-validate data and prepare what should be done
|
||||||
|
actions = {
|
||||||
|
'created': [],
|
||||||
|
'updated': [],
|
||||||
|
'deleted': []
|
||||||
|
}
|
||||||
|
is_well_formatted = True
|
||||||
|
|
||||||
|
def to_list(str_list):
|
||||||
|
L = str_list.split(',') if str_list else []
|
||||||
|
L = [l.strip() for l in L]
|
||||||
|
return L
|
||||||
|
|
||||||
|
existing_users = user_list()['users']
|
||||||
|
existing_groups = user_group_list()["groups"]
|
||||||
|
existing_domains = domain_list()["domains"]
|
||||||
|
|
||||||
|
reader = csv.DictReader(csvfile, delimiter=';', quotechar='"')
|
||||||
|
users_in_csv = []
|
||||||
|
|
||||||
|
missing_columns = [key for key in FIELDS_FOR_IMPORT.keys() if key not in reader.fieldnames]
|
||||||
|
if missing_columns:
|
||||||
|
raise YunohostValidationError("user_import_missing_columns", columns=', '.join(missing_columns))
|
||||||
|
|
||||||
|
for user in reader:
|
||||||
|
|
||||||
|
# Validate column values against regexes
|
||||||
|
format_errors = [f"{key}: '{user[key]}' doesn't match the expected format"
|
||||||
|
for key, validator in FIELDS_FOR_IMPORT.items()
|
||||||
|
if user[key] is None or not re.match(validator, user[key])]
|
||||||
|
|
||||||
|
# Check for duplicated username lines
|
||||||
|
if user['username'] in users_in_csv:
|
||||||
|
format_errors.append(f"username '{user['username']}' duplicated")
|
||||||
|
users_in_csv.append(user['username'])
|
||||||
|
|
||||||
|
# Validate that groups exist
|
||||||
|
user['groups'] = to_list(user['groups'])
|
||||||
|
unknown_groups = [g for g in user['groups'] if g not in existing_groups]
|
||||||
|
if unknown_groups:
|
||||||
|
format_errors.append(f"username '{user['username']}': unknown groups %s" % ', '.join(unknown_groups))
|
||||||
|
|
||||||
|
# Validate that domains exist
|
||||||
|
user['mail-alias'] = to_list(user['mail-alias'])
|
||||||
|
user['mail-forward'] = to_list(user['mail-forward'])
|
||||||
|
user['domain'] = user['mail'].split('@')[1]
|
||||||
|
|
||||||
|
unknown_domains = []
|
||||||
|
if user['domain'] not in existing_domains:
|
||||||
|
unknown_domains.append(user['domain'])
|
||||||
|
|
||||||
|
unknown_domains += [mail.split('@', 1)[1] for mail in user['mail-alias'] if mail.split('@', 1)[1] not in existing_domains]
|
||||||
|
unknown_domains = set(unknown_domains)
|
||||||
|
|
||||||
|
if unknown_domains:
|
||||||
|
format_errors.append(f"username '{user['username']}': unknown domains %s" % ', '.join(unknown_domains))
|
||||||
|
|
||||||
|
if format_errors:
|
||||||
|
logger.error(m18n.n('user_import_bad_line',
|
||||||
|
line=reader.line_num,
|
||||||
|
details=', '.join(format_errors)))
|
||||||
|
is_well_formatted = False
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Choose what to do with this line and prepare data
|
||||||
|
user['mailbox-quota'] = user['mailbox-quota'] or "0"
|
||||||
|
|
||||||
|
# User creation
|
||||||
|
if user['username'] not in existing_users:
|
||||||
|
# Generate password if not exists
|
||||||
|
# This could be used when reset password will be merged
|
||||||
|
if not user['password']:
|
||||||
|
user['password'] = random_ascii(70)
|
||||||
|
actions['created'].append(user)
|
||||||
|
# User update
|
||||||
|
elif update:
|
||||||
|
actions['updated'].append(user)
|
||||||
|
|
||||||
|
if delete:
|
||||||
|
actions['deleted'] = [user for user in existing_users if user not in users_in_csv]
|
||||||
|
|
||||||
|
if delete and not users_in_csv:
|
||||||
|
logger.error("You used the delete option with an empty csv file ... You probably did not really mean to do that, did you !?")
|
||||||
|
is_well_formatted = False
|
||||||
|
|
||||||
|
if not is_well_formatted:
|
||||||
|
raise YunohostValidationError('user_import_bad_file')
|
||||||
|
|
||||||
|
total = len(actions['created'] + actions['updated'] + actions['deleted'])
|
||||||
|
|
||||||
|
if total == 0:
|
||||||
|
logger.info(m18n.n('user_import_nothing_to_do'))
|
||||||
|
return
|
||||||
|
|
||||||
|
# Apply creation, update and deletion operation
|
||||||
|
result = {
|
||||||
|
'created': 0,
|
||||||
|
'updated': 0,
|
||||||
|
'deleted': 0,
|
||||||
|
'errors': 0
|
||||||
|
}
|
||||||
|
|
||||||
|
def progress(info=""):
|
||||||
|
progress.nb += 1
|
||||||
|
width = 20
|
||||||
|
bar = int(progress.nb * width / total)
|
||||||
|
bar = "[" + "#" * bar + "." * (width - bar) + "]"
|
||||||
|
if info:
|
||||||
|
bar += " > " + info
|
||||||
|
if progress.old == bar:
|
||||||
|
return
|
||||||
|
progress.old = bar
|
||||||
|
logger.info(bar)
|
||||||
|
progress.nb = 0
|
||||||
|
progress.old = ""
|
||||||
|
|
||||||
|
def on_failure(user, exception):
|
||||||
|
result['errors'] += 1
|
||||||
|
logger.error(user + ': ' + str(exception))
|
||||||
|
|
||||||
|
def update(new_infos, old_infos=False):
|
||||||
|
remove_alias = None
|
||||||
|
remove_forward = None
|
||||||
|
remove_groups = []
|
||||||
|
add_groups = new_infos["groups"]
|
||||||
|
if old_infos:
|
||||||
|
new_infos['mail'] = None if old_infos['mail'] == new_infos['mail'] else new_infos['mail']
|
||||||
|
remove_alias = list(set(old_infos['mail-alias']) - set(new_infos['mail-alias']))
|
||||||
|
remove_forward = list(set(old_infos['mail-forward']) - set(new_infos['mail-forward']))
|
||||||
|
new_infos['mail-alias'] = list(set(new_infos['mail-alias']) - set(old_infos['mail-alias']))
|
||||||
|
new_infos['mail-forward'] = list(set(new_infos['mail-forward']) - set(old_infos['mail-forward']))
|
||||||
|
|
||||||
|
remove_groups = list(set(old_infos["groups"]) - set(new_infos["groups"]))
|
||||||
|
add_groups = list(set(new_infos["groups"]) - set(old_infos["groups"]))
|
||||||
|
|
||||||
|
for group, infos in existing_groups.items():
|
||||||
|
# Loop only on groups in 'remove_groups'
|
||||||
|
# Ignore 'all_users' and primary group
|
||||||
|
if group in ["all_users", new_infos['username']] or group not in remove_groups:
|
||||||
|
continue
|
||||||
|
# If the user is in this group (and it's not the primary group),
|
||||||
|
# remove the member from the group
|
||||||
|
if new_infos['username'] in infos["members"]:
|
||||||
|
user_group_update(group, remove=new_infos['username'], sync_perm=False, from_import=True)
|
||||||
|
|
||||||
|
user_update(new_infos['username'],
|
||||||
|
new_infos['firstname'], new_infos['lastname'],
|
||||||
|
new_infos['mail'], new_infos['password'],
|
||||||
|
mailbox_quota=new_infos['mailbox-quota'],
|
||||||
|
mail=new_infos['mail'], add_mailalias=new_infos['mail-alias'],
|
||||||
|
remove_mailalias=remove_alias,
|
||||||
|
remove_mailforward=remove_forward,
|
||||||
|
add_mailforward=new_infos['mail-forward'], from_import=True)
|
||||||
|
|
||||||
|
for group in add_groups:
|
||||||
|
if group in ["all_users", new_infos['username']]:
|
||||||
|
continue
|
||||||
|
user_group_update(group, add=new_infos['username'], sync_perm=False, from_import=True)
|
||||||
|
|
||||||
|
users = user_list(list(FIELDS_FOR_IMPORT.keys()))['users']
|
||||||
|
operation_logger.start()
|
||||||
|
# We do delete and update before to avoid mail uniqueness issues
|
||||||
|
for user in actions['deleted']:
|
||||||
|
try:
|
||||||
|
user_delete(user, purge=True, from_import=True)
|
||||||
|
result['deleted'] += 1
|
||||||
|
except YunohostError as e:
|
||||||
|
on_failure(user, e)
|
||||||
|
progress(f"Deleting {user}")
|
||||||
|
|
||||||
|
for user in actions['updated']:
|
||||||
|
try:
|
||||||
|
update(user, users[user['username']])
|
||||||
|
result['updated'] += 1
|
||||||
|
except YunohostError as e:
|
||||||
|
on_failure(user['username'], e)
|
||||||
|
progress(f"Updating {user['username']}")
|
||||||
|
|
||||||
|
for user in actions['created']:
|
||||||
|
try:
|
||||||
|
user_create(user['username'],
|
||||||
|
user['firstname'], user['lastname'],
|
||||||
|
user['domain'], user['password'],
|
||||||
|
user['mailbox-quota'], from_import=True)
|
||||||
|
update(user)
|
||||||
|
result['created'] += 1
|
||||||
|
except YunohostError as e:
|
||||||
|
on_failure(user['username'], e)
|
||||||
|
progress(f"Creating {user['username']}")
|
||||||
|
|
||||||
|
permission_sync_to_user()
|
||||||
|
app_ssowatconf()
|
||||||
|
|
||||||
|
if result['errors']:
|
||||||
|
msg = m18n.n('user_import_partial_failed')
|
||||||
|
if result['created'] + result['updated'] + result['deleted'] == 0:
|
||||||
|
msg = m18n.n('user_import_failed')
|
||||||
|
logger.error(msg)
|
||||||
|
operation_logger.error(msg)
|
||||||
|
else:
|
||||||
|
logger.success(m18n.n('user_import_success'))
|
||||||
|
operation_logger.success()
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
#
|
#
|
||||||
# Group subcategory
|
# Group subcategory
|
||||||
#
|
#
|
||||||
|
@ -738,9 +1038,15 @@ def user_group_delete(operation_logger, groupname, force=False, sync_perm=True):
|
||||||
logger.debug(m18n.n("group_deleted", group=groupname))
|
logger.debug(m18n.n("group_deleted", group=groupname))
|
||||||
|
|
||||||
|
|
||||||
@is_unit_operation([("groupname", "group")])
|
@is_unit_operation([('groupname', 'group')])
|
||||||
def user_group_update(
|
def user_group_update(
|
||||||
operation_logger, groupname, add=None, remove=None, force=False, sync_perm=True
|
operation_logger,
|
||||||
|
groupname,
|
||||||
|
add=None,
|
||||||
|
remove=None,
|
||||||
|
force=False,
|
||||||
|
sync_perm=True,
|
||||||
|
from_import=False
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
Update user informations
|
Update user informations
|
||||||
|
@ -810,7 +1116,8 @@ def user_group_update(
|
||||||
]
|
]
|
||||||
|
|
||||||
if set(new_group) != set(current_group):
|
if set(new_group) != set(current_group):
|
||||||
operation_logger.start()
|
if not from_import:
|
||||||
|
operation_logger.start()
|
||||||
ldap = _get_ldap_interface()
|
ldap = _get_ldap_interface()
|
||||||
try:
|
try:
|
||||||
ldap.update(
|
ldap.update(
|
||||||
|
@ -820,14 +1127,16 @@ def user_group_update(
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise YunohostError("group_update_failed", group=groupname, error=e)
|
raise YunohostError("group_update_failed", group=groupname, error=e)
|
||||||
|
|
||||||
if groupname != "all_users":
|
|
||||||
logger.success(m18n.n("group_updated", group=groupname))
|
|
||||||
else:
|
|
||||||
logger.debug(m18n.n("group_updated", group=groupname))
|
|
||||||
|
|
||||||
if sync_perm:
|
if sync_perm:
|
||||||
permission_sync_to_user()
|
permission_sync_to_user()
|
||||||
return user_group_info(groupname)
|
|
||||||
|
if not from_import:
|
||||||
|
if groupname != "all_users":
|
||||||
|
logger.success(m18n.n("group_updated", group=groupname))
|
||||||
|
else:
|
||||||
|
logger.debug(m18n.n("group_updated", group=groupname))
|
||||||
|
|
||||||
|
return user_group_info(groupname)
|
||||||
|
|
||||||
|
|
||||||
def user_group_info(groupname):
|
def user_group_info(groupname):
|
||||||
|
|
Loading…
Add table
Reference in a new issue