husanjon
ab73d05ecc
Some checks failed
Build and Push to Docker Hub / build-test-push (push) Failing after 1m55s
111 lines
3.2 KiB
Python
111 lines
3.2 KiB
Python
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}"
|