اتصال به دیتابیس MySQL در برنامه‌های Python


برای اتصال به دیتابیس MySQL در برنامه‌های Python، در ابتدا باید ماژول مربوط به آن‌را با اجرای دستور زیر، نصب کنید:

کپی
pip install mysql-connector-python

در ادامه، بایستی متغیر محیطی مربوط به دیتابیس را، به برنامه خود اضافه کنید؛ به عنوان مثال:

کپی
DB_USER=root
DB_PASSWORD=Wc9yvejxaWm6RrysATmcUeew
DB_HOST=monte-rosa.liara.cloud
DB_PORT=31983
DB_NAME=awesome_swanson

در نهایت، می‌توانید مشابه قطعه کد زیر، به دیتابیس‌تان متصل شده و از آن، استفاده کنید:

کپی
import http.server
import socketserver
import mysql.connector
import os

db_config = {
    'user': os.getenv('DB_USER'),
    'password': os.getenv('DB_PASSWORD'),
    'host': os.getenv('DB_HOST'),
    'port': int(os.getenv('DB_PORT')),
    'database': os.getenv('DB_NAME')
}

def check_db_connection():
    try:
        connection = mysql.connector.connect(**db_config)
        if connection.is_connected():
            connection.close()
            return "connection successful"
    except mysql.connector.Error as e:
        return f"error:  {e}"
    return "connection failed"

class RequestHandler(http.server.SimpleHTTPRequestHandler):
    def do_GET(self):
        self.send_response(200)
        self.send_header("Content-type", "text/html; charset=utf-8")
        self.end_headers()

        connection_status = check_db_connection()

        html_content = f"""
        <!DOCTYPE html>
        <html lang="en">
        <head>
            <meta charset="UTF-8">
            <title>connection status</title>
            <style>
                body {{ font-family: Arial, sans-serif; background-color: #2B333F; color: #FFF; text-align: center; }}
                .status {{ margin-top: 20%; font-size: 24px; }}
            </style>
        </head>
        <body>
            <div class="status">{connection_status}</div>
        </body>
        </html>
        """
        self.wfile.write(html_content.encode("utf-8"))

PORT = 8000
with socketserver.TCPServer(("", PORT), RequestHandler) as httpd:
    print(f"Serving on port {PORT}")
    httpd.serve_forever()

استفاده از Connection Pooling

مفهوم Connection pooling به معنای استفاده از یک مجموعه اتصالات از پیش ساخته شده برای اتصال به پایگاه داده است. این تکنیک باعث می‌شود به جای ایجاد و بستن مکرر اتصالات، از اتصالات موجود در مجموعه استفاده شود که کارایی را افزایش می‌دهد.

برای استفاده از قابلیت connection pooling در دیتابیس MySQL، می‌توانید مشابه قطعه کد زیر، عمل کنید:

کپی
import http.server
import socketserver
from mysql.connector import pooling, Error
import os


db_config = {
    'user': os.getenv('DB_USER'),
    'password': os.getenv('DB_PASSWORD'),
    'host': os.getenv('DB_HOST'),
    'port': int(os.getenv('DB_PORT')),
    'database': os.getenv('DB_NAME')
}

try:
    connection_pool = pooling.MySQLConnectionPool(
        pool_name="mypool",
        pool_size=5,  # connection amount
        **db_config
    )
    print("Connection pool created successfully.")
except Error as e:
    print(f"Error while creating connection pool: {e}")

def check_db_connection():
    try:
        connection = connection_pool.get_connection()
        if connection.is_connected():
            connection.close()  
            return "connection successfull"
    except Error as e:
        return f"error: {e}"
    return "connection failed"

class RequestHandler(http.server.SimpleHTTPRequestHandler):
    def do_GET(self):
        self.send_response(200)
        self.send_header("Content-type", "text/html; charset=utf-8")
        self.end_headers()

        connection_status = check_db_connection()

        html_content = f"""
        <!DOCTYPE html>
        <html lang="en">
        <head>
            <meta charset="UTF-8">
            <title>connection status</title>
            <style>
                body {{ font-family: Arial, sans-serif; background-color: #2B333F; color: #FFF; text-align: center; }}
                .status {{ margin-top: 20%; font-size: 24px; }}
            </style>
        </head>
        <body>
            <div class="status">{connection_status}</div>
        </body>
        </html>
        """
        self.wfile.write(html_content.encode("utf-8"))

PORT = 8000
with socketserver.TCPServer(("", PORT), RequestHandler) as httpd:
    print(f"Serving on port {PORT}")
    httpd.serve_forever()