add all domain in ALIAS_DOMAINS to public domain table

This commit is contained in:
Son NK 2020-07-04 23:18:30 +02:00
parent ee9b796b7d
commit dc9ee048a2
1 changed files with 14 additions and 1 deletions

View File

@ -1,5 +1,6 @@
"""Initial loading script"""
from app.models import Mailbox, Contact
from app.config import ALIAS_DOMAINS
from app.models import Mailbox, Contact, PublicDomain
from app.log import LOG
from app.extensions import db
from app.pgp_utils import load_public_key
@ -32,8 +33,20 @@ def load_pgp_public_keys():
LOG.d("Finish load_pgp_public_keys")
def add_public_domains():
for alias_domain in ALIAS_DOMAINS:
if PublicDomain.get_by(domain=alias_domain):
LOG.d("%s is already a public domain", alias_domain)
else:
LOG.info("Add %s to public domain", alias_domain)
PublicDomain.create(domain=alias_domain)
db.session.commit()
if __name__ == "__main__":
app = create_app()
with app.app_context():
load_pgp_public_keys()
add_public_domains()