95 lines
3.6 KiB
Python
95 lines
3.6 KiB
Python
from django.db import models
|
|
from django.contrib.contenttypes.fields import GenericForeignKey
|
|
from django.contrib.contenttypes.models import ContentType
|
|
import uuid
|
|
|
|
|
|
class KYCRequest(models.Model):
|
|
"""Главная модель KYC заявки. Ссылается на детали страны через ContentType."""
|
|
WORKFLOW_STATUS_CHOICES = [
|
|
('pending', 'Ожидает обработки'),
|
|
('active', 'Активен'),
|
|
('error', 'Ошибка'),
|
|
]
|
|
|
|
uuid = models.CharField(max_length=100, unique=True, default=uuid.uuid4)
|
|
user_id = models.CharField(max_length=255, db_index=True)
|
|
team_name = models.CharField(max_length=200, blank=True)
|
|
country_code = models.CharField(max_length=2, blank=True)
|
|
workflow_status = models.CharField(
|
|
max_length=20,
|
|
choices=WORKFLOW_STATUS_CHOICES,
|
|
default='pending',
|
|
)
|
|
score = models.IntegerField(default=0)
|
|
|
|
# Общие контактные данные
|
|
contact_person = models.CharField(max_length=255, blank=True, default='')
|
|
contact_email = models.EmailField(blank=True, default='')
|
|
contact_phone = models.CharField(max_length=50, blank=True, default='')
|
|
|
|
# Ссылка на детали страны через ContentType
|
|
content_type = models.ForeignKey(
|
|
ContentType,
|
|
on_delete=models.SET_NULL,
|
|
null=True,
|
|
blank=True,
|
|
related_name='kyc_requests'
|
|
)
|
|
object_id = models.PositiveIntegerField(null=True, blank=True)
|
|
country_details = GenericForeignKey('content_type', 'object_id')
|
|
|
|
approved_by = models.CharField(max_length=255, null=True, blank=True)
|
|
approved_at = models.DateTimeField(null=True, blank=True)
|
|
created_at = models.DateTimeField(auto_now_add=True)
|
|
updated_at = models.DateTimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
db_table = 'kyc_requests'
|
|
|
|
def __str__(self):
|
|
return f"KYC {self.user_id} - {self.workflow_status}"
|
|
|
|
def get_country_data(self) -> dict:
|
|
"""Получить данные страны как словарь для Temporal workflow."""
|
|
if self.country_details and hasattr(self.country_details, 'to_dict'):
|
|
return self.country_details.to_dict()
|
|
return {}
|
|
|
|
|
|
class KYCRequestRussia(models.Model):
|
|
"""Детали KYC для России. Отдельная модель без наследования."""
|
|
|
|
# Данные компании от DaData
|
|
company_name = models.CharField(max_length=255)
|
|
company_full_name = models.TextField()
|
|
inn = models.CharField(max_length=12)
|
|
kpp = models.CharField(max_length=9, blank=True)
|
|
ogrn = models.CharField(max_length=15, blank=True)
|
|
address = models.TextField()
|
|
|
|
# Банковские реквизиты
|
|
bank_name = models.CharField(max_length=255)
|
|
bik = models.CharField(max_length=9)
|
|
correspondent_account = models.CharField(max_length=20, blank=True)
|
|
|
|
class Meta:
|
|
db_table = 'kyc_details_russia'
|
|
|
|
def __str__(self):
|
|
return f"KYC Russia: {self.company_name} (ИНН: {self.inn})"
|
|
|
|
def to_dict(self) -> dict:
|
|
"""Конвертировать в словарь для передачи в Temporal workflow."""
|
|
return {
|
|
"company_name": self.company_name,
|
|
"company_full_name": self.company_full_name,
|
|
"inn": self.inn,
|
|
"kpp": self.kpp,
|
|
"ogrn": self.ogrn,
|
|
"address": self.address,
|
|
"bank_name": self.bank_name,
|
|
"bik": self.bik,
|
|
"correspondent_account": self.correspondent_account,
|
|
}
|