From db66f7a1bb643c74b999318359168811f7610734 Mon Sep 17 00:00:00 2001 From: Dominic Kanrabat Date: Wed, 24 Apr 2019 03:34:09 -0400 Subject: [PATCH] Fixes MySQLdb feature --- deefuzzer/station.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/deefuzzer/station.py b/deefuzzer/station.py index 123b3f2..896b324 100644 --- a/deefuzzer/station.py +++ b/deefuzzer/station.py @@ -440,12 +440,18 @@ class Station(Thread): port = self.mdb_port, use_unicode = True) cur = con.cursor() - cur.execute("SELECT %s FROM %s".format(self.mdb_field, self.mdb_table)) - file_list = cur.fetchall() + cur.execute("SELECT %s FROM %s" % (self.mdb_field, self.mdb_table)) + rows = cur.fetchall() + for row in rows: + file_list.append(row[0]) except mdb.Error, e: self._err('Could not get playlist from MySQLdb, Error %d: %s' % (e.args[0], e.args[1])) + finally: + if con: + con.close() + # Directory elif os.path.isdir(self.media_source): self.q.get(1) -- 2.47.3