From f5951e966ccc40fc6cc08bcd52cccffe9e259f60 Mon Sep 17 00:00:00 2001
From: kuldeepkhatke <kuldeepkhatke03@gmail.com>
Date: Sun, 8 Jun 2025 14:15:17 +0530
Subject: [PATCH] Modified access_attempt as pk

---
 .../0010_accessattemptexpiration.py           | 21 +++++++------------
 axes/models.py                                |  1 +
 2 files changed, 8 insertions(+), 14 deletions(-)

diff --git a/axes/migrations/0010_accessattemptexpiration.py b/axes/migrations/0010_accessattemptexpiration.py
index 509a154..ef7e5a6 100644
--- a/axes/migrations/0010_accessattemptexpiration.py
+++ b/axes/migrations/0010_accessattemptexpiration.py
@@ -1,4 +1,4 @@
-# Generated by Django 5.2.1 on 2025-06-07 17:37
+# Generated by Django 5.2.1 on 2025-06-08 12:44
 
 import django.db.models.deletion
 from django.db import migrations, models
@@ -15,12 +15,14 @@ class Migration(migrations.Migration):
             name="AccessAttemptExpiration",
             fields=[
                 (
-                    "id",
-                    models.AutoField(
-                        auto_created=True,
+                    "access_attempt",
+                    models.OneToOneField(
+                        on_delete=django.db.models.deletion.CASCADE,
                         primary_key=True,
+                        related_name="expiration",
                         serialize=False,
-                        verbose_name="ID",
+                        to="axes.accessattempt",
+                        verbose_name="Access Attempt",
                     ),
                 ),
                 (
@@ -32,15 +34,6 @@ class Migration(migrations.Migration):
                         verbose_name="Expires At",
                     ),
                 ),
-                (
-                    "access_attempt",
-                    models.OneToOneField(
-                        on_delete=django.db.models.deletion.CASCADE,
-                        related_name="expiration",
-                        to="axes.accessattempt",
-                        verbose_name="Access Attempt",
-                    ),
-                ),
             ],
             options={
                 "verbose_name": "access attempt expiration",
diff --git a/axes/models.py b/axes/models.py
index 26f4857..3c51c6e 100644
--- a/axes/models.py
+++ b/axes/models.py
@@ -54,6 +54,7 @@ class AccessAttempt(AccessBase):
 class AccessAttemptExpiration(models.Model):
     access_attempt = models.OneToOneField(
         AccessAttempt,
+        primary_key=True,
         on_delete=models.CASCADE,
         related_name="expiration",
         verbose_name=_("Access Attempt"),