87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
import pymysql
|
||
|
||
|
||
# ==============================数据库模块===================================
|
||
def check_link_exists(host, user, password, database, link):
|
||
"""
|
||
检查指定的 link 是否存在于 MySQL 数据库表中，如果不存在，则插入该链接
|
||
:param host: MySQL 数据库主机地址
|
||
:param user: MySQL 用户名
|
||
:param password: MySQL 密码
|
||
:param database: 数据库名称
|
||
:param link: 需要检查的链接
|
||
:return: 如果链接存在，返回 True；如果链接不存在且插入成功，返回 False
|
||
"""
|
||
connection = None # 确保 connection 被初始化
|
||
|
||
try:
|
||
# 连接到 MySQL 数据库
|
||
connection = pymysql.connect(
|
||
host=host,
|
||
user=user,
|
||
password=password,
|
||
database=database
|
||
)
|
||
|
||
with connection.cursor() as cursor:
|
||
# 查询链接是否存在
|
||
cursor.execute("SELECT 1 FROM links WHERE link = %s", (link,))
|
||
result = cursor.fetchone()
|
||
|
||
# 如果链接存在
|
||
if result:
|
||
return True
|
||
else:
|
||
return False
|
||
|
||
except pymysql.MySQLError as e:
|
||
print(f"数据库错误: {e}")
|
||
return False
|
||
finally:
|
||
# 确保在结束时关闭连接
|
||
if connection:
|
||
connection.close()
|
||
|
||
|
||
def check_link_insert(host, user, password, database, link):
|
||
"""
|
||
检查指定的 link 是否存在于 MySQL 数据库表中，如果不存在，则插入该链接
|
||
:param host: MySQL 数据库主机地址
|
||
:param user: MySQL 用户名
|
||
:param password: MySQL 密码
|
||
:param database: 数据库名称
|
||
:param link: 需要检查的链接
|
||
:return: 如果链接存在，返回 True；如果链接不存在且插入成功，返回 False
|
||
"""
|
||
connection = None # 确保 connection 被初始化
|
||
try:
|
||
# 连接到 MySQL 数据库
|
||
connection = pymysql.connect(
|
||
host=host,
|
||
user=user,
|
||
password=password,
|
||
database=database
|
||
)
|
||
|
||
with connection.cursor() as cursor:
|
||
# 查询链接是否存在
|
||
cursor.execute("SELECT 1 FROM links WHERE link = %s", (link,))
|
||
result = cursor.fetchone()
|
||
if result:
|
||
# 如果链接已经存在，返回 True
|
||
return True
|
||
else:
|
||
# 插入链接
|
||
cursor.execute("INSERT INTO links (link) VALUES (%s)", (link,))
|
||
connection.commit() # 提交事务
|
||
print("链接已插入")
|
||
return False
|
||
except pymysql.MySQLError as e:
|
||
print(f"数据库错误: {e}")
|
||
return False
|
||
finally:
|
||
# 确保在结束时关闭连接
|
||
if connection:
|
||
connection.close()
|
||
|