chore: initial project scaffold and gitignore

This commit is contained in:
2026-04-03 07:33:32 +00:00
commit 3e3acbd13d
41 changed files with 3658 additions and 0 deletions
+46
View File
@@ -0,0 +1,46 @@
from __future__ import annotations
from logging.config import fileConfig
from alembic import context
from sqlalchemy import engine_from_config, pool
config = context.config
if config.config_file_name is not None:
fileConfig(config.config_file_name)
target_metadata = None
def run_migrations_offline() -> None:
url = config.get_main_option("sqlalchemy.url")
context.configure(
url=url,
target_metadata=target_metadata,
literal_binds=True,
dialect_opts={"paramstyle": "named"},
)
with context.begin_transaction():
context.run_migrations()
def run_migrations_online() -> None:
connectable = engine_from_config(
config.get_section(config.config_ini_section, {}),
prefix="sqlalchemy.",
poolclass=pool.NullPool,
)
with connectable.connect() as connection:
context.configure(connection=connection, target_metadata=target_metadata)
with context.begin_transaction():
context.run_migrations()
if context.is_offline_mode():
run_migrations_offline()
else:
run_migrations_online()
+20
View File
@@ -0,0 +1,20 @@
"""${message}
Revision ID: ${up_revision}
Revises: ${down_revision | comma,n}
Create Date: ${create_date}
"""
from __future__ import annotations
from alembic import op
import sqlalchemy as sa
${imports if imports else ""}
def upgrade() -> None:
${upgrades if upgrades else "pass"}
def downgrade() -> None:
${downgrades if downgrades else "pass"}
@@ -0,0 +1,90 @@
"""initial portfolio schema
Revision ID: 20260403_0001
Revises:
Create Date: 2026-04-03 00:00:00
"""
from __future__ import annotations
from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects import postgresql
revision = "20260403_0001"
down_revision = None
branch_labels = None
depends_on = None
def upgrade() -> None:
op.create_table(
"strategy_sets",
sa.Column("id", sa.Text(), primary_key=True),
sa.Column("strategy_name", sa.Text(), nullable=False),
sa.Column("strategy_config", sa.Text(), nullable=False, server_default=""),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("NOW()")),
sa.UniqueConstraint("strategy_name", "strategy_config", name="uq_strategy_sets_name_config"),
)
op.create_table(
"portfolios",
sa.Column("portfolio_id", sa.Text(), primary_key=True),
sa.Column("broker", sa.Text(), nullable=False, server_default="kis"),
sa.Column("status", sa.Text(), nullable=False, server_default="enabled"),
sa.Column("symbol", sa.Text(), nullable=False),
sa.Column("strategy_name", sa.Text(), nullable=False, server_default=""),
sa.Column("strategy_params", postgresql.JSONB(astext_type=sa.Text()), nullable=False, server_default=sa.text("'{}'::jsonb")),
sa.Column("cash_balance", sa.Float(), nullable=False),
sa.Column("asset_quantity", sa.Float(), nullable=False),
sa.Column("asset_price", sa.Float(), nullable=False, server_default="0"),
sa.Column("profit_buffer", sa.Float(), nullable=False, server_default="0"),
sa.Column("current_strategy_set_id", sa.Text(), nullable=True),
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("NOW()")),
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("NOW()")),
sa.Column("last_trade_at", sa.DateTime(timezone=True), nullable=True),
)
op.create_table(
"portfolio_ledger",
sa.Column("portfolio_id", sa.Text(), nullable=False),
sa.Column("recorded_at", sa.DateTime(timezone=True), nullable=False),
sa.Column("action", sa.Text(), nullable=False),
sa.Column("price", sa.Float(), nullable=False),
sa.Column("quantity", sa.Float(), nullable=False),
sa.Column("fee_amount", sa.Float(), nullable=False, server_default="0"),
sa.Column("tax_amount", sa.Float(), nullable=False, server_default="0"),
sa.Column("cash_balance_after", sa.Float(), nullable=False),
sa.Column("asset_quantity_after", sa.Float(), nullable=False),
sa.Column("total_value_after", sa.Float(), nullable=False),
sa.Column("profit_buffer_after", sa.Float(), nullable=False, server_default="0"),
sa.Column("strategy_set_id", sa.Text(), nullable=False, server_default="legacy"),
sa.Column("note", sa.Text(), nullable=False, server_default=""),
)
op.create_index(
"idx_portfolio_ledger_portfolio_time",
"portfolio_ledger",
["portfolio_id", sa.text("recorded_at DESC")],
unique=False,
)
op.create_index(
"idx_portfolios_broker_status_updated",
"portfolios",
["broker", "status", sa.text("updated_at DESC")],
unique=False,
)
bind = op.get_bind()
has_timescaledb = bind.execute(sa.text("SELECT EXISTS (SELECT 1 FROM pg_extension WHERE extname = 'timescaledb')")).scalar()
if has_timescaledb:
bind.execute(sa.text("SELECT create_hypertable('portfolio_ledger', 'recorded_at', if_not_exists => TRUE)"))
def downgrade() -> None:
op.drop_index("idx_portfolios_broker_status_updated", table_name="portfolios")
op.drop_index("idx_portfolio_ledger_portfolio_time", table_name="portfolio_ledger")
op.drop_table("portfolio_ledger")
op.drop_table("portfolios")
op.drop_table("strategy_sets")