Files
reverse_image_search_gpu/models/mysql.py
2024-11-04 05:46:22 +08:00

25 lines
677 B
Python

import pymysql
from configs.config import MYSQL_HOST, MYSQL_PORT, MYSQL_NAME, MYSQL_USER, MYSQL_PASS
# 连接 MySQL (开启 MySQL 服务)
conn = pymysql.connect(
host=MYSQL_HOST,
user=MYSQL_USER,
port=MYSQL_HOST,
password=MYSQL_PASS,
database=MYSQL_NAME,
local_infile=True,
cursorclass=pymysql.cursors.DictCursor
)
# 获取 MySQL 连接
def get_cursor():
try:
dx = conn.ping()
return conn.cursor()
except Exception:
conn = pymysql.connect(host=MYSQL_HOST, user="gameui", port=3306, password="gameui@2022", database='gameui', local_infile=True, cursorclass=pymysql.cursors.DictCursor)
return conn.cursor()