from flask import request from app.notify_client.events_api_client import events_api_client EVENT_SCHEMAS = { "sucessful_login": {"user_id"}, "update_user_email": {"user_id", "updated_by_id", "original_email_address", "new_email_address"}, "update_user_mobile_number": {"user_id", "updated_by_id", "original_mobile_number", "new_mobile_number"}, "remove_user_from_service": {"user_id", "removed_by_id", "service_id"}, "add_user_to_service": {"user_id", "invited_by_id", "service_id", "ui_permissions"}, "invite_user_to_service": {"email_address", "invited_by_id", "service_id", "ui_permissions"}, "cancel_user_invite_to_service": {"email_address", "canceled_by_id", "service_id"}, "set_user_permissions": {"user_id", "service_id", "original_ui_permissions", "new_ui_permissions", "set_by_id"}, "archive_user": {"user_id", "archived_by_id"}, "archive_service": {"service_id", "archived_by_id"}, "suspend_service": {"service_id", "suspended_by_id"}, "resume_service": {"service_id", "resumed_by_id"}, } def on_user_logged_in(_sender, user): _send_event('sucessful_login', user_id=user.id) def create_email_change_event(**kwargs): _send_event('update_user_email', **kwargs) def create_mobile_number_change_event(**kwargs): _send_event('update_user_mobile_number', **kwargs) def create_remove_user_from_service_event(**kwargs): _send_event('remove_user_from_service', **kwargs) def create_invite_user_to_service_event(**kwargs): _send_event('invite_user_to_service', **kwargs) def create_cancel_user_invite_to_service_event(**kwargs): _send_event('cancel_user_invite_to_service', **kwargs) def create_add_user_to_service_event(**kwargs): _send_event('add_user_to_service', **kwargs) def create_set_user_permissions_event(**kwargs): _send_event('set_user_permissions', **kwargs) def create_archive_user_event(**kwargs): _send_event('archive_user', **kwargs) def create_suspend_service_event(**kwargs): _send_event('suspend_service', **kwargs) def create_archive_service_event(**kwargs): _send_event('archive_service', **kwargs) def create_resume_service_event(**kwargs): _send_event('resume_service', **kwargs) def _send_event(event_type, **kwargs): expected_keys = EVENT_SCHEMAS[event_type] actual_keys = set(kwargs.keys()) if expected_keys != actual_keys: raise ValueError(f'Expected {expected_keys}, but got {actual_keys}') event_data = _construct_event_data(request) event_data.update(kwargs) events_api_client.create_event(event_type, event_data) 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