28 lines
704 B
Python
28 lines
704 B
Python
import pymysql
|
|
from configs.config import MYSQL_HOST, MYSQL_PORT, MYSQL_NAME, MYSQL_USER, MYSQL_PASS
|
|
|
|
# 创建 MySQL 连接
|
|
def create_connection():
|
|
return pymysql.connect(
|
|
host=MYSQL_HOST,
|
|
user=MYSQL_USER,
|
|
port=MYSQL_PORT, # 应该使用 MYSQL_PORT 而不是 MYSQL_HOST
|
|
password=MYSQL_PASS,
|
|
database=MYSQL_NAME,
|
|
local_infile=True,
|
|
cursorclass=pymysql.cursors.DictCursor
|
|
)
|
|
|
|
# 连接 MySQL (开启 MySQL 服务)
|
|
conn = create_connection()
|
|
|
|
# 获取 MySQL 连接
|
|
def get_cursor():
|
|
global conn
|
|
try:
|
|
conn.ping()
|
|
return conn.cursor()
|
|
except Exception:
|
|
conn = create_connection()
|
|
return conn.cursor()
|