Fixes MySQLdb feature

diff --git a/deefuzzer/station.py b/deefuzzer/station.py
index 123b3f2668639687ee34980339e8722ea5695ee5..896b324520c17792e9bb587e81e069cb03f0f29d 100644 (file)
--- 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)