Marius Mutu
20448f7aa0
- Add TvaEntry schema supporting multiple TVA rates (A, B, C, D codes) - Update OCR extractor to extract multiple TVA entries from receipts - Support both old (19%, 9%, 5%) and new Romanian rates (21%, 11% from Aug 2025) - Add tva_breakdown, tva_total, items_count, vendor_address to Receipt model - Update OCRPreview.vue to display TVA entries with rate badges - Add "Detalii Suplimentare" section in ReceiptCreateView with editable TVA table - Add TVA breakdown display in ReceiptDetailView - Create database migration for new TVA columns 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
221 lines
7.2 KiB
Python
221 lines
7.2 KiB
Python
"""Pydantic schemas for receipts API."""
|
|
|
|
from datetime import datetime, date
|
|
from decimal import Decimal
|
|
from typing import Optional, List
|
|
from pydantic import BaseModel, Field, ConfigDict
|
|
|
|
from app.db.models.receipt import ReceiptType, ReceiptDirection, ReceiptStatus
|
|
from app.db.models.accounting_entry import EntryType
|
|
|
|
|
|
# ============ Accounting Entry Schemas ============
|
|
|
|
class AccountingEntryBase(BaseModel):
|
|
"""Base schema for accounting entry."""
|
|
entry_type: EntryType
|
|
account_code: str = Field(max_length=20)
|
|
account_name: Optional[str] = Field(default=None, max_length=200)
|
|
amount: Decimal
|
|
partner_id: Optional[int] = None
|
|
cost_center_id: Optional[int] = None
|
|
|
|
|
|
class AccountingEntryCreate(AccountingEntryBase):
|
|
"""Schema for creating an accounting entry."""
|
|
pass
|
|
|
|
|
|
class AccountingEntryUpdate(BaseModel):
|
|
"""Schema for updating an accounting entry."""
|
|
entry_type: Optional[EntryType] = None
|
|
account_code: Optional[str] = Field(default=None, max_length=20)
|
|
account_name: Optional[str] = Field(default=None, max_length=200)
|
|
amount: Optional[Decimal] = None
|
|
partner_id: Optional[int] = None
|
|
cost_center_id: Optional[int] = None
|
|
|
|
|
|
class AccountingEntryResponse(AccountingEntryBase):
|
|
"""Schema for accounting entry response."""
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
receipt_id: int
|
|
is_auto_generated: bool
|
|
modified_by: Optional[str] = None
|
|
modified_at: Optional[datetime] = None
|
|
sort_order: int
|
|
|
|
|
|
# ============ Attachment Schemas ============
|
|
|
|
class AttachmentResponse(BaseModel):
|
|
"""Schema for attachment response."""
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
receipt_id: int
|
|
filename: str
|
|
stored_filename: str
|
|
file_path: str
|
|
file_size: int
|
|
mime_type: str
|
|
uploaded_at: datetime
|
|
|
|
|
|
# ============ TVA Schema ============
|
|
|
|
class TvaEntrySchema(BaseModel):
|
|
"""Single TVA entry with code, percentage and amount."""
|
|
code: Optional[str] = Field(default=None, description="TVA code: A, B, C, D")
|
|
percent: int = Field(description="TVA percentage: 0, 5, 9, 19, 21")
|
|
amount: Decimal = Field(description="TVA amount for this rate")
|
|
|
|
|
|
# ============ Receipt Schemas ============
|
|
|
|
class ReceiptBase(BaseModel):
|
|
"""Base schema for receipt."""
|
|
receipt_type: ReceiptType = ReceiptType.BON_FISCAL
|
|
direction: ReceiptDirection = ReceiptDirection.CHELTUIALA
|
|
receipt_number: Optional[str] = Field(default=None, max_length=50)
|
|
receipt_series: Optional[str] = Field(default=None, max_length=20)
|
|
receipt_date: date
|
|
amount: Decimal = Field(gt=0)
|
|
description: Optional[str] = Field(default=None, max_length=500)
|
|
# TVA info (multiple entries support)
|
|
tva_breakdown: Optional[List[TvaEntrySchema]] = Field(default=None, description="List of TVA entries")
|
|
tva_total: Optional[Decimal] = Field(default=None, description="Total TVA amount")
|
|
items_count: Optional[int] = Field(default=None, description="Number of items")
|
|
vendor_address: Optional[str] = Field(default=None, max_length=500, description="Vendor address")
|
|
# Other fields
|
|
expense_type_code: Optional[str] = Field(default=None, max_length=20)
|
|
company_id: int
|
|
partner_id: Optional[int] = None
|
|
partner_name: Optional[str] = Field(default=None, max_length=200)
|
|
cash_register_id: Optional[int] = None
|
|
cash_register_name: Optional[str] = Field(default=None, max_length=100)
|
|
cash_register_account: Optional[str] = Field(default=None, max_length=20)
|
|
|
|
|
|
class ReceiptCreate(ReceiptBase):
|
|
"""Schema for creating a receipt."""
|
|
pass
|
|
|
|
|
|
class ReceiptUpdate(BaseModel):
|
|
"""Schema for updating a receipt (DRAFT only)."""
|
|
receipt_type: Optional[ReceiptType] = None
|
|
direction: Optional[ReceiptDirection] = None
|
|
receipt_number: Optional[str] = Field(default=None, max_length=50)
|
|
receipt_series: Optional[str] = Field(default=None, max_length=20)
|
|
receipt_date: Optional[date] = None
|
|
amount: Optional[Decimal] = Field(default=None, gt=0)
|
|
description: Optional[str] = Field(default=None, max_length=500)
|
|
# TVA info (multiple entries support)
|
|
tva_breakdown: Optional[List[TvaEntrySchema]] = Field(default=None, description="List of TVA entries")
|
|
tva_total: Optional[Decimal] = Field(default=None, description="Total TVA amount")
|
|
items_count: Optional[int] = Field(default=None, description="Number of items")
|
|
vendor_address: Optional[str] = Field(default=None, max_length=500, description="Vendor address")
|
|
# Other fields
|
|
expense_type_code: Optional[str] = Field(default=None, max_length=20)
|
|
partner_id: Optional[int] = None
|
|
partner_name: Optional[str] = Field(default=None, max_length=200)
|
|
cash_register_id: Optional[int] = None
|
|
cash_register_name: Optional[str] = Field(default=None, max_length=100)
|
|
cash_register_account: Optional[str] = Field(default=None, max_length=20)
|
|
|
|
|
|
class ReceiptResponse(ReceiptBase):
|
|
"""Schema for receipt response with all fields."""
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
status: ReceiptStatus
|
|
created_by: str
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
submitted_at: Optional[datetime] = None
|
|
reviewed_by: Optional[str] = None
|
|
reviewed_at: Optional[datetime] = None
|
|
rejection_reason: Optional[str] = None
|
|
oracle_synced_at: Optional[datetime] = None
|
|
oracle_act_id: Optional[int] = None
|
|
oracle_error: Optional[str] = None
|
|
|
|
# Relationships (optional, loaded when needed)
|
|
attachments: List[AttachmentResponse] = []
|
|
entries: List[AccountingEntryResponse] = []
|
|
|
|
|
|
class ReceiptListResponse(BaseModel):
|
|
"""Schema for paginated receipt list response."""
|
|
items: List[ReceiptResponse]
|
|
total: int
|
|
page: int
|
|
page_size: int
|
|
pages: int
|
|
|
|
|
|
class ReceiptFilter(BaseModel):
|
|
"""Schema for filtering receipts."""
|
|
status: Optional[ReceiptStatus] = None
|
|
company_id: Optional[int] = None
|
|
created_by: Optional[str] = None
|
|
date_from: Optional[date] = None
|
|
date_to: Optional[date] = None
|
|
search: Optional[str] = None # Search in description, partner_name
|
|
page: int = Field(default=1, ge=1)
|
|
page_size: int = Field(default=20, ge=1, le=100)
|
|
|
|
|
|
# ============ Workflow Schemas ============
|
|
|
|
class WorkflowAction(BaseModel):
|
|
"""Schema for workflow action response."""
|
|
success: bool
|
|
message: str
|
|
receipt: Optional[ReceiptResponse] = None
|
|
|
|
|
|
class RejectRequest(BaseModel):
|
|
"""Schema for rejection request."""
|
|
reason: str = Field(min_length=5, max_length=500)
|
|
|
|
|
|
class EntriesUpdateRequest(BaseModel):
|
|
"""Schema for bulk updating accounting entries."""
|
|
entries: List[AccountingEntryCreate]
|
|
|
|
|
|
# ============ Nomenclature Schemas ============
|
|
|
|
class PartnerOption(BaseModel):
|
|
"""Schema for partner dropdown option."""
|
|
id: int
|
|
name: str
|
|
code: Optional[str] = None
|
|
|
|
|
|
class AccountOption(BaseModel):
|
|
"""Schema for account dropdown option."""
|
|
code: str
|
|
name: str
|
|
|
|
|
|
class CashRegisterOption(BaseModel):
|
|
"""Schema for cash register dropdown option."""
|
|
id: int
|
|
name: str
|
|
account_code: str # 5311, 5121, etc.
|
|
|
|
|
|
class ExpenseTypeOption(BaseModel):
|
|
"""Schema for expense type dropdown option."""
|
|
code: str
|
|
name: str
|
|
account_code: str
|
|
has_vat: bool
|
|
vat_percent: Decimal = Decimal("19")
|