from django.db import models
from apps.utils.models import CommonBDModel, BaseModel

# Create your models here.
class Work(CommonBDModel):
    name = models.CharField('名称', max_length=100)
    type = models.CharField('类型', max_length=50, help_text="项目类型: 项目/企业")
    description = models.TextField('描述', blank=True, null=True)
    score_dq = models.FloatField('贷前评分', null=True, blank=True)
    fingerprint_dq = models.TextField('贷前指纹', null=True, blank=True)
    score_dh = models.FloatField('贷后评分', null=True, blank=True)
    fingerprint_dh = models.TextField('贷后指纹', null=True, blank=True)
    # 企业低碳转型战略与总体规划
    dq_file1 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dq_file1')
    # 碳排放数据监测、核查与信息披露报告
    dq_file2 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dq_file2')
    # 环境、社会与治理（ESG）尽职调查报告
    dq_file3 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dq_file3')
    # 项目融资与可行性研究方案
    dq_file4 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dq_file4')
    # 数字化与智能控制系统技术方案
    dq_file5 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dq_file5')
    # 企业技术说明及改造方案
    dq_file6 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dq_file6')
    dh_file1 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dh_file1')
    dh_file2 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dh_file2')
    dh_file3 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dh_file3')
    dh_file4 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dh_file4')
    dh_file5 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dh_file5')
    dh_file6 = models.ForeignKey("system.file", on_delete=models.SET_NULL, null=True, blank=True, related_name='work_dh_file6')


class Fingerprint(BaseModel):
    fp_hex = models.CharField(
        max_length=16, unique=True, verbose_name="simhash(hex)"
    )
    fp_int = models.BigIntegerField(
        db_index=True, verbose_name="simhash(int)"
    )

    seg1 = models.IntegerField(db_index=True)
    seg2 = models.IntegerField(db_index=True)
    seg3 = models.IntegerField(db_index=True)
    seg4 = models.IntegerField(db_index=True)

    score = models.FloatField(verbose_name="得分")

    class Meta:
        indexes = [
            models.Index(fields=["seg1"]),
            models.Index(fields=["seg2"]),
            models.Index(fields=["seg3"]),
            models.Index(fields=["seg4"]),
        ]