import pymysql
from dotenv import load_dotenv
import os

load_dotenv(dotenv_path="keys.env")

db_name = os.getenv("DATABASE_NAME")
db_user = os.getenv("MYSQL_USER")
mysql_host = os.getenv("MYSQL_HOST")
mysql_password = os.getenv("MYSQL_PASSWORD")
db_port = int(os.getenv("MYSQL_PORT"))


timeout = 20

def get_db():
    connection = pymysql.connect(
        charset="utf8mb4",
        connect_timeout=timeout,
        cursorclass=pymysql.cursors.DictCursor,
        db=db_name,
        host=mysql_host,
        password=mysql_password,
        read_timeout=timeout,
        port=db_port,
        user=db_user,
        write_timeout=timeout,
        ssl={"ssl": True},
    )
    return connection