"""
创建股票基础数据相关表（使用 SQLAlchemy ORM，兼容 SQLite 和 PostgreSQL）
"""
from app.db.session import engine, SessionLocal
from app.db.base import Base
from app.models.stock_basic import StockBasic, IndexBasic, IndexTrade

def create_tables():
    """创建所有股票基础数据相关表"""
    print("开始创建数据库表...")
    
    Base.metadata.create_all(bind=engine)
    
    print("数据库表创建完成")
    print(f"创建的表: stock_basic, index_basic, index_trade")

if __name__ == "__main__":
    create_tables()