Initial commit

backend/load_trading_rules.py

@@ -0,0 +1,101 @@
+#!/usr/bin/env python3
+"""
+매매 규칙 설정을 데이터베이스에 로드하는 스크립트
+
+Usage:
+    python load_trading_rules.py
+"""
+
+import asyncio
+import json
+from pathlib import Path
+from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy import select
+
+from app.core.config import settings
+from app.models.models import TradingRule
+
+
+async def load_trading_rules():
+    """trading_rules_config.json 파일을 읽어서 데이터베이스에 저장"""
+    
+    # 설정 파일 읽기
+    config_path = Path(__file__).parent / "trading_rules_config.json"
+    
+    if not config_path.exists():
+        print(f"❌ 설정 파일을 찾을 수 없습니다: {config_path}")
+        return
+    
+    with open(config_path, "r", encoding="utf-8") as f:
+        config = json.load(f)
+    
+    # 데이터베이스 연결
+    engine = create_async_engine(settings.DATABASE_URL, echo=True)
+    async_session = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
+    
+    async with async_session() as session:
+        print("\n📋 매매 규칙 로드 중...\n")
+        
+        # 기존 규칙 삭제 (선택적)
+        result = await session.execute(select(TradingRule))
+        existing_rules = result.scalars().all()
+        
+        if existing_rules:
+            print(f"⚠️  기존 규칙 {len(existing_rules)}개 발견")
+            response = input("기존 규칙을 모두 삭제하고 새로 로드하시겠습니까? (y/N): ")
+            if response.lower() == 'y':
+                for rule in existing_rules:
+                    await session.delete(rule)
+                await session.commit()
+                print("✅ 기존 규칙 삭제 완료\n")
+        
+        # 새 규칙 추가
+        rules = config.get("rules", [])
+        
+        for idx, rule_data in enumerate(rules, 1):
+            trading_rule = TradingRule(
+                name=rule_data["name"],
+                description=rule_data.get("description", ""),
+                rule_type=rule_data["rule_type"],
+                conditions=rule_data["conditions"],
+                actions=rule_data["actions"],
+                is_active=rule_data.get("is_active", True),
+                priority=rule_data.get("priority", idx)
+            )
+            
+            session.add(trading_rule)
+            print(f"✅ [{idx}/{len(rules)}] {rule_data['name']} - {'활성화' if rule_data.get('is_active') else '비활성화'}")
+        
+        await session.commit()
+        
+        print(f"\n🎉 총 {len(rules)}개의 매매 규칙이 로드되었습니다!")
+        
+        # 포트폴리오 제약 조건 및 리스크 관리 설정 출력
+        print("\n" + "="*60)
+        print("📊 포트폴리오 제약 조건:")
+        print("="*60)
+        for key, value in config.get("portfolio_constraints", {}).items():
+            if isinstance(value, float):
+                print(f"  • {key}: {value*100:.1f}%" if value < 1 else f"  • {key}: {value:,.0f}")
+            else:
+                print(f"  • {key}: {value}")
+        
+        print("\n" + "="*60)
+        print("⚠️  리스크 관리:")
+        print("="*60)
+        for key, value in config.get("risk_management", {}).items():
+            if isinstance(value, float) and value < 1:
+                print(f"  • {key}: {value*100:.1f}%")
+            else:
+                print(f"  • {key}: {value}")
+        
+        print("\n" + "="*60)
+        print("✅ 설정 로드 완료!")
+        print("="*60)
+    
+    await engine.dispose()
+
+
+if __name__ == "__main__":
+    asyncio.run(load_trading_rules())