File size: 1,652 Bytes
d5684b3
 
4761dd9
 
d5684b3
81ec8d6
d5684b3
 
 
 
 
 
 
81ec8d6
d5684b3
 
 
 
 
81ec8d6
 
 
 
 
 
 
 
 
 
 
 
 
d5684b3
 
 
 
 
f92858b
d5684b3
 
81ec8d6
d5684b3
 
 
 
 
81ec8d6
 
0dbb03c
 
 
 
81ec8d6
d5684b3
a4ef0c7
 
00277e2
 
 
 
 
 
 
2508dde
 
4761dd9
a4ef0c7
 
 
 
 
4761dd9
 
 
b905bd6
 
4761dd9
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
from enum import Enum
from datetime import datetime
from typing import Dict, List
from typing_extensions import TypedDict

from app.schema.base import BaseModel, PydanticBaseModel


class TransactionType(str, Enum):
    INCOME = "income"
    EXPENSE = "expense"


class UserCreate(PydanticBaseModel):
    name: str
    email: str
    hashed_password: str


class UserUpdate(PydanticBaseModel):
    name: str
    email: str
    hashed_password: str


class UserResponse(PydanticBaseModel):
    id: int
    name: str
    email: str
    is_deleted: bool


class User(BaseModel):
    name: str
    email: str
    hashed_password: str
    is_deleted: bool = False
    transactions: "List[Transaction]" = []


class TransactionResponse(PydanticBaseModel):
    transaction_date: datetime
    category: str
    name_description: str
    amount: float
    type: TransactionType


class TransactionCreate(TransactionResponse):
    user_id: int


class Transaction(TransactionResponse):
    user: User


class FileUploadCreate(PydanticBaseModel):
    source: str
    date: datetime
    category: str
    name_description: str
    amount: float
    type: str


class IncomeStatementCreateRequest(PydanticBaseModel):
    user_id: int
    date_from: datetime
    date_to: datetime


class IncomeStatementDetail(TypedDict):
    total: float
    category_totals: List[Dict[str, str | float]]

class IncomeStatementLLMResponse(PydanticBaseModel):
    income: IncomeStatementDetail
    expenses: IncomeStatementDetail


class IncomeStatement(IncomeStatementCreateRequest, IncomeStatementLLMResponse):
    pass

class IncomeStatementResponse(IncomeStatement):
    id: int