import os import pymysql def _params() -> dict: return { "host": os.environ["MYSQL_HOST"], "port": int(os.environ.get("MYSQL_PORT", 3306)), "user": os.environ["MYSQL_USER"], "password": os.environ["MYSQL_PASSWORD"], "database": os.environ["MYSQL_DATABASE"], "connect_timeout": 3, } def check_db() -> bool: try: conn = pymysql.connect(**_params()) conn.close() return True except Exception: return False