bolade
142 lines
2.9 KiB
Python
142 lines
2.9 KiB
Python
from datetime import datetime
|
|
from typing import List, Optional
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class AddressRequest(BaseModel):
|
|
province: str
|
|
city: str
|
|
postal_code: str
|
|
country: str = "Canada"
|
|
|
|
|
|
class ReceiptRequest(BaseModel):
|
|
id: str
|
|
file_name: str
|
|
upload_date: datetime
|
|
receipt_date: datetime
|
|
amount: float
|
|
tax: float
|
|
vendor: str
|
|
category: str
|
|
description: str
|
|
# Tax rule fields
|
|
billing_address: Optional[AddressRequest] = None
|
|
shipping_address: Optional[AddressRequest] = None
|
|
currency: str = "CAD"
|
|
is_meals_entertainment: bool = False
|
|
|
|
|
|
class TransactionRequest(BaseModel):
|
|
id: str
|
|
transaction_date: datetime
|
|
amount: float
|
|
vendor: str
|
|
notes: str
|
|
# Tax rule fields
|
|
currency: str = "CAD"
|
|
fx_rate: Optional[float] = None
|
|
|
|
|
|
class AssetRequest(BaseModel):
|
|
id: str
|
|
name: str
|
|
purchase_date: datetime
|
|
purchase_amount: float
|
|
useful_life_years: int
|
|
residual_value: float
|
|
cca_rate: float
|
|
asset_class: str
|
|
|
|
|
|
class MatchingRequest(BaseModel):
|
|
receipt_ids: List[str]
|
|
transaction_ids: List[str]
|
|
|
|
|
|
class MatchResponse(BaseModel):
|
|
receipt_id: str
|
|
transaction_id: str
|
|
confidence_score: float
|
|
match_reason: str
|
|
receipt_vendor: str
|
|
receipt_amount: float
|
|
receipt_description: str
|
|
receipt_category: str
|
|
receipt_tax_amount: float
|
|
transaction_vendor: str
|
|
transaction_amount: float
|
|
|
|
|
|
class MatchingResponse(BaseModel):
|
|
matches: List[MatchResponse]
|
|
stats: dict
|
|
|
|
|
|
class ApprovalRequest(BaseModel):
|
|
match_id: str
|
|
approved: bool
|
|
reason: Optional[str] = None
|
|
|
|
|
|
class RuleRequest(BaseModel):
|
|
name: str
|
|
condition: str
|
|
action: str
|
|
source: str = "user"
|
|
|
|
|
|
class DocumentUploadResponse(BaseModel):
|
|
file_id: str
|
|
filename: str
|
|
upload_date: datetime
|
|
status: str
|
|
|
|
|
|
class DocumentProcessResponse(BaseModel):
|
|
file_id: str
|
|
extraction_success: bool
|
|
vendor: Optional[str] = None
|
|
description: Optional[str] = None
|
|
total_amount: Optional[float] = None
|
|
tax_amount: Optional[float] = None
|
|
date: Optional[str] = None
|
|
category: Optional[str] = None
|
|
confidence: Optional[float] = None
|
|
error: Optional[str] = None
|
|
|
|
|
|
# New tax-related models
|
|
class TaxCalculationRequest(BaseModel):
|
|
receipt_id: str
|
|
transaction_id: Optional[str] = None
|
|
|
|
|
|
class TaxCalculationResponse(BaseModel):
|
|
receipt_id: str
|
|
rules_applied: List[str]
|
|
sales_tax: dict
|
|
fx_analysis: Optional[dict] = None
|
|
meals_entertainment: dict
|
|
|
|
|
|
class DepreciationRequest(BaseModel):
|
|
asset: AssetRequest
|
|
year: int
|
|
method: str # "straight_line" or "cca"
|
|
|
|
|
|
class DepreciationResponse(BaseModel):
|
|
asset_id: str
|
|
year: int
|
|
method: str
|
|
depreciation: float
|
|
book_value: float
|
|
total_depreciation: Optional[float] = None
|
|
success: bool
|
|
error: Optional[str] = None
|
|
|
|
class MatchSpecificRequest(BaseModel):
|
|
file_ids: List[str]
|
|
categorization_id: str
|
|
|