59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
"""
|
|
Tenant-aware middleware for multi-tenancy
|
|
"""
|
|
from django.utils.deprecation import MiddlewareMixin
|
|
|
|
|
|
class TenantMiddleware(MiddlewareMixin):
|
|
"""
|
|
Middleware that sets the current hospital context from the authenticated user.
|
|
|
|
This middleware ensures that:
|
|
- authenticated users have their tenant_hospital set from their profile
|
|
- PX admins can switch between hospitals via session
|
|
- Source Users have their source context available
|
|
- All requests have tenant context available
|
|
"""
|
|
|
|
def process_request(self, request):
|
|
"""Set tenant hospital context on each request."""
|
|
if request.user and request.user.is_authenticated:
|
|
# Store user's role for quick access
|
|
request.user_roles = request.user.get_role_names()
|
|
|
|
# Set source user context
|
|
request.source_user = None
|
|
request.source_user_profile = None
|
|
if request.user.is_source_user():
|
|
profile = request.user.get_source_user_profile_active()
|
|
if profile:
|
|
request.source_user = profile
|
|
request.source_user_profile = profile
|
|
|
|
# PX Admins can switch hospitals via session
|
|
if request.user.is_px_admin():
|
|
hospital_id = request.session.get('selected_hospital_id')
|
|
if hospital_id:
|
|
from apps.organizations.models import Hospital
|
|
try:
|
|
# Validate that the hospital exists
|
|
request.tenant_hospital = Hospital.objects.get(id=hospital_id)
|
|
except Hospital.DoesNotExist:
|
|
# Invalid hospital ID, fall back to default
|
|
request.tenant_hospital = None
|
|
# Clear invalid session data
|
|
request.session.pop('selected_hospital_id', None)
|
|
else:
|
|
# No hospital selected yet
|
|
request.tenant_hospital = None
|
|
else:
|
|
# Non-PX Admin users use their assigned hospital
|
|
request.tenant_hospital = request.user.hospital
|
|
else:
|
|
request.tenant_hospital = None
|
|
request.user_roles = []
|
|
request.source_user = None
|
|
request.source_user_profile = None
|
|
|
|
return None
|