fix: make REMOTE_ADDR optional in all cases

soynatan · Mar 14, 2024 · cafc7c0 · cafc7c0
1 parent ba825f8
commit cafc7c0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/easyaudit/signals/auth_signals.py b/easyaudit/signals/auth_signals.py
@@ -18,7 +18,7 @@ def user_logged_in(sender, request, user, **kwargs):
                 'login_type': LoginEvent.LOGIN,
                 'username': getattr(user, user.USERNAME_FIELD),
                 'user_id': getattr(user, 'id', None),
-                'remote_ip': request.META.get(REMOTE_ADDR_HEADER, "")
+                'remote_ip': request.META.get(REMOTE_ADDR_HEADER, '')
             })
     except Exception:
         if should_propagate_exceptions():
@@ -32,7 +32,7 @@ def user_logged_out(sender, request, user, **kwargs):
                 'login_type': LoginEvent.LOGOUT,
                 'username': getattr(user, user.USERNAME_FIELD),
                 'user_id': getattr(user, 'id', None),
-                'remote_ip': request.META[REMOTE_ADDR_HEADER]
+                'remote_ip': request.META.get(REMOTE_ADDR_HEADER, '')
             })
     except Exception:
         if should_propagate_exceptions():
@@ -47,7 +47,7 @@ def user_login_failed(sender, credentials, **kwargs):
             login_event = audit_logger.login({
                 'login_type': LoginEvent.FAILED,
                 'username': credentials[user_model.USERNAME_FIELD],
-                'remote_ip': request.META[REMOTE_ADDR_HEADER]
+                'remote_ip': request.META.get(REMOTE_ADDR_HEADER, '')
             })
     except Exception:
         if should_propagate_exceptions():