"""MySQL 接続ヘルパ（collectors / ai 共通）"""
from __future__ import annotations

import os
import sys

sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

try:
    import pymysql
except ImportError as e:
    raise ImportError("pip install pymysql") from e

from config import DB_HOST, DB_NAME, DB_PASSWORD, DB_PORT, DB_USER, DB_CHARSET


def connect():
    return pymysql.connect(
        host=DB_HOST,
        port=DB_PORT,
        user=DB_USER,
        password=DB_PASSWORD,
        database=DB_NAME,
        charset=DB_CHARSET,
        cursorclass=pymysql.cursors.DictCursor,
    )