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
86
87
88
89
90
91
92
93
94
95
96
|
"""Initial schema
Revision ID: 001
Revises:
Create Date: 2026-04-01
"""
from collections.abc import Sequence
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "001"
down_revision: str | None = None
branch_labels: str | Sequence[str] | None = None
depends_on: str | Sequence[str] | None = None
def upgrade() -> None:
op.create_table(
"candles",
sa.Column("symbol", sa.Text, primary_key=True),
sa.Column("timeframe", sa.Text, primary_key=True),
sa.Column("open_time", sa.DateTime(timezone=True), primary_key=True),
sa.Column("open", sa.Numeric, nullable=False),
sa.Column("high", sa.Numeric, nullable=False),
sa.Column("low", sa.Numeric, nullable=False),
sa.Column("close", sa.Numeric, nullable=False),
sa.Column("volume", sa.Numeric, nullable=False),
)
op.create_table(
"signals",
sa.Column("id", sa.Text, primary_key=True),
sa.Column("strategy", sa.Text, nullable=False),
sa.Column("symbol", sa.Text, nullable=False),
sa.Column("side", sa.Text, nullable=False),
sa.Column("price", sa.Numeric, nullable=False),
sa.Column("quantity", sa.Numeric, nullable=False),
sa.Column("reason", sa.Text),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
)
op.create_table(
"orders",
sa.Column("id", sa.Text, primary_key=True),
sa.Column("signal_id", sa.Text, sa.ForeignKey("signals.id")),
sa.Column("symbol", sa.Text, nullable=False),
sa.Column("side", sa.Text, nullable=False),
sa.Column("type", sa.Text, nullable=False),
sa.Column("price", sa.Numeric, nullable=False),
sa.Column("quantity", sa.Numeric, nullable=False),
sa.Column("status", sa.Text, nullable=False, server_default="PENDING"),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False),
sa.Column("filled_at", sa.DateTime(timezone=True)),
)
op.create_table(
"trades",
sa.Column("id", sa.Text, primary_key=True),
sa.Column("order_id", sa.Text, sa.ForeignKey("orders.id")),
sa.Column("symbol", sa.Text, nullable=False),
sa.Column("side", sa.Text, nullable=False),
sa.Column("price", sa.Numeric, nullable=False),
sa.Column("quantity", sa.Numeric, nullable=False),
sa.Column("fee", sa.Numeric, nullable=False, server_default="0"),
sa.Column("traded_at", sa.DateTime(timezone=True), nullable=False),
)
op.create_table(
"positions",
sa.Column("symbol", sa.Text, primary_key=True),
sa.Column("quantity", sa.Numeric, nullable=False),
sa.Column("avg_entry_price", sa.Numeric, nullable=False),
sa.Column("current_price", sa.Numeric, nullable=False),
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False),
)
op.create_table(
"portfolio_snapshots",
sa.Column("id", sa.Integer, primary_key=True, autoincrement=True),
sa.Column("total_value", sa.Numeric, nullable=False),
sa.Column("realized_pnl", sa.Numeric, nullable=False),
sa.Column("unrealized_pnl", sa.Numeric, nullable=False),
sa.Column("snapshot_at", sa.DateTime(timezone=True), nullable=False),
)
def downgrade() -> None:
op.drop_table("portfolio_snapshots")
op.drop_table("positions")
op.drop_table("trades")
op.drop_table("orders")
op.drop_table("signals")
op.drop_table("candles")
|
