diff --git a/DB.py b/DB.py index cceec3b..d8d9a50 100644 --- a/DB.py +++ b/DB.py @@ -109,14 +109,12 @@ class DBVidcon: self.cursor.execute(sql) rows = self.cursor.fetchall() result = {row['rn']: row['parameter'] for row in rows} - self.close() return result def get_proxy_parameter(self, rn: str) -> str: sql = "SELECT parameter FROM proxy_agent WHERE rn = %s LIMIT 1" self.cursor.execute(sql, (rn,)) result = self.cursor.fetchone() - self.close() return result[0] if result else None def rollback_records(self, raws):