23 lines
534 B
Python
23 lines
534 B
Python
import mysql.connector
|
|
from mysql.connector import Error
|
|
import config
|
|
|
|
def get_db_connection():
|
|
"""创建数据库连接"""
|
|
try:
|
|
conn = mysql.connector.connect(
|
|
host=config.DB_HOST,
|
|
user=config.DB_USER,
|
|
password=config.DB_PASSWORD,
|
|
database=config.DB_NAME
|
|
)
|
|
return conn
|
|
except Error as e:
|
|
print(f"数据库连接错误: {e}")
|
|
return None
|
|
|
|
def close_connection(conn):
|
|
"""关闭数据库连接"""
|
|
if conn:
|
|
conn.close()
|