fix: 修正代理数据提取逻辑以使用字典格式

This commit is contained in:
晓丰 2025-05-17 11:40:43 +08:00
parent 7822ef9943
commit d310c8a418

6
DB.py
View File

@ -108,7 +108,7 @@ class DBVidcon:
sql = "SELECT rn, parameter FROM proxy_agent" sql = "SELECT rn, parameter FROM proxy_agent"
self.cursor.execute(sql) self.cursor.execute(sql)
rows = self.cursor.fetchall() rows = self.cursor.fetchall()
result = {rn: param for rn, param in rows} result = {row['rn']: row['parameter'] for row in rows}
self.close() self.close()
return result return result
@ -245,8 +245,6 @@ class DBVidcon:
print("[交由Redis处理]") print("[交由Redis处理]")
break break
def flush(self): def flush(self):
"""批量执行完后手动提交。""" """批量执行完后手动提交。"""
self.conn.commit() self.conn.commit()
@ -269,4 +267,4 @@ class DBVidcon:
time.sleep(10) time.sleep(10)
else: else:
break break
return proxy return proxy