def process_servers(server_type, except_re):
es = EmailServer(db)
eq = EmailQuota(db)
existing_servers = {}
# This lists all hosts, so we need to filter on server_type later.
for row in es.list_traits(co.trait_email_server_weight):
existing_servers[int(row['entity_id'])] = True
assigned = {}
for row in es.list_email_server_ext(server_type=server_type):
# logger.debug("Processing %r" % row.dict())
if except_re and except_re.match(row['name']):
logger.debug("Skipping server named '%s'" % row['name'])
continue
srv = int(row['server_id'])
assigned[srv] = eq.get_quota_stats_by_server(srv)['total_quota'] or 0
logger.debug("%s has assigned quota %d" % (row['name'], assigned[srv]))
max_weight = max(assigned.values()) * 110 / 100
for srv in assigned:
es.clear()
es.find(srv)
# We add 1 to handle the case with only one server
weight = max_weight - assigned[srv] + 1
logger.debug("Assigning weight %d to server ID %d" % (weight, srv))
es.populate_trait(co.trait_email_server_weight, numval=weight)
es.write_db()
if srv in existing_servers:
del existing_servers[srv]
for obsolete in existing_servers:
es.clear()
es.find(obsolete)
if server_type and es.email_server_type != server_type:
continue
logger.info("Deleting old weight trait for %s" % es.name)
es.delete_trait(co.trait_email_server_weight)
def process_servers(server_type, except_re):
es = EmailServer(db)
eq = EmailQuota(db)
existing_servers = {}
# This lists all hosts, so we need to filter on server_type later.
for row in es.list_traits(co.trait_email_server_weight):
existing_servers[int(row['entity_id'])] = True
assigned = {}
for row in es.list_email_server_ext(server_type=server_type):
# logger.debug("Processing %r" % row.dict())
if except_re and except_re.match(row['name']):
logger.debug("Skipping server named '%s'" % row['name'])
continue
srv = int(row['server_id'])
assigned[srv] = eq.get_quota_stats_by_server(srv)['total_quota'] or 0
logger.debug("%s has assigned quota %d" % (row['name'], assigned[srv]))
max_weight = max(assigned.values()) * 110 / 100
for srv in assigned:
es.clear()
es.find(srv)
# We add 1 to handle the case with only one server
weight = max_weight - assigned[srv] + 1
logger.debug("Assigning weight %d to server ID %d" % (weight, srv))
es.populate_trait(co.trait_email_server_weight, numval=weight)
es.write_db()
if srv in existing_servers:
del existing_servers[srv]
for obsolete in existing_servers:
es.clear()
es.find(obsolete)
if server_type and es.email_server_type != server_type:
continue
logger.info("Deleting old weight trait for %s" % es.name)
es.delete_trait(co.trait_email_server_weight)
