# app/extensions.py
from flask_sqlalchemy import SQLAlchemy
from flask_login import LoginManager
from flask_compress import Compress
from flask_cors import CORS
from clickhouse_driver import Client as Cclient
from sqlalchemy import create_engine

# Database instances
db = SQLAlchemy()

# Other extensions
login_manager = LoginManager()
compress = Compress()
cors = CORS()

# Database engines (如果仍然需要直接使用 engine)
engine = create_engine("mysql+pymysql://root:Zzl33818!@111.198.58.126:33060/stock", echo=False)
engine_med = create_engine("mysql+pymysql://root:Zzl33818!@111.198.58.126:33060/med", echo=False)
engine_2 = create_engine("mysql+pymysql://root:Zzl33818!@111.198.58.126:33060/valuefrontier", echo=False)

# ClickHouse client factory
def get_clickhouse_client():
    return Cclient(
        host='111.198.58.126',
        port=18778,
        user='default',
        password='Zzl5588161!',
        database='stock'
    )