From 51029b7bbac7d76a35bfe2f680e887c60aee6156 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Thu, 15 Aug 2024 10:42:12 -0500
Subject: Import specific classes from the datetime module.

---
 gn_auth/auth/authorisation/users/views.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'gn_auth/auth/authorisation')

diff --git a/gn_auth/auth/authorisation/users/views.py b/gn_auth/auth/authorisation/users/views.py
index 363da92..0922e1e 100644
--- a/gn_auth/auth/authorisation/users/views.py
+++ b/gn_auth/auth/authorisation/users/views.py
@@ -1,12 +1,12 @@
 """User authorisation endpoints."""
 import sqlite3
 import secrets
-import datetime
 import traceback
 from typing import Any
 from functools import partial
 from dataclasses import asdict
 from urllib.parse import urljoin
+from datetime import datetime, timedelta
 from email.headerregistry import Address
 from email_validator import validate_email, EmailNotValidError
 from flask import (
@@ -123,7 +123,7 @@ def send_verification_email(
     """Send an email verification message."""
     subject="GeneNetwork: Please Verify Your Email"
     verification_code = secrets.token_urlsafe(64)
-    generated = datetime.datetime.now()
+    generated = datetime.now()
     expiration_minutes = 15
     def __render__(template):
         return render_template(template,
@@ -148,7 +148,7 @@ def send_verification_email(
                 "generated": int(generated.timestamp()),
                 "expires": int(
                     (generated +
-                     datetime.timedelta(
+                     timedelta(
                          minutes=expiration_minutes)).timestamp())
             })
         send_message(smtp_user=current_app.config["SMTP_USER"],
@@ -236,8 +236,8 @@ def verify_user():
             return loginuri
 
         results = results[0]
-        if (datetime.datetime.fromtimestamp(
-                int(results["expires"])) < datetime.datetime.now()):
+        if (datetime.fromtimestamp(
+                int(results["expires"])) < datetime.now()):
             delete_verification_code(cursor, verificationcode)
             flash("Invalid verification code: code has expired.",
                   "alert-danger")
-- 
cgit v1.2.3