gold eggs backend

core/http/models/user.py

@@ -0,0 +1,110 @@
+import math
+from datetime import datetime, timedelta
+
+from django.contrib.auth import models as auth_models
+from django.db import models
+from django.utils.translation import gettext_lazy as _
+
+from core.http import managers
+
+
+class User(auth_models.AbstractUser):
+    ROLE_CHOICES = (
+        ("admin", _("Admin")),
+        ("market", _("Market")),
+        ("courier", _("Courier")),
+        ("sklad", _("Sklad")),
+    )
+
+    phone = models.CharField(max_length=255, unique=True)
+    username = models.CharField(max_length=255, null=True, blank=True)
+    created_at = models.DateTimeField(auto_now_add=True)
+    updated_at = models.DateTimeField(auto_now=True)
+    validated_at = models.DateTimeField(null=True, blank=True)
+    USERNAME_FIELD = "phone"
+    objects = managers.UserManager()
+
+    avatar = models.ImageField(
+        upload_to="avatars/",
+        null=True,
+        blank=True,
+        default="avatars/golden_eggs.png",
+    )
+    role = models.CharField(
+        max_length=10,
+        choices=ROLE_CHOICES,
+        default="admin",
+    )
+    fcm_token = models.CharField(max_length=255, null=True, blank=True)
+
+    def __str__(self):
+        return f"{self.phone} | {self.id}"
+
+
+class SmsConfirm(models.Model):
+    SMS_EXPIRY_SECONDS = 120
+    RESEND_BLOCK_MINUTES = 10
+    TRY_BLOCK_MINUTES = 2
+    RESEND_COUNT = 5
+    TRY_COUNT = 10
+
+    code = models.IntegerField()
+    try_count = models.IntegerField(default=0)
+    resend_count = models.IntegerField(default=0)
+    phone = models.CharField(max_length=20)
+    expire_time = models.DateTimeField(null=True, blank=True)
+    unlock_time = models.DateTimeField(null=True, blank=True)
+    resend_unlock_time = models.DateTimeField(null=True, blank=True)
+
+    def sync_limits(self):
+        if self.resend_count >= self.RESEND_COUNT:
+            self.try_count = 0
+            self.resend_count = 0
+            self.resend_unlock_time = datetime.now() + timedelta(
+                minutes=self.RESEND_BLOCK_MINUTES
+            )
+        elif self.try_count >= self.TRY_COUNT:
+            self.try_count = 0
+            self.unlock_time = datetime.now() + timedelta(
+                minutes=self.TRY_BLOCK_MINUTES
+            )
+
+        if (
+            self.resend_unlock_time is not None
+            and self.resend_unlock_time.timestamp()
+            < datetime.now().timestamp()
+        ):
+            self.resend_unlock_time = None
+
+        if (
+            self.unlock_time is not None
+            and self.unlock_time.timestamp() < datetime.now().timestamp()
+        ):
+            self.unlock_time = None
+        self.save()
+
+    def is_expired(self):
+        return (
+            self.expire_time.timestamp() < datetime.now().timestamp()
+            if hasattr(self.expire_time, "timestamp")
+            else None
+        )
+
+    def is_block(self):
+        return self.unlock_time is not None
+
+    def reset_limits(self):
+        self.try_count = 0
+        self.resend_count = 0
+        self.unlock_time = None
+
+    def interval(self, time):
+        expire = time.timestamp() - datetime.now().timestamp()
+        minutes = math.floor(expire / 60)
+        expire -= minutes * 60
+        expire = math.floor(expire)
+
+        return f"{minutes:02d}:{expire:02d}"
+
+    def __str__(self) -> str:
+        return f"{self.phone} | {self.code}"