1
0
mirror of https://github.com/Mailu/Mailu.git synced 2024-12-16 10:59:53 +02:00
Mailu/core/admin/manage.py

299 lines
9.6 KiB
Python
Raw Normal View History

2017-09-17 14:37:10 +02:00
from mailu import app, manager, db, models
2016-06-25 14:02:50 +02:00
import os
import socket
import uuid
@manager.command
def advertise():
""" Advertise this server against statistic services.
"""
if os.path.isfile(app.config["INSTANCE_ID_PATH"]):
with open(app.config["INSTANCE_ID_PATH"], "r") as handle:
instance_id = handle.read()
else:
instance_id = str(uuid.uuid4())
with open(app.config["INSTANCE_ID_PATH"], "w") as handle:
handle.write(instance_id)
if app.config["DISABLE_STATISTICS"].lower() != "true":
try:
socket.gethostbyname(app.config["STATS_ENDPOINT"].format(instance_id))
except:
pass
2016-06-25 14:02:50 +02:00
@manager.command
def admin(localpart, domain_name, password):
""" Create an admin user
"""
domain = models.Domain.query.get(domain_name)
if not domain:
domain = models.Domain(name=domain_name)
db.session.add(domain)
2016-06-25 14:02:50 +02:00
user = models.User(
localpart=localpart,
domain=domain,
global_admin=True
2016-06-25 14:02:50 +02:00
)
user.set_password(password)
2016-06-25 14:02:50 +02:00
db.session.add(user)
db.session.commit()
2016-06-25 12:57:47 +02:00
2017-09-28 17:28:10 +02:00
2016-12-14 09:16:48 +02:00
@manager.command
2017-09-28 17:37:39 +02:00
def user(localpart, domain_name, password,
hash_scheme=app.config['PASSWORD_SCHEME']):
""" Create a user
2016-12-14 09:16:48 +02:00
"""
domain = models.Domain.query.get(domain_name)
if not domain:
domain = models.Domain(name=domain_name)
db.session.add(domain)
user = models.User(
localpart=localpart,
domain=domain,
global_admin=False
2016-12-14 09:16:48 +02:00
)
2017-08-24 16:23:54 +02:00
user.set_password(password, hash_scheme=hash_scheme)
db.session.add(user)
db.session.commit()
2017-09-28 17:28:10 +02:00
@manager.option('-n', '--domain_name', dest='domain_name')
@manager.option('-u', '--max_users', dest='max_users')
@manager.option('-a', '--max_aliases', dest='max_aliases')
@manager.option('-q', '--max_quota_bytes', dest='max_quota_bytes')
2017-09-26 22:20:33 +02:00
def domain(domain_name, max_users=0, max_aliases=0, max_quota_bytes=0):
domain = models.Domain.query.get(domain_name)
if not domain:
domain = models.Domain(name=domain_name)
db.session.add(domain)
db.session.commit()
2017-09-22 09:42:37 +02:00
2017-09-28 17:28:10 +02:00
2017-08-24 16:23:54 +02:00
@manager.command
def user_import(localpart, domain_name, password_hash,
2017-09-28 17:37:39 +02:00
hash_scheme=app.config['PASSWORD_SCHEME']):
""" Import a user along with password hash. Available hashes:
'SHA512-CRYPT'
'SHA256-CRYPT'
'MD5-CRYPT'
'CRYPT'
2017-08-24 16:23:54 +02:00
"""
domain = models.Domain.query.get(domain_name)
if not domain:
domain = models.Domain(name=domain_name)
db.session.add(domain)
user = models.User(
localpart=localpart,
domain=domain,
global_admin=False
)
user.set_password(password_hash, hash_scheme=hash_scheme, raw=True)
2016-12-14 09:16:48 +02:00
db.session.add(user)
db.session.commit()
2016-06-25 12:57:47 +02:00
2017-09-28 17:28:10 +02:00
@manager.command
def config_update(verbose=False, delete_objects=False):
2017-08-25 22:07:07 +02:00
"""sync configuration with data from YAML-formatted stdin"""
2017-09-28 17:28:10 +02:00
import yaml
import sys
new_config = yaml.load(sys.stdin)
# print new_config
2017-09-28 17:28:10 +02:00
domains = new_config.get('domains', [])
tracked_domains = set()
2017-09-26 22:20:33 +02:00
for domain_config in domains:
if verbose:
print(str(domain_config))
domain_name = domain_config['name']
2017-09-28 17:28:10 +02:00
max_users = domain_config.get('max_users', 0)
max_aliases = domain_config.get('max_aliases', 0)
max_quota_bytes = domain_config.get('max_quota_bytes', 0)
tracked_domains.add(domain_name)
domain = models.Domain.query.get(domain_name)
if not domain:
domain = models.Domain(name=domain_name,
2017-09-28 17:28:10 +02:00
max_users=max_users,
max_aliases=max_aliases,
max_quota_bytes=max_quota_bytes)
db.session.add(domain)
2017-09-28 17:28:10 +02:00
print("Added " + str(domain_config))
else:
domain.max_users = max_users
domain.max_aliases = max_aliases
domain.max_quota_bytes = max_quota_bytes
db.session.add(domain)
2017-09-28 17:28:10 +02:00
print("Updated " + str(domain_config))
2017-09-26 22:20:33 +02:00
2017-09-28 17:28:10 +02:00
users = new_config.get('users', [])
tracked_users = set()
user_optional_params = ('comment', 'quota_bytes', 'global_admin',
2017-09-28 17:37:39 +02:00
'enable_imap', 'enable_pop', 'forward_enabled',
'forward_destination', 'reply_enabled',
'reply_subject', 'reply_body', 'displayed_name',
'spam_enabled', 'email', 'spam_threshold')
for user_config in users:
if verbose:
2017-09-11 23:06:00 +02:00
print(str(user_config))
2017-09-28 17:28:10 +02:00
localpart = user_config['localpart']
domain_name = user_config['domain']
password_hash = user_config.get('password_hash', None)
hash_scheme = user_config.get('hash_scheme', None)
domain = models.Domain.query.get(domain_name)
2017-09-28 17:28:10 +02:00
email = '{0}@{1}'.format(localpart, domain_name)
optional_params = {}
2017-09-28 06:29:42 +02:00
for k in user_optional_params:
2017-09-28 06:34:10 +02:00
if k in user_config:
2017-09-28 17:28:10 +02:00
optional_params[k] = user_config[k]
if not domain:
domain = models.Domain(name=domain_name)
db.session.add(domain)
2017-08-25 22:07:07 +02:00
user = models.User.query.get(email)
tracked_users.add(email)
2017-09-26 22:20:33 +02:00
tracked_domains.add(domain_name)
if not user:
user = models.User(
localpart=localpart,
domain=domain,
2017-09-28 06:29:42 +02:00
**optional_params
)
2017-09-28 06:48:20 +02:00
else:
for k in optional_params:
setattr(user, k, optional_params[k])
user.set_password(password_hash, hash_scheme=hash_scheme, raw=True)
db.session.add(user)
2017-09-28 17:28:10 +02:00
aliases = new_config.get('aliases', [])
tracked_aliases = set()
for alias_config in aliases:
if verbose:
2017-09-11 23:06:00 +02:00
print(str(alias_config))
2017-09-28 17:28:10 +02:00
localpart = alias_config['localpart']
domain_name = alias_config['domain']
if type(alias_config['destination']) is str:
2017-09-28 06:29:42 +02:00
destination = alias_config['destination'].split(',')
2017-09-22 09:42:37 +02:00
else:
2017-09-28 06:29:42 +02:00
destination = alias_config['destination']
2017-09-28 17:28:10 +02:00
wildcard = alias_config.get('wildcard', False)
domain = models.Domain.query.get(domain_name)
2017-09-28 17:28:10 +02:00
email = '{0}@{1}'.format(localpart, domain_name)
if not domain:
domain = models.Domain(name=domain_name)
db.session.add(domain)
2017-08-25 22:07:07 +02:00
alias = models.Alias.query.get(email)
tracked_aliases.add(email)
2017-09-26 22:20:33 +02:00
tracked_domains.add(domain_name)
if not alias:
alias = models.Alias(
localpart=localpart,
domain=domain,
2017-09-28 06:29:42 +02:00
wildcard=wildcard,
2017-09-22 09:42:37 +02:00
destination=destination,
2017-08-25 22:07:07 +02:00
email=email
)
else:
2017-09-22 09:42:37 +02:00
alias.destination = destination
2017-09-28 17:28:10 +02:00
alias.wildcard = wildcard
db.session.add(alias)
2017-09-26 22:20:33 +02:00
db.session.commit()
2017-09-28 17:28:10 +02:00
managers = new_config.get('managers', [])
2017-09-26 22:20:33 +02:00
# tracked_managers=set()
for manager_config in managers:
if verbose:
print(str(manager_config))
domain_name = manager_config['domain']
2017-09-28 04:44:19 +02:00
user_name = manager_config['user']
2017-09-26 22:20:33 +02:00
domain = models.Domain.query.get(domain_name)
manageruser = models.User.query.get(user_name + '@' + domain_name)
2017-09-28 17:37:39 +02:00
if manageruser not in domain.managers:
2017-09-28 05:52:03 +02:00
domain.managers.append(manageruser)
2017-09-26 22:20:33 +02:00
db.session.add(domain)
2017-09-28 17:28:10 +02:00
db.session.commit()
2017-08-25 22:07:07 +02:00
if delete_objects:
for user in db.session.query(models.User).all():
2017-09-28 17:28:10 +02:00
if not (user.email in tracked_users):
2017-09-28 05:16:23 +02:00
if verbose:
2017-09-28 17:28:10 +02:00
print("Deleting user: " + str(user.email))
2017-08-25 22:07:07 +02:00
db.session.delete(user)
for alias in db.session.query(models.Alias).all():
2017-09-28 17:28:10 +02:00
if not (alias.email in tracked_aliases):
2017-09-28 05:16:23 +02:00
if verbose:
2017-09-28 17:28:10 +02:00
print("Deleting alias: " + str(alias.email))
2017-08-25 22:07:07 +02:00
db.session.delete(alias)
2017-09-26 22:20:33 +02:00
for domain in db.session.query(models.Domain).all():
2017-09-28 17:28:10 +02:00
if not (domain.name in tracked_domains):
2017-09-28 05:16:23 +02:00
if verbose:
2017-09-28 17:28:10 +02:00
print("Deleting domain: " + str(domain.name))
2017-09-26 22:20:33 +02:00
db.session.delete(domain)
db.session.commit()
2017-09-28 17:28:10 +02:00
@manager.command
def user_delete(email):
"""delete user"""
user = models.User.query.get(email)
if user:
db.session.delete(user)
db.session.commit()
2017-09-28 17:28:10 +02:00
@manager.command
def alias_delete(email):
"""delete alias"""
alias = models.Alias.query.get(email)
if alias:
db.session.delete(alias)
db.session.commit()
2017-09-28 17:28:10 +02:00
2016-12-14 10:40:49 +02:00
@manager.command
def alias(localpart, domain_name, destination):
""" Create an alias
"""
domain = models.Domain.query.get(domain_name)
if not domain:
domain = models.Domain(name=domain_name)
db.session.add(domain)
alias = models.Alias(
2016-12-14 10:40:49 +02:00
localpart=localpart,
domain=domain,
destination=destination.split(','),
email="%s@%s" % (localpart, domain_name)
)
db.session.add(alias)
db.session.commit()
# Set limits to a domain
2017-09-28 17:28:10 +02:00
@manager.command
def setlimits(domain_name, max_users, max_aliases, max_quota_bytes):
2017-09-28 17:28:10 +02:00
domain = models.Domain.query.get(domain_name)
domain.max_users = max_users
domain.max_aliases = max_aliases
domain.max_quota_bytes = max_quota_bytes
2017-09-28 17:28:10 +02:00
db.session.add(domain)
db.session.commit()
# Make the user manager of a domain
2017-09-28 17:28:10 +02:00
@manager.command
def setmanager(domain_name, user_name='manager'):
2017-09-28 17:28:10 +02:00
domain = models.Domain.query.get(domain_name)
manageruser = models.User.query.get(user_name + '@' + domain_name)
domain.managers.append(manageruser)
db.session.add(domain)
db.session.commit()
2016-06-25 12:57:47 +02:00
if __name__ == "__main__":
manager.run()