| """Invoice recognition models.""" |
|
|
| from __future__ import annotations |
|
|
| from typing import ClassVar |
|
|
| from django.conf import settings |
| from django.db import models |
| from django.utils.translation import gettext_lazy as _ |
|
|
|
|
| class InvoiceCategory(models.TextChoices): |
| """发票类目""" |
|
|
| VAT_SPECIAL = "vat_special", _("增值税专用发票") |
| VAT_NORMAL = "vat_normal", _("增值税普通发票") |
| VAT_ELECTRONIC = "vat_electronic", _("增值税电子普通发票") |
| VEHICLE_SALES = "vehicle_sales", _("机动车销售统一发票") |
| TRAIN_TICKET = "train_ticket", _("火车票") |
| TAXI_RECEIPT = "taxi_receipt", _("出租车发票") |
| QUOTA_INVOICE = "quota_invoice", _("定额发票") |
| AIR_ITINERARY = "air_itinerary", _("飞机行程单") |
| TOLL_RECEIPT = "toll_receipt", _("过路费发票") |
| OTHER = "other", _("其他") |
|
|
|
|
| class InvoiceRecognitionTaskStatus(models.TextChoices): |
| """发票识别任务状态""" |
|
|
| PENDING = "pending", _("待处理") |
| PROCESSING = "processing", _("处理中") |
| COMPLETED = "completed", _("已完成") |
| FAILED = "failed", _("失败") |
|
|
|
|
| class InvoiceRecordStatus(models.TextChoices): |
| """发票记录识别状态""" |
|
|
| PENDING = "pending", _("待识别") |
| SUCCESS = "success", _("识别成功") |
| FAILED = "failed", _("识别失败") |
|
|
|
|
| class InvoiceRecognitionTask(models.Model): |
| """发票识别任务。""" |
|
|
| id: int |
| name = models.CharField(max_length=255, verbose_name=_("任务名称")) |
| status = models.CharField( |
| max_length=32, |
| choices=InvoiceRecognitionTaskStatus.choices, |
| default=InvoiceRecognitionTaskStatus.PENDING, |
| verbose_name=_("任务状态"), |
| ) |
| created_by = models.ForeignKey( |
| settings.AUTH_USER_MODEL, |
| on_delete=models.SET_NULL, |
| null=True, |
| blank=True, |
| related_name="+", |
| verbose_name=_("创建人"), |
| ) |
| created_at = models.DateTimeField(auto_now_add=True, verbose_name=_("创建时间")) |
| finished_at = models.DateTimeField(null=True, blank=True, verbose_name=_("完成时间")) |
| merge_config = models.JSONField(default=list, blank=True, verbose_name=_("分组合并配置")) |
|
|
| class Meta: |
| managed = False |
| db_table = "automation_invoicerecognitiontask" |
| ordering: ClassVar[list[str]] = ["-created_at"] |
| verbose_name = _("发票识别任务") |
| verbose_name_plural = _("发票识别任务") |
|
|
| def __str__(self) -> str: |
| return f"{self.name} ({self.get_status_display()})" |
|
|
|
|
| class InvoiceRecord(models.Model): |
| """发票记录。""" |
|
|
| id: int |
| task = models.ForeignKey( |
| InvoiceRecognitionTask, |
| on_delete=models.CASCADE, |
| related_name="records", |
| verbose_name=_("所属任务"), |
| ) |
| file_path = models.CharField(max_length=1024, verbose_name=_("文件路径")) |
| original_filename = models.CharField(max_length=255, verbose_name=_("原始文件名")) |
| invoice_code = models.CharField(max_length=50, blank=True, default="", verbose_name=_("发票代码")) |
| invoice_number = models.CharField(max_length=50, blank=True, default="", verbose_name=_("发票号码")) |
| invoice_date = models.DateField(null=True, blank=True, verbose_name=_("开票日期")) |
| amount = models.DecimalField( |
| max_digits=12, |
| decimal_places=2, |
| null=True, |
| blank=True, |
| verbose_name=_("金额（不含税）"), |
| ) |
| tax_amount = models.DecimalField(max_digits=12, decimal_places=2, null=True, blank=True, verbose_name=_("税额")) |
| total_amount = models.DecimalField( |
| max_digits=12, |
| decimal_places=2, |
| null=True, |
| blank=True, |
| verbose_name=_("价税合计"), |
| ) |
| buyer_name = models.CharField(max_length=255, blank=True, default="", verbose_name=_("购买方名称")) |
| seller_name = models.CharField(max_length=255, blank=True, default="", verbose_name=_("销售方名称")) |
| project_name = models.CharField(max_length=255, blank=True, default="", verbose_name=_("项目名称")) |
| category = models.CharField( |
| max_length=32, |
| choices=InvoiceCategory.choices, |
| default=InvoiceCategory.OTHER, |
| verbose_name=_("发票类目"), |
| ) |
| raw_text = models.TextField(blank=True, default="", verbose_name=_("OCR 原始文本")) |
| is_duplicate = models.BooleanField(default=False, verbose_name=_("是否重复")) |
| duplicate_of_id = models.IntegerField(null=True, blank=True, verbose_name=_("重复的原始记录ID")) |
| status = models.CharField( |
| max_length=32, |
| choices=InvoiceRecordStatus.choices, |
| default=InvoiceRecordStatus.PENDING, |
| verbose_name=_("识别状态"), |
| ) |
| created_at = models.DateTimeField(auto_now_add=True, verbose_name=_("创建时间")) |
|
|
| class Meta: |
| managed = False |
| db_table = "automation_invoicerecord" |
| verbose_name = _("发票记录") |
| verbose_name_plural = _("发票记录") |
|
|
| def __str__(self) -> str: |
| return f"{self.original_filename} ({self.get_status_display()})" |
|
|
