from __future__ import absolute_import from django.dispatch import receiver from django.contrib.auth.signals import user_logged_in from django.core.mail import send_mail from django.conf import settings from django.template import loader from django.utils import timezone from typing import Any, Dict, Optional from zerver.models import UserProfile def get_device_browser(user_agent): # type: (str) -> Optional[str] user_agent = user_agent.lower() if "chrome" in user_agent and "chromium" not in user_agent: return 'Chrome' elif "firefox" in user_agent and "seamonkey" not in user_agent and "chrome" not in user_agent: return "Firefox" elif "chromium" in user_agent: return "Chromium" elif "safari" in user_agent and "chrome" not in user_agent and "chromium" not in user_agent: return "Safari" elif "opera" in user_agent: return "Opera" elif "msie" in user_agent or "trident" in user_agent: return "Internet Explorer" elif "edge" in user_agent: return "Edge" else: return None def get_device_os(user_agent): # type: (str) -> Optional[str] user_agent = user_agent.lower() if "windows" in user_agent: return "Windows" elif "macintosh" in user_agent: return "MacOS" elif "linux" in user_agent and "android" not in user_agent: return "Linux" elif "android" in user_agent: return "Android" elif "like mac os x" in user_agent: return "iOS" else: return None @receiver(user_logged_in, dispatch_uid="only_on_login") def email_on_new_login(sender, user, request, **kwargs): # type: (Any, UserProfile, Any, Any) -> None # We import here to minimize the dependencies of this module, # since it runs as part of `manage.py` initialization from zerver.context_processors import common_context if not settings.SEND_LOGIN_EMAILS: return if request: # Login emails are for returning users, not new registrations. # Determine if login request was from new registration. path = request.META.get('PATH_INFO', None) if path: if path == "/accounts/register/": return login_time = timezone.now().strftime('%A, %B %d, %Y at %I:%M%p ') + \ timezone.get_current_timezone_name() user_agent = request.META.get('HTTP_USER_AGENT', "").lower() device_browser = get_device_browser(user_agent) device_os = get_device_os(user_agent) device_ip = request.META.get('REMOTE_ADDR') or "Uknown IP address" device_info = {"device_browser": device_browser, "device_os": device_os, "device_ip": device_ip, "login_time": login_time } context = common_context(user) context['device_info'] = device_info context['zulip_support'] = settings.ZULIP_ADMINISTRATOR context['user'] = user text_template = 'zerver/emails/new_login/new_login_alert.txt' html_template = 'zerver/emails/new_login/new_login_alert.html' text_content = loader.render_to_string(text_template, context) html_content = loader.render_to_string(html_template, context) sender = settings.NOREPLY_EMAIL_ADDRESS recipients = [user.email] subject = loader.render_to_string('zerver/emails/new_login/new_login_alert.subject').strip() send_mail(subject, text_content, sender, recipients, html_message=html_content)