From 4adee5748c8069ddf537aab17d165014762cd7a2 Mon Sep 17 00:00:00 2001 From: Jim Driessen Date: Mon, 30 Jan 2017 03:57:53 +0100 Subject: [PATCH] Clear reset_token upon successful login. --- marietje/marietje/__init__.py | 1 + marietje/marietje/signals.py | 10 ++++++++++ 2 files changed, 11 insertions(+) diff --git a/marietje/marietje/__init__.py b/marietje/marietje/__init__.py index e69de29..756bffc 100644 --- a/marietje/marietje/__init__.py +++ b/marietje/marietje/__init__.py @@ -0,0 +1 @@ +import marietje.signals diff --git a/marietje/marietje/signals.py b/marietje/marietje/signals.py index e69de29..1480c54 100644 --- a/marietje/marietje/signals.py +++ b/marietje/marietje/signals.py @@ -0,0 +1,10 @@ +from django.contrib.auth.signals import user_logged_in + + +def on_login(sender, user, request, **kwargs): + if user.reset_token: + user.reset_token = None + user.save() + + +user_logged_in.connect(on_login)