husanjon
ab73d05ecc
Some checks failed
Build and Push to Docker Hub / build-test-push (push) Failing after 1m55s
67 lines
2.0 KiB
Python
Executable File
67 lines
2.0 KiB
Python
Executable File
from datetime import datetime
|
|
from datetime import timedelta
|
|
|
|
from core import exceptions
|
|
from core.http import models
|
|
from core.http import tasks
|
|
|
|
|
|
class SmsService:
|
|
@staticmethod
|
|
def send_confirm(phone):
|
|
# TODO: Deploy this change when deploying -> code = random.randint(1000, 9999) # noqa
|
|
code = 1111
|
|
|
|
sms_confirm, status = models.SmsConfirm.objects.get_or_create(
|
|
phone=phone, defaults={"code": code}
|
|
)
|
|
|
|
sms_confirm.sync_limits()
|
|
|
|
if sms_confirm.resend_unlock_time is not None:
|
|
expired = sms_confirm.interval(sms_confirm.resend_unlock_time)
|
|
exception = exceptions.SmsException(
|
|
f"Resend blocked, try again in {expired}", expired=expired
|
|
)
|
|
raise exception
|
|
|
|
sms_confirm.code = code
|
|
sms_confirm.try_count = 0
|
|
sms_confirm.resend_count += 1
|
|
sms_confirm.phone = phone
|
|
sms_confirm.expired_time = datetime.now() + timedelta(
|
|
seconds=models.SmsConfirm.SMS_EXPIRY_SECONDS
|
|
) # noqa
|
|
sms_confirm.resend_unlock_time = datetime.now() + timedelta(
|
|
seconds=models.SmsConfirm.SMS_EXPIRY_SECONDS
|
|
) # noqa
|
|
sms_confirm.save()
|
|
|
|
tasks.SendConfirm.delay(phone, code)
|
|
return True
|
|
|
|
@staticmethod
|
|
def check_confirm(phone, code):
|
|
sms_confirm = models.SmsConfirm.objects.filter(phone=phone).first()
|
|
|
|
if sms_confirm is None:
|
|
raise exceptions.SmsException("Invalid confirmation code")
|
|
|
|
sms_confirm.sync_limits()
|
|
|
|
if sms_confirm.is_expired():
|
|
raise exceptions.SmsException("Time for confirmation has expired")
|
|
|
|
if sms_confirm.is_block():
|
|
expired = sms_confirm.interval(sms_confirm.unlock_time)
|
|
raise exceptions.SmsException(f"Try again in {expired}")
|
|
|
|
if sms_confirm.code == code:
|
|
sms_confirm.delete()
|
|
return True
|
|
|
|
sms_confirm.try_count += 1
|
|
sms_confirm.save()
|
|
|
|
raise exceptions.SmsException("Invalid confirmation code")
|