129 lines
4.9 KiB
Python
Executable File
129 lines
4.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
데이터베이스 초기화 스크립트
|
|
PostgreSQL 데이터베이스에 테이블을 생성하고 기본 데이터를 삽입합니다.
|
|
"""
|
|
import sys
|
|
import os
|
|
from dotenv import load_dotenv
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
# 환경 변수 로드
|
|
load_dotenv()
|
|
|
|
# app 모듈 import를 위한 경로 추가
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
from app.models import Base, Asset, UserAsset, AlertSetting
|
|
from app.database import DATABASE_URL
|
|
import json
|
|
|
|
def init_database():
|
|
"""데이터베이스 초기화"""
|
|
print("🔧 데이터베이스 초기화 시작...")
|
|
|
|
try:
|
|
# 엔진 생성
|
|
engine = create_engine(DATABASE_URL, echo=True)
|
|
|
|
# 테이블 생성
|
|
print("\n📋 테이블 생성 중...")
|
|
Base.metadata.create_all(bind=engine)
|
|
print("✅ 테이블 생성 완료")
|
|
|
|
# 세션 생성
|
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
db = SessionLocal()
|
|
|
|
try:
|
|
# 자산 마스터 데이터 초기화
|
|
print("\n📊 자산 마스터 데이터 초기화 중...")
|
|
assets_data = [
|
|
("XAU/USD", "금/달러", "귀금속"),
|
|
("XAU/CNY", "금/위안", "귀금속"),
|
|
("XAU/GBP", "금/파운드", "귀금속"),
|
|
("USD/DXY", "달러인덱스", "환율"),
|
|
("USD/KRW", "달러/원", "환율"),
|
|
("BTC/USD", "비트코인/달러", "암호화폐"),
|
|
("BTC/KRW", "비트코인/원", "암호화폐"),
|
|
("KRX/GLD", "금 현물", "귀금속"),
|
|
("XAU/KRW", "금/원", "귀금속"),
|
|
]
|
|
|
|
for symbol, name, category in assets_data:
|
|
existing = db.query(Asset).filter(Asset.symbol == symbol).first()
|
|
if not existing:
|
|
asset = Asset(symbol=symbol, name=name, category=category)
|
|
db.add(asset)
|
|
print(f" ✓ {symbol} ({name}) 추가")
|
|
else:
|
|
print(f" - {symbol} 이미 존재")
|
|
|
|
db.commit()
|
|
print("✅ 자산 마스터 데이터 초기화 완료")
|
|
|
|
# 사용자 자산 초기화
|
|
print("\n👤 사용자 자산 데이터 초기화 중...")
|
|
assets = db.query(Asset).all()
|
|
for asset in assets:
|
|
existing = db.query(UserAsset).filter(UserAsset.asset_id == asset.id).first()
|
|
if not existing:
|
|
user_asset = UserAsset(
|
|
asset_id=asset.id,
|
|
previous_close=0,
|
|
average_price=0,
|
|
quantity=0
|
|
)
|
|
db.add(user_asset)
|
|
print(f" ✓ {asset.symbol} 사용자 데이터 추가")
|
|
else:
|
|
print(f" - {asset.symbol} 사용자 데이터 이미 존재")
|
|
|
|
db.commit()
|
|
print("✅ 사용자 자산 데이터 초기화 완료")
|
|
|
|
# 알림 설정 초기화
|
|
print("\n🔔 알림 설정 초기화 중...")
|
|
default_settings = {
|
|
"급등락_감지": False,
|
|
"급등락_임계값": 3.0,
|
|
"목표수익률_감지": False,
|
|
"목표수익률": 10.0,
|
|
"특정가격_감지": False,
|
|
"금_목표가격": 100000,
|
|
"BTC_목표가격": 100000000,
|
|
}
|
|
|
|
for key, value in default_settings.items():
|
|
existing = db.query(AlertSetting).filter(AlertSetting.setting_key == key).first()
|
|
if not existing:
|
|
setting = AlertSetting(setting_key=key, setting_value=json.dumps(value))
|
|
db.add(setting)
|
|
print(f" ✓ {key}: {value}")
|
|
else:
|
|
print(f" - {key} 이미 존재")
|
|
|
|
db.commit()
|
|
print("✅ 알림 설정 초기화 완료")
|
|
|
|
print("\n🎉 데이터베이스 초기화 완료!")
|
|
print("\n다음 명령으로 서버를 시작하세요:")
|
|
print(" python main.py")
|
|
|
|
finally:
|
|
db.close()
|
|
|
|
except Exception as e:
|
|
print(f"\n❌ 오류 발생: {e}")
|
|
print("\n문제 해결 방법:")
|
|
print("1. PostgreSQL이 실행 중인지 확인하세요:")
|
|
print(" sudo systemctl status postgresql")
|
|
print("\n2. 데이터베이스가 생성되었는지 확인하세요:")
|
|
print(" sudo -u postgres psql -c '\\l'")
|
|
print("\n3. .env 파일의 DATABASE_URL이 올바른지 확인하세요")
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
init_database()
|