shuim
b4a1fc79ab
설계: docs/multi_tenant_design.md / 검수: docs/multi_tenant_phase1_review.md (High 0, MT=False 배포 가능)
데이터·인프라
- user_kis_credentials(Fernet 암호화, PBKDF2 키 파생, key_version), stock_bans_v2 신설,
portfolio/trade_history/active_trades/trade_journals/trading_sessions 등 user_id nullable 추가
- alembic 0009~0011(idempotent, legacy owner 백필). main.py 기동 시 portfolio.user_id 스키마 검사(H1)
- Jenkinsfile: DB Migration 을 Deploy 앞으로, alembic 이력 없으면 stamp 0008 후 upgrade(N1)
- bootstrap: users 비어 있으면 BOOTSTRAP_ADMIN_* 로 관리자 시드 + .env KIS 자격증명 이관 + 백필(SAVEPOINT)
백엔드
- core/crypto.py, core/tenant_auth.py(get_current_user — 비활성 사용자 401, API 키→legacy owner,
deny_non_owner_session_when_mt), core/dt_utils.py
- services/tenant: context, kis_pool, registry(UserContextRegistry, 락 최소화·복호화 to_thread), risk_factory
- KoreaInvestmentAPI.for_user(사용자별 인스턴스, 토큰캐시 users/{id}/, 시세는 공용 위임)
- trade_executor.safe_place_order_v2(사용자별 락/세마포어, legacy ctx 는 기존 키)
- mdd_manager/loss_tracker _v2 + ban_until tz-aware/naive 경계 정규화(레거시 경로 버그 포함 수정)
- db_worker: payload user_id 옵션 필터, upsert_stock_ban_v2, legacy ActiveTrade 조회 LIMIT 1
- API: /account/credentials(등록·마스킹조회·삭제·연결테스트·자동매매), /auth/admin/tenant/*,
/portfolio/v2(조회·요약·sync·buy·sell), /trades/v2(+pending). PortfolioResponse Optional 수정
프론트
- /account 내 계좌 설정, useTenantMode(세션=v2 / API키=legacy, 미확정 시 쿼리·버튼 비활성),
계좌 미등록 인라인 안내, v2 매수/매도/거래이력 분기
테스트: tests/integration(격리 Postgres, KIS 전면 mock) 20 passed / 1 xfailed(0012 이전 제약)
2단계(스케줄러 사용자화·legacy 라우터 격리) 전까지 MULTI_TENANT_ENABLED=True 전환 금지
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
66 lines
2.7 KiB
Python
66 lines
2.7 KiB
Python
"""add user_kis_credentials table (multi-tenant phase 1)
|
|
|
|
Revision ID: 0009
|
|
Revises: 0008
|
|
Create Date: 2026-09-21
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "0009"
|
|
down_revision = "0008"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
conn = op.get_bind()
|
|
exists = conn.execute(sa.text(
|
|
"SELECT 1 FROM information_schema.tables WHERE table_name='user_kis_credentials'"
|
|
)).fetchone()
|
|
if exists:
|
|
return # create_all 로 이미 생성됨
|
|
op.create_table(
|
|
"user_kis_credentials",
|
|
sa.Column("id", sa.Integer(), primary_key=True),
|
|
sa.Column("user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("mode", sa.String(10), nullable=False),
|
|
sa.Column("app_key_enc", sa.Text(), nullable=False),
|
|
sa.Column("app_secret_enc", sa.Text(), nullable=False),
|
|
sa.Column("account_no_enc", sa.Text(), nullable=False),
|
|
sa.Column("account_product_code", sa.String(2), nullable=False, server_default="01"),
|
|
sa.Column("app_key_hash", sa.String(64), nullable=False),
|
|
sa.Column("app_key_hint", sa.String(8), nullable=True),
|
|
sa.Column("is_enabled", sa.Boolean(), nullable=False, server_default=sa.false()),
|
|
sa.Column("auto_trade_enabled", sa.Boolean(), nullable=False, server_default=sa.false()),
|
|
sa.Column("last_verified_at", sa.DateTime(), nullable=True),
|
|
sa.Column("key_version", sa.Integer(), nullable=False, server_default="1"),
|
|
sa.Column("created_at", sa.DateTime(), nullable=False, server_default=sa.func.now()),
|
|
sa.Column("updated_at", sa.DateTime(), nullable=True),
|
|
)
|
|
op.create_index(
|
|
"ix_user_kis_credentials_user_id", "user_kis_credentials", ["user_id"], unique=True
|
|
)
|
|
op.create_index(
|
|
"ix_user_kis_credentials_app_key_hash", "user_kis_credentials", ["app_key_hash"], unique=True
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
# L3: 테이블이 `create_all`(SQLAlchemy 모델의 `unique=True` 컬럼)로 먼저 생성된
|
|
# 경우 실제로는 named index가 아니라 unique constraint(`..._key`)로 존재해
|
|
# `drop_index`가 실패한다. 존재할 때만 드롭한다.
|
|
conn = op.get_bind()
|
|
for index_name in (
|
|
"ix_user_kis_credentials_app_key_hash",
|
|
"ix_user_kis_credentials_user_id",
|
|
):
|
|
exists = conn.execute(sa.text(
|
|
"SELECT 1 FROM pg_indexes WHERE tablename='user_kis_credentials' AND indexname=:name"
|
|
), {"name": index_name}).fetchone()
|
|
if exists:
|
|
op.drop_index(index_name, table_name="user_kis_credentials")
|
|
op.drop_table("user_kis_credentials")
|