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

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

کپی

pip install mysql-connector-python

سپس، با فرض فعال بودن محیط مجازی در پروژه پایتونی‌تان (virtualenv)، دستور زیر را اجرا کنید تا فایل requirements.txt به‌روز شود:

کپی

pip freeze > requirements.txt

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

کپی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

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

کپی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()

ویرایش در گیت‌هاب