158 lines
3.5 KiB
Python
158 lines
3.5 KiB
Python
# Description: Pydantic data models dictate what is passed between frontend and backend.
|
|
|
|
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
# DCA Client Models
|
|
class CreateDcaClientData(BaseModel):
|
|
user_id: str
|
|
wallet_id: str
|
|
dca_mode: str = "flow" # 'flow' or 'fixed'
|
|
fixed_mode_daily_limit: Optional[int] = None
|
|
|
|
|
|
class DcaClient(BaseModel):
|
|
id: str
|
|
user_id: str
|
|
wallet_id: str
|
|
dca_mode: str
|
|
fixed_mode_daily_limit: Optional[int]
|
|
status: str
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
|
|
class UpdateDcaClientData(BaseModel):
|
|
dca_mode: Optional[str] = None
|
|
fixed_mode_daily_limit: Optional[int] = None
|
|
status: Optional[str] = None
|
|
|
|
|
|
# Deposit Models
|
|
class CreateDepositData(BaseModel):
|
|
client_id: str
|
|
amount: int # Amount in smallest currency unit (centavos for GTQ)
|
|
currency: str = "GTQ"
|
|
notes: Optional[str] = None
|
|
|
|
|
|
class DcaDeposit(BaseModel):
|
|
id: str
|
|
client_id: str
|
|
amount: int
|
|
currency: str
|
|
status: str # 'pending' or 'confirmed'
|
|
notes: Optional[str]
|
|
created_at: datetime
|
|
confirmed_at: Optional[datetime]
|
|
|
|
|
|
class UpdateDepositStatusData(BaseModel):
|
|
status: str
|
|
notes: Optional[str] = None
|
|
|
|
|
|
# Payment Models
|
|
class CreateDcaPaymentData(BaseModel):
|
|
client_id: str
|
|
amount_sats: int
|
|
amount_fiat: int
|
|
exchange_rate: float
|
|
transaction_type: str # 'flow', 'fixed', 'manual', 'commission'
|
|
lamassu_transaction_id: Optional[str] = None
|
|
payment_hash: Optional[str] = None
|
|
|
|
|
|
class DcaPayment(BaseModel):
|
|
id: str
|
|
client_id: str
|
|
amount_sats: int
|
|
amount_fiat: int
|
|
exchange_rate: float
|
|
transaction_type: str
|
|
lamassu_transaction_id: Optional[str]
|
|
payment_hash: Optional[str]
|
|
status: str # 'pending', 'confirmed', 'failed'
|
|
created_at: datetime
|
|
|
|
|
|
# Client Balance Summary
|
|
class ClientBalanceSummary(BaseModel):
|
|
client_id: str
|
|
total_deposits: int # Total confirmed deposits
|
|
total_payments: int # Total payments made
|
|
remaining_balance: int # Available balance for DCA
|
|
currency: str
|
|
|
|
|
|
# Transaction Processing Models
|
|
class LamassuTransaction(BaseModel):
|
|
transaction_id: str
|
|
amount_fiat: int
|
|
amount_crypto: int
|
|
exchange_rate: float
|
|
transaction_type: str # 'cash_in' or 'cash_out'
|
|
status: str
|
|
timestamp: datetime
|
|
|
|
|
|
# Lamassu Configuration Models
|
|
class CreateLamassuConfigData(BaseModel):
|
|
host: str
|
|
port: int = 5432
|
|
database_name: str
|
|
username: str
|
|
password: str
|
|
|
|
|
|
class LamassuConfig(BaseModel):
|
|
id: str
|
|
host: str
|
|
port: int
|
|
database_name: str
|
|
username: str
|
|
password: str
|
|
is_active: bool
|
|
test_connection_last: Optional[datetime]
|
|
test_connection_success: Optional[bool]
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
|
|
class UpdateLamassuConfigData(BaseModel):
|
|
host: Optional[str] = None
|
|
port: Optional[int] = None
|
|
database_name: Optional[str] = None
|
|
username: Optional[str] = None
|
|
password: Optional[str] = None
|
|
is_active: Optional[bool] = None
|
|
|
|
|
|
# Legacy models (keep for backward compatibility during transition)
|
|
class CreateMyExtensionData(BaseModel):
|
|
id: Optional[str] = ""
|
|
name: str
|
|
lnurlpayamount: int
|
|
lnurlwithdrawamount: int
|
|
wallet: str
|
|
total: int = 0
|
|
|
|
|
|
class MyExtension(BaseModel):
|
|
id: str
|
|
name: str
|
|
lnurlpayamount: int
|
|
lnurlwithdrawamount: int
|
|
wallet: str
|
|
total: int
|
|
lnurlpay: Optional[str] = ""
|
|
lnurlwithdraw: Optional[str] = ""
|
|
|
|
|
|
class CreatePayment(BaseModel):
|
|
myextension_id: str
|
|
amount: int
|
|
memo: str
|