import flask_migrate from IPython import embed from sqlalchemy_utils import create_database, database_exists, drop_database from app.config import DB_URI, ALIAS_DOMAINS from app.email_utils import send_email, render, get_email_domain_part from app.models import * from server import create_app from time import sleep from sqlalchemy import * from sqlalchemy.orm import * def create_db(): if not database_exists(DB_URI): LOG.debug("db not exist, create database") create_database(DB_URI) # Create all tables # Use flask-migrate instead of db.create_all() flask_migrate.upgrade() def change_password(user_id, new_password): user = User.get(user_id) user.set_password(new_password) db.session.commit() def reset_db(): if database_exists(DB_URI): drop_database(DB_URI) create_db() def send_safari_extension_newsletter(): for user in User.query.all(): send_email( user.email, "Quickly create alias with our Safari extension", render("com/safari-extension.txt", user=user), render("com/safari-extension.html", user=user), ) def send_mailbox_newsletter(): for user in User.query.order_by(User.id).all(): if user.notification and user.activated: try: LOG.d("Send newsletter to %s", user) send_email( user.email, "Introducing Mailbox - our most requested feature", render("com/newsletter/mailbox.txt", user=user), render("com/newsletter/mailbox.html", user=user), ) sleep(1) except Exception: LOG.warning("Cannot send to user %s", user) def send_pgp_newsletter(): for user in User.query.order_by(User.id).all(): if user.notification and user.activated: try: LOG.d("Send PGP newsletter to %s", user) send_email( user.email, "Introducing PGP - encrypt your emails so only you can read them", render("com/newsletter/pgp.txt", user=user), render("com/newsletter/pgp.html", user=user), ) sleep(1) except Exception: LOG.warning("Cannot send to user %s", user) def send_mobile_newsletter(): count = 0 for user in User.query.order_by(User.id).all(): if user.notification and user.activated: count += 1 try: LOG.d("#%s: send to %s", count, user) send_email( user.email, "Mobile and Dark Mode", render("com/newsletter/mobile-darkmode.txt", user=user), render("com/newsletter/mobile-darkmode.html", user=user), ) except Exception: LOG.warning("Cannot send to user %s", user) if count % 5 == 0: # sleep every 5 sends to avoid hitting email limits LOG.d("Sleep 1s") sleep(1) def migrate_domain_trash(): """Move aliases from global trash to domain trash if applicable""" for deleted_alias in DeletedAlias.query.all(): alias_domain = get_email_domain_part(deleted_alias.email) if alias_domain not in ALIAS_DOMAINS: domain = CustomDomain.get_by(domain=alias_domain) if domain: LOG.d("move %s to domain %s trash", deleted_alias, domain) db.session.add( DomainDeletedAlias( user_id=domain.user_id, email=deleted_alias.email, domain_id=domain.id, created_at=deleted_alias.created_at, ) ) DeletedAlias.delete(deleted_alias.id) db.session.commit() def disable_mailbox(mailbox_id): """disable a mailbox and all of its aliases""" mailbox = Mailbox.get(mailbox_id) mailbox.verified = False for alias in mailbox.aliases: alias.enabled = False db.session.commit() email_msg = f"""Hi, Your mailbox {mailbox.email} cannot receive emails. To avoid forwarding emails to an invalid mailbox, we have disabled this mailbox along with all of its aliases. If this is a mistake, please reply to this email. Thanks, SimpleLogin team. """ try: send_email( mailbox.user.email, f"{mailbox.email} is disabled", email_msg, email_msg.replace("\n", "
"), ) except Exception: LOG.exception("Cannot send disable mailbox email to %s", mailbox.user) app = create_app() with app.app_context(): # to test email template # with open("/tmp/email.html", "w") as f: # user = User.get(1) # f.write( # render( # "transactional/reset-password.html", # email=user.email, # user=user, # name=user.name, # activation_link="https://ab.cd", # alias="alias@ab.cd", # directory="dir", # domain="domain", # new_email="new@email.com", # current_email="current@email.com", # link="https://link.com", # mailbox_email="mailbox_email@email.com", # sender="sender@example.com", # reset_password_link="http://reset_password_link", # ) # ) embed()