mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-16 00:14:18 -05:00
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
from app import events_api_client
|
|
from flask_login import current_user
|
|
|
|
|
|
def on_user_logged_in(sender, user):
|
|
_send_event(sender, event_type='sucessful_login', user=user)
|
|
|
|
|
|
# should change the event type? This is a remember me login.
|
|
def on_user_login_confirmed(sender):
|
|
_send_event(sender, event_type='sucessful_login_remember_me', user=current_user)
|
|
|
|
|
|
def _send_event(sender, **kwargs):
|
|
from flask import request
|
|
try:
|
|
event_data = _construct_event_data(request)
|
|
if kwargs.get('user'):
|
|
event_data['user_id'] = kwargs.get('user').id
|
|
if not kwargs.get('event_type'):
|
|
return
|
|
events_api_client.create_event(kwargs['event_type'], event_data)
|
|
except Exception as e:
|
|
sender.logger.error('Error creating event')
|
|
sender.logger.error(e)
|
|
|
|
|
|
def _construct_event_data(request):
|
|
return {'ip_address': _get_remote_addr(request),
|
|
'browser_fingerprint': _get_browser_fingerprint(request)}
|
|
|
|
|
|
# This might not be totally correct depending on proxy setup
|
|
def _get_remote_addr(request):
|
|
if request.headers.getlist("X-Forwarded-For"):
|
|
return request.headers.getlist("X-Forwarded-For")[0]
|
|
else:
|
|
return request.remote_addr
|
|
|
|
|
|
def _get_browser_fingerprint(request):
|
|
browser = request.user_agent.browser
|
|
version = request.user_agent.version
|
|
platform = request.user_agent.platform
|
|
user_agent_string = request.user_agent.string
|
|
# at some point this may be hashed?
|
|
finger_print = {'browser': browser,
|
|
'platform': platform,
|
|
'version': version,
|
|
'user_agent_string': user_agent_string}
|
|
|
|
return finger_print
|