Verified Commit 8182a349 authored by Baptiste Beauplat's avatar Baptiste Beauplat
Browse files

Explicitly add _num_days and _today method to EmailChangeTokenGenerator (removed in django3)

parent 7d87726d
...@@ -26,6 +26,7 @@ ...@@ -26,6 +26,7 @@
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
# OTHER DEALINGS IN THE SOFTWARE. # OTHER DEALINGS IN THE SOFTWARE.
from datetime import date
from django.conf import settings from django.conf import settings
from django.utils.crypto import constant_time_compare, salted_hmac from django.utils.crypto import constant_time_compare, salted_hmac
...@@ -87,5 +88,12 @@ class EmailChangeTokenGenerator(PasswordResetTokenGenerator): ...@@ -87,5 +88,12 @@ class EmailChangeTokenGenerator(PasswordResetTokenGenerator):
str(login_timestamp) + \ str(login_timestamp) + \
str(timestamp) str(timestamp)
def _num_days(self, dt):
return (dt - date(2001, 1, 1)).days
def _today(self):
# Used for mocking in tests
return date.today() # pragma: no cover
email_change_token_generator = EmailChangeTokenGenerator() email_change_token_generator = EmailChangeTokenGenerator()
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment