132 lines
4.7 KiB
Python
132 lines
4.7 KiB
Python
from django.db import models
|
||
from django.contrib.contenttypes.fields import GenericForeignKey
|
||
from django.contrib.contenttypes.models import ContentType
|
||
import uuid
|
||
|
||
|
||
class AbstractKycBase(models.Model):
|
||
"""Общая база для KYC сущностей с идентичными полями."""
|
||
|
||
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='%(class)s_set'
|
||
)
|
||
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:
|
||
abstract = True
|
||
|
||
|
||
class KYCApplication(AbstractKycBase):
|
||
"""KYC заявка на верификацию. Одноразовый процесс."""
|
||
|
||
class Meta:
|
||
db_table = 'kyc_requests' # Сохраняем для совместимости
|
||
verbose_name = 'KYC Application'
|
||
verbose_name_plural = 'KYC Applications'
|
||
|
||
def __str__(self):
|
||
return f"KYC Application {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 KYCProfile(AbstractKycBase):
|
||
"""KYC профиль компании для долгосрочного мониторинга.
|
||
|
||
Создается после успешной верификации (approval) заявки.
|
||
Используется Dagster для сбора данных о компании.
|
||
"""
|
||
|
||
class Meta:
|
||
db_table = 'kyc_monitoring' # Сохраняем для совместимости
|
||
verbose_name = 'KYC Profile'
|
||
verbose_name_plural = 'KYC Profiles'
|
||
|
||
def __str__(self):
|
||
return f"KYC Profile {self.user_id} - {self.workflow_status}"
|
||
|
||
|
||
# Aliases for backwards compatibility
|
||
KYCRequest = KYCApplication
|
||
KYCMonitoring = KYCProfile
|
||
|
||
|
||
class KYCDetailsRussia(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'
|
||
verbose_name = 'KYC Details Russia'
|
||
verbose_name_plural = '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,
|
||
}
|
||
|
||
|
||
# Alias for backwards compatibility
|
||
KYCRequestRussia = KYCDetailsRussia
|