2758be8680
* Fix pylint warnings across all 24 Python files in web_server - Add module, class, and function docstrings (C0114, C0115, C0116) - Fix import ordering: stdlib before third-party before local (C0411) - Replace wildcard imports with explicit named imports (W0401) - Remove trailing whitespace and add missing final newlines (C0303, C0304) - Replace dict() with dict literals (R1735) - Remove unused imports and variables (W0611, W0612) - Narrow broad Exception catches to specific exceptions (W0718) - Replace f-string logging with lazy % formatting (W1203) - Fix variable naming: UPPER_CASE for constants, snake_case for locals (C0103) - Add pylint disable comments for necessary global statements (W0603) - Fix no-else-return, simplifiable-if-expression, singleton-comparison - Fix bad indentation in stripe.py (W0311) - Add encoding="utf-8" to open() calls (W1514) - Add check=True to subprocess.run() calls (W1510) - Register Celery task modules via conf.include * Update `package-lock.json` add peer dependencies
218 lines
7.1 KiB
Python
218 lines
7.1 KiB
Python
"""Email sending utilities for password reset, account confirmation, and newsletters."""
|
|
|
|
import smtplib
|
|
from email.mime.text import MIMEText
|
|
from os import getenv
|
|
from secrets import token_hex
|
|
|
|
from dotenv import load_dotenv
|
|
from utils.auth import generate_token
|
|
from .user_utils import get_session_info_email
|
|
import redis
|
|
from database.database import Database
|
|
|
|
REDIS_URL = "redis://redis:6379/1"
|
|
r = redis.from_url(REDIS_URL, decode_responses=True)
|
|
|
|
load_dotenv()
|
|
|
|
url = getenv("HOMEPAGE_URL")
|
|
|
|
|
|
def send_email(email, func) -> None:
|
|
"""
|
|
Send a verification email to the user.
|
|
"""
|
|
|
|
# Setup the sender email details
|
|
smtp_server = "smtp.gmail.com"
|
|
smtp_port = 587
|
|
smtp_email = getenv("EMAIL")
|
|
smtp_password = getenv("EMAIL_PASSWORD")
|
|
|
|
# Setup up the receiver details
|
|
body, subject = func()
|
|
|
|
msg = MIMEText(body, "html")
|
|
msg["Subject"] = subject
|
|
msg["From"] = smtp_email
|
|
msg["To"] = email
|
|
|
|
# Send the email using smtplib
|
|
with smtplib.SMTP(smtp_server, smtp_port) as smtp:
|
|
try:
|
|
smtp.starttls() # TLS handshake to start the connection
|
|
smtp.login(smtp_email, smtp_password)
|
|
smtp.ehlo()
|
|
smtp.send_message(msg)
|
|
|
|
except TimeoutError:
|
|
print("Server timed out", flush=True)
|
|
|
|
except smtplib.SMTPException as e:
|
|
print("Error: ", e, flush=True)
|
|
|
|
def forgot_password_body(email) -> str:
|
|
"""
|
|
Handles the creation of the email body for resetting password
|
|
"""
|
|
salt = token_hex(32)
|
|
|
|
token = generate_token(email, salt)
|
|
token += "R3sET"
|
|
r.setex(token, 3600, salt)
|
|
username = (get_session_info_email(email))["username"]
|
|
|
|
full_url = url + "/reset_password/" + token
|
|
content = f"""
|
|
<html>
|
|
<head>
|
|
<meta charset="UTF-8">
|
|
<style>
|
|
body {{ font-family: Arial, sans-serif; background-color: #f4f4f4; padding: 20px; text-align: center; }}
|
|
.container {{ background: white; padding: 20px; border-radius: 8px; box-shadow: 0 0 10px rgba(0, 0, 0, 0.1); }}
|
|
.btn {{ display: inline-block; padding: 10px 20px; color: white; background-color: #FFFFFF; text-decoration: none; border-radius: 5px; border: 1px solid #000000; font-weight: bold; }}
|
|
.btn:hover {{ background-color: #E0E0E0; }}
|
|
p {{ color: #000000; }}
|
|
a.btn {{ color: #000000; }}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<div class="container">
|
|
<h1>Gander</h1>
|
|
<h2>Password Reset Request</h2>
|
|
<p>Click the button below to reset your password for your account {username}. This link is valid for 1 hour.</p>
|
|
<a href="{full_url}" class="btn">Reset Password</a>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
|
|
return content, "Gander - Forgot Password"
|
|
|
|
def confirm_account_creation_body(email) -> str:
|
|
"""
|
|
Handles account confirmation email body for account creation
|
|
"""
|
|
salt = token_hex(32)
|
|
|
|
token = generate_token(email, salt)
|
|
token += "CrEaTe"
|
|
r.setex(token, 3600, salt)
|
|
|
|
full_url = url + "/confirm_account_creation/" + token
|
|
|
|
content = f"""
|
|
<html>
|
|
<head>
|
|
<meta charset="UTF-8">
|
|
<style>
|
|
body {{ font-family: Arial, sans-serif; background-color: #f4f4f4; padding: 20px; text-align: center; }}
|
|
.container {{ background: white; padding: 20px; border-radius: 8px; box-shadow: 0 0 10px rgba(0, 0, 0, 0.1); }}
|
|
.btn {{ display: inline-block; padding: 10px 20px; color: white; background-color: #FFFFFF; text-decoration: none; border-radius: 5px; border: 1px solid #000000; font-weight: bold; }}
|
|
.btn:hover {{ background-color: #E0E0E0; }}
|
|
p {{ color: #000000; }}
|
|
a.btn {{ color: #000000; }}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<div class="container">
|
|
<h1>Gander</h1>
|
|
<h2>Confirm Account Creation</h2>
|
|
<p>Click the button below to create your account. This link is valid for 1 hour.</p>
|
|
<a href="{full_url}" class="btn">Create Account</a>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
return content, "Gander - Confirm Account Creation"
|
|
|
|
def newsletter_conf(email):
|
|
"""
|
|
Handles sending a confirmation email that a user has joined a newsletter
|
|
"""
|
|
salt = token_hex(32)
|
|
|
|
token = generate_token(email, salt)
|
|
token += "DaNeWs"
|
|
r.setex(token, 3600, salt)
|
|
|
|
full_url = url + "/user/unsubscribe/" + token
|
|
|
|
content = f"""
|
|
<html>
|
|
<head>
|
|
<meta charset="UTF-8">
|
|
<style>
|
|
body {{ font-family: Arial, sans-serif; background-color: #f4f4f4; padding: 20px; text-align: center; }}
|
|
.container {{ background: white; padding: 20px; border-radius: 8px; box-shadow: 0 0 10px rgba(0, 0, 0, 0.1); }}
|
|
.btn {{ display: inline-block; padding: 10px 20px; color: white; background-color: #FFFFFF; text-decoration: none; border-radius: 5px; border: 1px solid #000000; font-weight: bold; }}
|
|
.btn:hover {{ background-color: #E0E0E0; }}
|
|
p {{ color: #000000; }}
|
|
a.btn {{ color: #000000; }}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<div class="container">
|
|
<h1>Gander</h1>
|
|
<h2>Welcome to the Official Gander Newsletter!</h2>
|
|
<p>If you are receiving this email, it means that you have been officially added to the Monthly Gander newsletter.</p>
|
|
<p>In this newsletter, you will receive updates about: your favourite streamers; important Gander updates; and more!</p>
|
|
<small><a href="{full_url}">Unsubscribe?</a></small>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
"""
|
|
|
|
# Check if user is already in database
|
|
with Database() as db:
|
|
user_exists = db.fetchone("""
|
|
SELECT *
|
|
FROM newsletter
|
|
WHERE email = ?;""",
|
|
(email,))
|
|
print(user_exists, flush=True)
|
|
|
|
|
|
if user_exists is None:
|
|
add_to_newsletter(email)
|
|
|
|
return content, "Gander - Newsletter"
|
|
|
|
def add_to_newsletter(email):
|
|
"""
|
|
Add a person to the newsletter database
|
|
"""
|
|
# Create connection to the database
|
|
with Database() as db:
|
|
# Add the users email to the newsletter table
|
|
db.execute("""
|
|
INSERT INTO newsletter (email)
|
|
VALUES (?);
|
|
""", (email,))
|
|
|
|
def remove_from_newsletter(email):
|
|
"""
|
|
Remove a person from the newsletter database
|
|
"""
|
|
# Create connection to the database
|
|
with Database() as db:
|
|
# Remove the users email from the newsletter table
|
|
db.execute("""
|
|
DELETE FROM newsletter
|
|
WHERE email = ?;
|
|
""", (email,))
|
|
|
|
def email_exists(email):
|
|
"""
|
|
Returns whether email exists within database
|
|
"""
|
|
with Database() as db:
|
|
data = db.fetchone("""
|
|
SELECT * FROM users
|
|
WHERE email = ?
|
|
""", (email,))
|
|
return bool(data)
|