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(localpart=localpart, domain=domain, destination=destination.split(','), email="%s@%s" % (localpart, domain_name)) db.session.add(alias) db.session.commit()
def alias(localpart, domain_name, destination, wildcard=False): """ 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(localpart=localpart, domain=domain, wildcard=wildcard, destination=destination.split(','), email=f'{localpart}@{domain_name}') db.session.add(alias) db.session.commit()
def alias_create(domain_name): domain = models.Domain.query.get(domain_name) or flask.abort(404) if domain.max_aliases and len(domain.aliases) >= domain.max_aliases: flask.flash('Too many aliases for domain %s' % domain, 'error') return flask.redirect( flask.url_for('.alias_list', domain_name=domain.name)) form = forms.AliasForm() if form.validate_on_submit(): if domain.has_email(form.localpart.data): flask.flash('Email is already used', 'error') else: alias = models.Alias(domain=domain) form.populate_obj(alias) db.session.add(alias) db.session.commit() flask.flash('Alias %s created' % alias) return flask.redirect( flask.url_for('.alias_list', domain_name=domain.name)) return flask.render_template('alias/create.html', domain=domain, form=form)
def config_update(verbose=False, delete_objects=False): """sync configuration with data from YAML-formatted stdin""" import yaml import sys new_config = yaml.load(sys.stdin) # print new_config domains = new_config.get('domains', []) tracked_domains = set() for domain_config in domains: if verbose: print(str(domain_config)) domain_name = domain_config['name'] 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, max_users=max_users, max_aliases=max_aliases, max_quota_bytes=max_quota_bytes) db.session.add(domain) 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) print("Updated " + str(domain_config)) users = new_config.get('users', []) tracked_users = set() user_optional_params = ('comment', 'quota_bytes', 'global_admin', '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: print(str(user_config)) 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) email = '{0}@{1}'.format(localpart, domain_name) optional_params = {} for k in user_optional_params: if k in user_config: optional_params[k] = user_config[k] if not domain: domain = models.Domain(name=domain_name) db.session.add(domain) user = models.User.query.get(email) tracked_users.add(email) tracked_domains.add(domain_name) if not user: user = models.User(localpart=localpart, domain=domain, **optional_params) 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) aliases = new_config.get('aliases', []) tracked_aliases = set() for alias_config in aliases: if verbose: print(str(alias_config)) localpart = alias_config['localpart'] domain_name = alias_config['domain'] if type(alias_config['destination']) is str: destination = alias_config['destination'].split(',') else: destination = alias_config['destination'] wildcard = alias_config.get('wildcard', False) domain = models.Domain.query.get(domain_name) email = '{0}@{1}'.format(localpart, domain_name) if not domain: domain = models.Domain(name=domain_name) db.session.add(domain) alias = models.Alias.query.get(email) tracked_aliases.add(email) tracked_domains.add(domain_name) if not alias: alias = models.Alias(localpart=localpart, domain=domain, wildcard=wildcard, destination=destination, email=email) else: alias.destination = destination alias.wildcard = wildcard db.session.add(alias) db.session.commit() managers = new_config.get('managers', []) # tracked_managers=set() for manager_config in managers: if verbose: print(str(manager_config)) domain_name = manager_config['domain'] user_name = manager_config['user'] domain = models.Domain.query.get(domain_name) manageruser = models.User.query.get(user_name + '@' + domain_name) if manageruser not in domain.managers: domain.managers.append(manageruser) db.session.add(domain) db.session.commit() if delete_objects: for user in db.session.query(models.User).all(): if not (user.email in tracked_users): if verbose: print("Deleting user: "******"Deleting alias: " + str(alias.email)) db.session.delete(alias) for domain in db.session.query(models.Domain).all(): if not (domain.name in tracked_domains): if verbose: print("Deleting domain: " + str(domain.name)) db.session.delete(domain) db.session.commit()
def config_update(verbose=False, delete_objects=False): """ Sync configuration with data from YAML (deprecated) """ new_config = yaml.safe_load(sys.stdin) # print new_config domains = new_config.get('domains', []) tracked_domains = set() for domain_config in domains: if verbose: print(str(domain_config)) domain_name = domain_config['name'] max_users = domain_config.get('max_users', -1) max_aliases = domain_config.get('max_aliases', -1) 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, max_users=max_users, max_aliases=max_aliases, max_quota_bytes=max_quota_bytes) db.session.add(domain) print(f'Added {domain_config}') else: domain.max_users = max_users domain.max_aliases = max_aliases domain.max_quota_bytes = max_quota_bytes db.session.add(domain) print(f'Updated {domain_config}') users = new_config.get('users', []) tracked_users = set() user_optional_params = ('comment', 'quota_bytes', 'global_admin', '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: print(str(user_config)) localpart = user_config['localpart'] domain_name = user_config['domain'] password_hash = user_config.get('password_hash', None) domain = models.Domain.query.get(domain_name) email = f'{localpart}@{domain_name}' optional_params = {} for k in user_optional_params: if k in user_config: optional_params[k] = user_config[k] if not domain: domain = models.Domain(name=domain_name) db.session.add(domain) user = models.User.query.get(email) tracked_users.add(email) tracked_domains.add(domain_name) if not user: user = models.User(localpart=localpart, domain=domain, **optional_params) else: for k in optional_params: setattr(user, k, optional_params[k]) user.set_password(password_hash, raw=True) db.session.add(user) aliases = new_config.get('aliases', []) tracked_aliases = set() for alias_config in aliases: if verbose: print(str(alias_config)) localpart = alias_config['localpart'] domain_name = alias_config['domain'] if isinstance(alias_config['destination'], str): destination = alias_config['destination'].split(',') else: destination = alias_config['destination'] wildcard = alias_config.get('wildcard', False) domain = models.Domain.query.get(domain_name) email = f'{localpart}@{domain_name}' if not domain: domain = models.Domain(name=domain_name) db.session.add(domain) alias = models.Alias.query.get(email) tracked_aliases.add(email) tracked_domains.add(domain_name) if not alias: alias = models.Alias(localpart=localpart, domain=domain, wildcard=wildcard, destination=destination, email=email) else: alias.destination = destination alias.wildcard = wildcard db.session.add(alias) db.session.commit() managers = new_config.get('managers', []) # tracked_managers=set() for manager_config in managers: if verbose: print(str(manager_config)) domain_name = manager_config['domain'] user_name = manager_config['user'] domain = models.Domain.query.get(domain_name) manageruser = models.User.query.get(f'{user_name}@{domain_name}') if manageruser not in domain.managers: domain.managers.append(manageruser) db.session.add(domain) db.session.commit() if delete_objects: for user in db.session.query(models.User).all(): if not user.email in tracked_users: if verbose: print(f'Deleting user: {user.email}') db.session.delete(user) for alias in db.session.query(models.Alias).all(): if not alias.email in tracked_aliases: if verbose: print(f'Deleting alias: {alias.email}') db.session.delete(alias) for domain in db.session.query(models.Domain).all(): if not domain.name in tracked_domains: if verbose: print(f'Deleting domain: {domain.name}') db.session.delete(domain) db.session.commit()