fix: add otp fields

ocadotechnology · Sep 27, 2023 · 634d13e · 634d13e
1 parent 1d04dce
commit 634d13e
Show file tree

Hide file tree

Showing 4 changed files with 109 additions and 92 deletions.
diff --git a/Pipfile.lock b/Pipfile.lock
diff --git a/...mon/migrations/0045_auto_20230927_1004.py → ...mon/migrations/0045_auto_20230927_1223.py b/...mon/migrations/0045_auto_20230927_1004.py → ...mon/migrations/0045_auto_20230927_1223.py
@@ -1,8 +1,9 @@
-# Generated by Django 3.2.20 on 2023-09-27 09:04
+# Generated by Django 3.2.20 on 2023-09-27 11:23
 
 from django.conf import settings
 from django.db import migrations, models
 import django.db.models.deletion
+import pyotp
 
 
 class Migration(migrations.Migration):
@@ -15,8 +16,13 @@ class Migration(migrations.Migration):
     operations = [
         migrations.AddField(
             model_name='userprofile',
-            name='enabled_auth_factor_csv',
-            field=models.TextField(default=''),
+            name='last_otp_for_time',
+            field=models.DateTimeField(null=True),
+        ),
+        migrations.AddField(
+            model_name='userprofile',
+            name='otp_secret',
+            field=models.CharField(default=pyotp.random_base32, max_length=40),
         ),
         migrations.AlterField(
             model_name='userprofile',

diff --git a/cfl_common/common/models.py b/cfl_common/common/models.py
@@ -3,6 +3,7 @@
 from uuid import uuid4
 
 import pgeocode
+import pyotp
 from django.contrib.auth.models import User
 from django.db import models
 from django.utils import timezone
@@ -18,7 +19,8 @@ class UserProfile(models.Model):
         on_delete=models.CASCADE,
     )
 
-    enabled_auth_factor_csv = models.TextField(default="")
+    otp_secret = models.CharField(max_length=40, default=pyotp.random_base32)
+    last_otp_for_time = models.DateTimeField(null=True)
     developer = models.BooleanField(default=False)
     is_verified = models.BooleanField(default=False)
 

diff --git a/cfl_common/setup.py b/cfl_common/setup.py
@@ -5,7 +5,9 @@
 
 with open("../portal/__init__.py", "r") as fd:
     version = re.search(
-        r'^__version__\s*=\s*[\'"]([^\'"]*)[\'"]', fd.read(), re.MULTILINE
+        r'^__version__\s*=\s*[\'"]([^\'"]*)[\'"]',
+        fd.read(),
+        re.MULTILINE,
     ).group(1)
 
 setup(
@@ -20,6 +22,7 @@
         "django-countries==7.3.1",
         "pyjwt==2.6.0",
         "pgeocode==0.4.0",
+        "pyotp==2.9.0",
     ],
     tests_require=[],
     test_suite="tests",