72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
import os
|
|
import sqlite3
|
|
import smtplib
|
|
from email.mime.text import MIMEText
|
|
from flask import Flask, render_template, request, redirect, url_for, flash
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
app = Flask(__name__)
|
|
|
|
app.secret_key = os.getenv('SECRET_KEY')
|
|
|
|
DATABASE_URL = os.getenv('DATABASE_URL')
|
|
SMTP_SERVER = os.getenv('SMTP_SERVER')
|
|
SMTP_PORT = int(os.getenv("SMTP_PORT", 465))
|
|
SMTP_USER = os.getenv('SMTP_USER')
|
|
SMTP_PASSWORD = os.getenv('SMTP_PASSWORD')
|
|
|
|
def get_all_emails():
|
|
"""Retrieve all subscriber emails from the database"""
|
|
try:
|
|
conn = sqlite3.connect(DATABASE_URL)
|
|
cursor = conn.cursor()
|
|
cursor.execute('SELECT email FROM subscribers')
|
|
results = cursor.fetchall()
|
|
conn.close()
|
|
return [row[0] for row in results]
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
return []
|
|
|
|
def send_update_email(subject, body):
|
|
"""Send an update email"""
|
|
subscribers = get_all_emails()
|
|
if not subscribers:
|
|
return "No subscribers found"
|
|
try:
|
|
server = smtplib.SMTP(SMTP_SERVER, SMTP_PORT, timeout=10)
|
|
server.set_debuglevel(True)
|
|
server.login(SMTP_USER, SMTP_PASSWORD)
|
|
for email in subscribers:
|
|
msg = MIMEText(body, 'html', 'utf-8')
|
|
msg['Subject'] = subject
|
|
msg['From'] = SMTP_USER
|
|
msg['To'] = email
|
|
server.sendmail(SMTP_USER, email, msg.as_string())
|
|
print(f"Updated email for {email} has been sent.")
|
|
server.quit()
|
|
return "Email has been sent."
|
|
except Exception as e:
|
|
print(f"Failed to send email: {e}")
|
|
return f"Failed to send email: {e}"
|
|
|
|
@app.route('/')
|
|
def index():
|
|
"""Displays all subscriber emails"""
|
|
emails = get_all_emails()
|
|
return render_template("admin_index.html", emails=emails)
|
|
|
|
@app.route('/send_update_email', methods=['GET', 'POST'])
|
|
def send_update_email():
|
|
"""Display a form to send an update email"""
|
|
if request.method == 'POST':
|
|
subject = request.form['subject']
|
|
body = request.form['body']
|
|
result_message = send_update_email(subject, body)
|
|
flash(result_message)
|
|
return redirect(url_for("send_update_email"))
|
|
return render_template("send_update.html")
|
|
|
|
if __name__ == '__main__':
|
|
app.run(port=5001, debug=True)
|