看股功能更新

This commit is contained in:
2026-08-16 00:05:26 +08:00
parent 9cce670b74
commit fc86fe0674
28 changed files with 3823 additions and 96 deletions

View File

@@ -0,0 +1,36 @@
"""add adj_factor table (复权因子底座)
Revision ID: 20260815_01
Revises: 208b0c5d302a
Create Date: 2026-08-15
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
revision: str = "20260815_01"
down_revision: Union[str, Sequence[str], None] = "208b0c5d302a"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.create_table(
"adj_factor",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("trade_date", sa.DateTime(), nullable=False),
sa.Column("ts_code", sa.String(length=12), nullable=False),
sa.Column("adj_factor", sa.Float(), nullable=False),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("ts_code", "trade_date", name="uq_adj_code_date"),
)
op.create_index("ix_adj_factor_trade_date", "adj_factor", ["trade_date"])
op.create_index("ix_adj_factor_ts_code", "adj_factor", ["ts_code"])
def downgrade() -> None:
op.drop_index("ix_adj_factor_ts_code", table_name="adj_factor")
op.drop_index("ix_adj_factor_trade_date", table_name="adj_factor")
op.drop_table("adj_factor")

View File

@@ -0,0 +1,69 @@
"""user data tables: preferences / watchlist / screener queries
Revision ID: 20260815_02
Revises: 20260815_01
Create Date: 2026-08-15
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
revision: str = "20260815_02"
down_revision: Union[str, Sequence[str], None] = "20260815_01"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.create_table(
"user_preferences",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("user_id", sa.BigInteger(), nullable=False),
sa.Column("key", sa.String(length=64), nullable=False),
sa.Column("value_json", sa.Text(), nullable=False, server_default="null"),
sa.Column("updated_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("user_id", "key", name="uq_user_pref_key"),
)
op.create_index("ix_user_preferences_user_id", "user_preferences", ["user_id"])
op.create_table(
"watchlist_items",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("user_id", sa.BigInteger(), nullable=False),
sa.Column("ts_code", sa.String(length=12), nullable=False),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint("user_id", "ts_code", name="uq_watch_user_code"),
)
op.create_index("ix_watchlist_items_user_id", "watchlist_items", ["user_id"])
op.create_index("ix_watchlist_items_ts_code", "watchlist_items", ["ts_code"])
op.create_table(
"screener_queries",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("user_id", sa.BigInteger(), nullable=False),
sa.Column("text", sa.String(length=500), nullable=False),
sa.Column("conditions_json", sa.Text(), nullable=True),
sa.Column("hit_count", sa.Integer(), nullable=True),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
sa.PrimaryKeyConstraint("id"),
)
op.create_index("ix_screener_queries_user_id", "screener_queries", ["user_id"])
op.create_index("ix_screener_queries_created_at", "screener_queries", ["created_at"])
def downgrade() -> None:
op.drop_index("ix_screener_queries_created_at", table_name="screener_queries")
op.drop_index("ix_screener_queries_user_id", table_name="screener_queries")
op.drop_table("screener_queries")
op.drop_index("ix_watchlist_items_ts_code", table_name="watchlist_items")
op.drop_index("ix_watchlist_items_user_id", table_name="watchlist_items")
op.drop_table("watchlist_items")
op.drop_index("ix_user_preferences_user_id", table_name="user_preferences")
op.drop_table("user_preferences")

View File

@@ -0,0 +1,30 @@
"""candles add amount/turnover columns
Revision ID: 20260815_03
Revises: 20260815_02
Create Date: 2026-08-15
- amount 成交额TDX .day 原生 float32/ Tushare daily amount 千元×1000
- turnover 换手率(%Tushare daily_basic.turnover_rate2000 年起)
均为可空列——历史回补前为 NULL前端 tooltip 显示 ""
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
revision: str = "20260815_03"
down_revision: Union[str, Sequence[str], None] = "20260815_02"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.add_column("candles", sa.Column("amount", sa.Float(), nullable=True))
op.add_column("candles", sa.Column("turnover", sa.Float(), nullable=True))
def downgrade() -> None:
op.drop_column("candles", "turnover")
op.drop_column("candles", "amount")

View File

@@ -0,0 +1,51 @@
"""user_trades: 交割单导入的实盘成交流水K线买卖点数据源
Revision ID: 20260815_04
Revises: 20260815_03
Create Date: 2026-08-15
"""
from typing import Sequence, Union
from alembic import op
import sqlalchemy as sa
revision: str = "20260815_04"
down_revision: Union[str, Sequence[str], None] = "20260815_03"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.create_table(
"user_trades",
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
sa.Column("user_id", sa.BigInteger(), nullable=False),
sa.Column("ts_code", sa.String(length=12), nullable=False),
sa.Column("code", sa.String(length=10), nullable=False),
sa.Column("name", sa.String(length=32), nullable=True),
sa.Column("trade_date", sa.Date(), nullable=False),
sa.Column("direction", sa.String(length=4), nullable=False),
sa.Column("price", sa.Float(), nullable=True),
sa.Column("qty", sa.Float(), nullable=False),
sa.Column("amount", sa.Float(), nullable=True),
sa.Column("fee", sa.Float(), nullable=False, server_default="0"),
sa.Column("raw_json", sa.Text(), nullable=True),
sa.Column("created_at", sa.DateTime(), nullable=False),
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="CASCADE"),
sa.PrimaryKeyConstraint("id"),
sa.UniqueConstraint(
"user_id", "trade_date", "ts_code", "direction", "price", "qty",
name="uq_user_trade_dedup",
),
)
op.create_index("ix_user_trades_user_id", "user_trades", ["user_id"])
op.create_index("ix_user_trades_ts_code", "user_trades", ["ts_code"])
op.create_index("ix_user_trades_trade_date", "user_trades", ["trade_date"])
def downgrade() -> None:
op.drop_index("ix_user_trades_trade_date", table_name="user_trades")
op.drop_index("ix_user_trades_ts_code", table_name="user_trades")
op.drop_index("ix_user_trades_user_id", table_name="user_trades")
op.drop_table("user_trades")