89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
import flask_migrate
|
|
from IPython import embed
|
|
from sqlalchemy_utils import create_database, database_exists, drop_database
|
|
|
|
from app.config import DB_URI
|
|
from app.email_utils import send_email, render
|
|
from app.models import *
|
|
from server import create_app
|
|
|
|
|
|
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 convert_user_full_mailbox(user):
|
|
# create a default mailbox
|
|
default_mb = Mailbox.get_by(user_id=user.id, email=user.email)
|
|
if not default_mb:
|
|
LOG.d("create default mailbox for user %s", user)
|
|
default_mb = Mailbox.create(user_id=user.id, email=user.email, verified=True)
|
|
db.session.commit()
|
|
|
|
# assign existing alias to this mailbox
|
|
for gen_email in GenEmail.query.filter_by(user_id=user.id):
|
|
if not gen_email.mailbox_id:
|
|
LOG.d("Set alias %s mailbox to default mailbox", gen_email)
|
|
gen_email.mailbox_id = default_mb.id
|
|
|
|
# finally set user to full_mailbox
|
|
user.full_mailbox = True
|
|
user.default_mailbox_id = default_mb.id
|
|
db.session.commit()
|
|
|
|
|
|
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()
|