diff --git a/init_app.py b/init_app.py index b77088ea..81ac645d 100644 --- a/init_app.py +++ b/init_app.py @@ -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()