Fix linting issues in email service and auth routes

Co-authored-by: cadtoolbox <12723486+cadtoolbox@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-02-10 00:49:01 +00:00
co-authored by cadtoolbox
parent 94e01499b7
commit 3aab9d7052
3 changed files with 71 additions and 71 deletions
+6 -6
View File
@@ -5,6 +5,7 @@ from sqlalchemy import select
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.orm import selectinload
from backend.app.api.routes.settings import get_external_login_url
from backend.app.core.auth import (
ACCESS_TOKEN_EXPIRE_MINUTES,
authenticate_user,
@@ -41,7 +42,6 @@ from backend.app.services.email_service import (
save_smtp_settings,
send_email,
)
from backend.app.api.routes.settings import get_external_login_url
def _user_to_response(user: User) -> UserResponse:
@@ -263,7 +263,7 @@ async def disable_auth(
@router.post("/login", response_model=LoginResponse)
async def login(request: LoginRequest, db: AsyncSession = Depends(get_db)):
"""Login and get access token.
Supports username or email-based login. Username lookup is case-insensitive.
"""
# Check if auth is enabled
@@ -276,13 +276,13 @@ async def login(request: LoginRequest, db: AsyncSession = Depends(get_db)):
# Try username-based authentication first
user = await authenticate_user(db, request.username, request.password)
# If username auth failed and advanced auth is enabled, try email-based authentication
if not user:
advanced_auth = await is_advanced_auth_enabled(db)
if advanced_auth:
user = await authenticate_user_by_email(db, request.username, request.password)
if not user:
raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED,
@@ -437,7 +437,7 @@ async def enable_advanced_auth(
db: AsyncSession = Depends(get_db),
):
"""Enable advanced authentication (admin only).
Requires SMTP settings to be configured and tested first.
"""
import logging
@@ -546,7 +546,7 @@ async def forgot_password(request: ForgotPasswordRequest, db: AsyncSession = Dep
# Find user by email
user = await get_user_by_email(db, request.email)
# Always return success message to prevent email enumeration
# but only send email if user exists
if user and user.is_active:
+2 -2
View File
@@ -3,6 +3,7 @@ from sqlalchemy import delete, func, select
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.orm import selectinload
from backend.app.api.routes.settings import get_external_login_url
from backend.app.core.auth import (
RequirePermissionIfAuthEnabled,
get_current_user_optional,
@@ -24,7 +25,6 @@ from backend.app.services.email_service import (
get_smtp_settings,
send_email,
)
from backend.app.api.routes.settings import get_external_login_url
router = APIRouter(prefix="/users", tags=["users"])
@@ -64,7 +64,7 @@ async def create_user(
db: AsyncSession = Depends(get_db),
):
"""Create a new user.
When advanced authentication is enabled:
- Email is required
- Password is auto-generated and emailed to user
+63 -63
View File
@@ -23,21 +23,21 @@ logger = logging.getLogger(__name__)
def generate_secure_password(length: int = 16) -> str:
"""Generate a secure random password.
Args:
length: Length of the password (default: 16)
Returns:
A secure random password containing uppercase, lowercase, digits, and special characters
"""
import random
# Define character sets
lowercase = string.ascii_lowercase
uppercase = string.ascii_uppercase
digits = string.digits
special = "!@#$%^&*()_+-=[]{}|;:,.<>?"
# Ensure at least one character from each set
password_chars = [
secrets.choice(lowercase),
@@ -45,24 +45,24 @@ def generate_secure_password(length: int = 16) -> str:
secrets.choice(digits),
secrets.choice(special),
]
# Fill the rest with random characters from all sets
all_chars = lowercase + uppercase + digits + special
password_chars.extend(secrets.choice(all_chars) for _ in range(length - 4))
# Shuffle to avoid predictable patterns
random.shuffle(password_chars)
return "".join(password_chars)
async def get_notification_template(db: AsyncSession, event_type: str) -> NotificationTemplate | None:
"""Get a notification template by event type from database.
Args:
db: Database session
event_type: Type of event (e.g., 'user_created', 'password_reset')
Returns:
NotificationTemplate object or None if not found
"""
@@ -74,11 +74,11 @@ async def get_notification_template(db: AsyncSession, event_type: str) -> Notifi
def render_template(template_str: str, variables: dict[str, Any]) -> str:
"""Render a template string with variables.
Args:
template_str: Template string with {variable} placeholders
variables: Dictionary of variables to substitute
Returns:
Rendered template string
"""
@@ -92,10 +92,10 @@ def render_template(template_str: str, variables: dict[str, Any]) -> str:
async def get_smtp_settings(db: AsyncSession) -> SMTPSettings | None:
"""Get SMTP settings from database.
Args:
db: Database session
Returns:
SMTPSettings object or None if not configured
"""
@@ -116,21 +116,21 @@ async def get_smtp_settings(db: AsyncSession) -> SMTPSettings | None:
)
)
settings_dict = {s.key: s.value for s in result.scalars().all()}
# Check if minimum required settings are present
required_keys = ["smtp_host", "smtp_port", "smtp_from_email"]
if not all(key in settings_dict for key in required_keys):
return None
# Handle migration: convert old smtp_use_tls to smtp_security if needed
smtp_security = settings_dict.get("smtp_security")
if not smtp_security:
# Migrate from old smtp_use_tls format
smtp_use_tls = settings_dict.get("smtp_use_tls", "true").lower() == "true"
smtp_security = "starttls" if smtp_use_tls else "ssl"
smtp_auth_enabled = settings_dict.get("smtp_auth_enabled", "true").lower() == "true"
return SMTPSettings(
smtp_host=settings_dict["smtp_host"],
smtp_port=int(settings_dict["smtp_port"]),
@@ -145,14 +145,14 @@ async def get_smtp_settings(db: AsyncSession) -> SMTPSettings | None:
async def save_smtp_settings(db: AsyncSession, smtp_settings: SMTPSettings) -> None:
"""Save SMTP settings to database.
Args:
db: Database session
smtp_settings: SMTP settings to save
"""
from sqlalchemy import func
from sqlalchemy.dialects.sqlite import insert as sqlite_insert
settings_data = {
"smtp_host": smtp_settings.smtp_host,
"smtp_port": str(smtp_settings.smtp_port),
@@ -161,15 +161,15 @@ async def save_smtp_settings(db: AsyncSession, smtp_settings: SMTPSettings) -> N
"smtp_from_email": smtp_settings.smtp_from_email,
"smtp_from_name": smtp_settings.smtp_from_name,
}
# Only save username if auth is enabled or if provided
if smtp_settings.smtp_username:
settings_data["smtp_username"] = smtp_settings.smtp_username
# Only save password if provided
if smtp_settings.smtp_password:
settings_data["smtp_password"] = smtp_settings.smtp_password
for key, value in settings_data.items():
stmt = sqlite_insert(Settings).values(key=key, value=value)
stmt = stmt.on_conflict_do_update(
@@ -187,14 +187,14 @@ def send_email(
body_html: str | None = None,
) -> None:
"""Send an email using SMTP.
Args:
smtp_settings: SMTP configuration
to_email: Recipient email address
subject: Email subject
body_text: Plain text body
body_html: Optional HTML body
Raises:
Exception: If email sending fails
"""
@@ -202,24 +202,24 @@ def send_email(
msg["From"] = f"{smtp_settings.smtp_from_name} <{smtp_settings.smtp_from_email}>"
msg["To"] = to_email
msg["Subject"] = subject
# Attach plain text part
msg.attach(MIMEText(body_text, "plain"))
# Attach HTML part if provided
if body_html:
msg.attach(MIMEText(body_html, "html"))
# Send email
try:
security = smtp_settings.smtp_security
auth_enabled = smtp_settings.smtp_auth_enabled
# Validate username is provided when authentication is enabled
if auth_enabled and smtp_settings.smtp_password:
if not smtp_settings.smtp_username:
raise ValueError("SMTP username is required when authentication is enabled")
if security == "ssl":
# Direct SSL connection (typically port 465)
with smtplib.SMTP_SSL(smtp_settings.smtp_host, smtp_settings.smtp_port, timeout=10) as server:
@@ -247,17 +247,17 @@ def send_email(
def create_welcome_email(username: str, password: str, login_url: str) -> tuple[str, str, str]:
"""Create welcome email content for new user.
Args:
username: Username of the new user
password: Auto-generated password
login_url: URL to login page
Returns:
Tuple of (subject, text_body, html_body)
"""
subject = "Welcome to BamBuddy - Your Account Details"
text_body = f"""Welcome to BamBuddy!
Your account has been created. Here are your login details:
@@ -272,7 +272,7 @@ For security reasons, please change your password after your first login.
Best regards,
BamBuddy Team
"""
html_body = f"""<!DOCTYPE html>
<html>
<head>
@@ -285,20 +285,20 @@ BamBuddy Team
</div>
<div style="background: #f9f9f9; padding: 30px; border-radius: 0 0 8px 8px; border: 1px solid #ddd; border-top: none;">
<p style="font-size: 16px;">Your account has been created. Here are your login details:</p>
<div style="background: white; padding: 20px; border-radius: 4px; margin: 20px 0; border-left: 4px solid #667eea;">
<p style="margin: 0 0 10px 0;"><strong>Username:</strong> <code style="background: #f0f0f0; padding: 2px 6px; border-radius: 3px;">{username}</code></p>
<p style="margin: 0;"><strong>Password:</strong> <code style="background: #f0f0f0; padding: 2px 6px; border-radius: 3px;">{password}</code></p>
</div>
<div style="text-align: center; margin: 30px 0;">
<a href="{login_url}" style="display: inline-block; background: #667eea; color: white; padding: 12px 30px; text-decoration: none; border-radius: 4px; font-weight: bold;">Login Now</a>
</div>
<p style="font-size: 14px; color: #666; border-top: 1px solid #ddd; padding-top: 20px; margin-top: 20px;">
<strong>Security Note:</strong> For security reasons, please change your password after your first login.
</p>
<p style="font-size: 14px; color: #999; margin-top: 30px;">
Best regards,<br>
BamBuddy Team
@@ -307,23 +307,23 @@ BamBuddy Team
</body>
</html>
"""
return subject, text_body, html_body
def create_password_reset_email(username: str, password: str, login_url: str) -> tuple[str, str, str]:
"""Create password reset email content.
Args:
username: Username of the user
password: New auto-generated password
login_url: URL to login page
Returns:
Tuple of (subject, text_body, html_body)
"""
subject = "BamBuddy - Your Password Has Been Reset"
text_body = f"""Your BamBuddy password has been reset.
Your login details:
@@ -340,7 +340,7 @@ If you did not request this password reset, please contact your administrator im
Best regards,
BamBuddy Team
"""
html_body = f"""<!DOCTYPE html>
<html>
<head>
@@ -353,26 +353,26 @@ BamBuddy Team
</div>
<div style="background: #f9f9f9; padding: 30px; border-radius: 0 0 8px 8px; border: 1px solid #ddd; border-top: none;">
<p style="font-size: 16px;">Your BamBuddy password has been reset.</p>
<div style="background: white; padding: 20px; border-radius: 4px; margin: 20px 0; border-left: 4px solid #667eea;">
<p style="margin: 0 0 10px 0;"><strong>Username:</strong> <code style="background: #f0f0f0; padding: 2px 6px; border-radius: 3px;">{username}</code></p>
<p style="margin: 0;"><strong>New Password:</strong> <code style="background: #f0f0f0; padding: 2px 6px; border-radius: 3px;">{password}</code></p>
</div>
<div style="text-align: center; margin: 30px 0;">
<a href="{login_url}" style="display: inline-block; background: #667eea; color: white; padding: 12px 30px; text-decoration: none; border-radius: 4px; font-weight: bold;">Login Now</a>
</div>
<div style="background: #fff3cd; border: 1px solid #ffc107; border-radius: 4px; padding: 15px; margin: 20px 0;">
<p style="margin: 0; font-size: 14px; color: #856404;">
<strong>⚠️ Security Alert:</strong> If you did not request this password reset, please contact your administrator immediately.
</p>
</div>
<p style="font-size: 14px; color: #666; border-top: 1px solid #ddd; padding-top: 20px; margin-top: 20px;">
<strong>Security Note:</strong> For security reasons, please change your password after logging in.
</p>
<p style="font-size: 14px; color: #999; margin-top: 30px;">
Best regards,<br>
BamBuddy Team
@@ -381,7 +381,7 @@ BamBuddy Team
</body>
</html>
"""
return subject, text_body, html_body
@@ -389,20 +389,20 @@ async def create_welcome_email_from_template(
db: AsyncSession, username: str, password: str, login_url: str, app_name: str = "BamBuddy"
) -> tuple[str, str, str]:
"""Create welcome email content using notification template from database.
Args:
db: Database session
username: Username of the new user
password: Auto-generated password
login_url: URL to login page
app_name: Application name (default: BamBuddy)
Returns:
Tuple of (subject, text_body, html_body)
"""
# Try to get template from database
template = await get_notification_template(db, "user_created")
if template:
# Render template with variables
variables = {
@@ -411,10 +411,10 @@ async def create_welcome_email_from_template(
"password": password,
"login_url": login_url,
}
subject = render_template(template.title_template, variables)
text_body = render_template(template.body_template, variables)
# Create HTML version with embedded login button
html_body = f"""<!DOCTYPE html>
<html>
@@ -428,7 +428,7 @@ async def create_welcome_email_from_template(
</div>
<div style="background: #f9f9f9; padding: 30px; border-radius: 0 0 8px 8px; border: 1px solid #ddd; border-top: none;">
<div style="white-space: pre-wrap; font-size: 16px;">{text_body}</div>
<div style="text-align: center; margin: 30px 0;">
<a href="{login_url}" style="display: inline-block; background: #667eea; color: white; padding: 12px 30px; text-decoration: none; border-radius: 4px; font-weight: bold;">Login Now</a>
</div>
@@ -436,7 +436,7 @@ async def create_welcome_email_from_template(
</body>
</html>
"""
logger.info("Using custom welcome email template from database")
return subject, text_body, html_body
else:
@@ -449,20 +449,20 @@ async def create_password_reset_email_from_template(
db: AsyncSession, username: str, password: str, login_url: str, app_name: str = "BamBuddy"
) -> tuple[str, str, str]:
"""Create password reset email content using notification template from database.
Args:
db: Database session
username: Username of the user
password: New auto-generated password
login_url: URL to login page
app_name: Application name (default: BamBuddy)
Returns:
Tuple of (subject, text_body, html_body)
"""
# Try to get template from database
template = await get_notification_template(db, "password_reset")
if template:
# Render template with variables
variables = {
@@ -471,10 +471,10 @@ async def create_password_reset_email_from_template(
"password": password,
"login_url": login_url,
}
subject = render_template(template.title_template, variables)
text_body = render_template(template.body_template, variables)
# Create HTML version with embedded login button
html_body = f"""<!DOCTYPE html>
<html>
@@ -488,11 +488,11 @@ async def create_password_reset_email_from_template(
</div>
<div style="background: #f9f9f9; padding: 30px; border-radius: 0 0 8px 8px; border: 1px solid #ddd; border-top: none;">
<div style="white-space: pre-wrap; font-size: 16px;">{text_body}</div>
<div style="text-align: center; margin: 30px 0;">
<a href="{login_url}" style="display: inline-block; background: #667eea; color: white; padding: 12px 30px; text-decoration: none; border-radius: 4px; font-weight: bold;">Login Now</a>
</div>
<div style="background: #fff3cd; border: 1px solid #ffc107; border-radius: 4px; padding: 15px; margin: 20px 0;">
<p style="margin: 0; font-size: 14px; color: #856404;">
<strong>⚠️ Security Alert:</strong> If you did not request this password reset, please contact your administrator immediately.
@@ -502,7 +502,7 @@ async def create_password_reset_email_from_template(
</body>
</html>
"""
logger.info("Using custom password reset email template from database")
return subject, text_body, html_body
else: