merged config_sync improvements/bugfixes

master
Dmytro Makovey 7 years ago
commit b05e924f3b

@ -90,24 +90,34 @@ def config_update(verbose=False, delete_objects=False):
max_aliases=max_aliases, max_aliases=max_aliases,
max_quota_bytes=max_quota_bytes) max_quota_bytes=max_quota_bytes)
db.session.add(domain) db.session.add(domain)
print("Added "+str(domain_config))
else: else:
domain.max_users = max_users domain.max_users = max_users
domain.max_aliases = max_aliases domain.max_aliases = max_aliases
domain.max_quota_bytes = max_quota_bytes domain.max_quota_bytes = max_quota_bytes
db.session.add(domain) db.session.add(domain)
print("Updated "+str(domain_config))
users=new_config.get('users',[]) users=new_config.get('users',[])
tracked_users=set() 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: for user_config in users:
if verbose: if verbose:
print(str(user_config)) print(str(user_config))
localpart=user_config['localpart'] localpart=user_config['localpart']
domain_name=user_config['domain'] domain_name=user_config['domain']
global_admin=user_config.get('global_admin',False)
password_hash=user_config.get('password_hash',None) password_hash=user_config.get('password_hash',None)
hash_scheme=user_config.get('hash_scheme',None) hash_scheme=user_config.get('hash_scheme',None)
domain = models.Domain.query.get(domain_name) domain = models.Domain.query.get(domain_name)
email='{0}@{1}'.format(localpart,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: if not domain:
domain = models.Domain(name=domain_name) domain = models.Domain(name=domain_name)
db.session.add(domain) db.session.add(domain)
@ -118,8 +128,11 @@ def config_update(verbose=False, delete_objects=False):
user = models.User( user = models.User(
localpart=localpart, localpart=localpart,
domain=domain, domain=domain,
global_admin=False **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) user.set_password(password_hash, hash_scheme=hash_scheme, raw=True)
db.session.add(user) db.session.add(user)
@ -130,11 +143,11 @@ def config_update(verbose=False, delete_objects=False):
print(str(alias_config)) print(str(alias_config))
localpart=alias_config['localpart'] localpart=alias_config['localpart']
domain_name=alias_config['domain'] domain_name=alias_config['domain']
pre_destination=alias_config['destination'] if type(alias_config['destination']) is str:
if type(pre_destination) is str: destination = alias_config['destination'].split(',')
destination = pre_destination.split(',')
else: else:
destination = pre_destination destination = alias_config['destination']
wildcard=alias_config.get('wildcard',False)
domain = models.Domain.query.get(domain_name) domain = models.Domain.query.get(domain_name)
email='{0}@{1}'.format(localpart,domain_name) email='{0}@{1}'.format(localpart,domain_name)
if not domain: if not domain:
@ -147,34 +160,47 @@ def config_update(verbose=False, delete_objects=False):
alias = models.Alias( alias = models.Alias(
localpart=localpart, localpart=localpart,
domain=domain, domain=domain,
wildcard=wildcard,
destination=destination, destination=destination,
email=email email=email
) )
else: else:
alias.destination = destination alias.destination = destination
alias.wildcard=wildcard
db.session.add(alias) db.session.add(alias)
db.session.commit()
managers=new_config.get('managers',[]) managers=new_config.get('managers',[])
# tracked_managers=set() # tracked_managers=set()
for manager_config in managers: for manager_config in managers:
if verbose: if verbose:
print(str(manager_config)) print(str(manager_config))
domain_name = manager_config['domain'] domain_name = manager_config['domain']
user_name = manage_config['user'] user_name = manager_config['user']
domain = models.Domain.query.get(domain_name) domain = models.Domain.query.get(domain_name)
manageruser = models.User.query.get(user_name + '@' + domain_name) manageruser = models.User.query.get(user_name + '@' + domain_name)
if not manageruser in domain.managers:
domain.managers.append(manageruser) domain.managers.append(manageruser)
db.session.add(domain) db.session.add(domain)
db.session.commit()
if delete_objects: if delete_objects:
for user in db.session.query(models.User).all(): for user in db.session.query(models.User).all():
if not ( user.email in tracked_users ): if not ( user.email in tracked_users ):
if verbose:
print("Deleting user: "+str(user.email))
db.session.delete(user) db.session.delete(user)
for alias in db.session.query(models.Alias).all(): for alias in db.session.query(models.Alias).all():
if not ( alias.email in tracked_aliases ): if not ( alias.email in tracked_aliases ):
if verbose:
print("Deleting alias: "+str(alias.email))
db.session.delete(alias) db.session.delete(alias)
for domain in db.session.query(models.Domain).all(): for domain in db.session.query(models.Domain).all():
if not ( domain.name in tracked_domains ): if not ( domain.name in tracked_domains ):
if verbose:
print("Deleting domain: "+str(domain.name))
db.session.delete(domain) db.session.delete(domain)
db.session.commit() db.session.commit()

Loading…
Cancel
Save