source push

modify_db_schema.py

@@ -0,0 +1,37 @@
+import sqlite3
+import os
+
+# 데이터베이스 파일 경로
+db_path = os.path.join(os.path.dirname(__file__), 'quant_bot.db')
+
+# 연결 생성
+conn = sqlite3.connect(db_path)
+cursor = conn.cursor()
+
+# 기존 필드 이름 변경
+print("기존 테이블 필드 변경 중...")
+cursor.execute("ALTER TABLE env_config RENAME COLUMN KIS_APP_KEY TO KIS_APP_KEY_REAL")
+cursor.execute("ALTER TABLE env_config RENAME COLUMN KIS_APP_SECRET TO KIS_APP_SECRET_REAL")
+cursor.execute("ALTER TABLE env_config RENAME COLUMN KIS_ACCOUNT_NO TO KIS_ACCOUNT_NO_REAL")
+cursor.execute("ALTER TABLE env_config RENAME COLUMN KIS_ACCOUNT_CODE TO KIS_ACCOUNT_CODE_REAL")
+
+# 모의계좌용 필드 추가
+print("모의계좌용 필드 추가 중...")
+cursor.execute("ALTER TABLE env_config ADD COLUMN KIS_APP_KEY_MOCK TEXT")
+cursor.execute("ALTER TABLE env_config ADD COLUMN KIS_APP_SECRET_MOCK TEXT")
+cursor.execute("ALTER TABLE env_config ADD COLUMN KIS_ACCOUNT_NO_MOCK TEXT")
+cursor.execute("ALTER TABLE env_config ADD COLUMN KIS_ACCOUNT_CODE_MOCK TEXT")
+
+# 변경된 필드명으로 데이터 이동 (기존 데이터 복사)
+print("기존 데이터 이동 중...")
+# 실제 데이터베이스에서 KIS_APP_KEY_REAL을 KIS_APP_KEY_MOCK로 복사 (모의계좌용으로 사용)
+cursor.execute("UPDATE env_config SET KIS_APP_KEY_MOCK = KIS_APP_KEY_REAL")
+cursor.execute("UPDATE env_config SET KIS_APP_SECRET_MOCK = KIS_APP_SECRET_REAL")
+cursor.execute("UPDATE env_config SET KIS_ACCOUNT_NO_MOCK = KIS_ACCOUNT_NO_REAL")
+cursor.execute("UPDATE env_config SET KIS_ACCOUNT_CODE_MOCK = KIS_ACCOUNT_CODE_REAL")
+
+# 변경 내용 저장
+conn.commit()
+
+print("데이터베이스 스키마 변경 완료")
+conn.close()