ppf: Update DatabaseAccess classes to initialize database in constructor.

[cs-p2p-next.git] / ppf / new / storage.py

diff --git a/ppf/new/storage.py b/ppf/new/storage.py
index 854a243..2bbc257 100644 (file)
--- a/ppf/new/storage.py
+++ b/ppf/new/storage.py
@@ -403,11 +403,10 @@ class TreeTextFileAccess(FileAccess):
         f.close()
 
 class DatabaseAccess(SwarmDataAccess):
-    def __init__(self):
-        pass
-
-    def connect(self, database):
+    def __init__(self, database):
         self.database = database
+
+    def connect(self):
         self.conn = None
         self.cursor = None
 
@@ -416,11 +415,10 @@ class DatabaseAccess(SwarmDataAccess):
         self.conn.close()
 
 class SQLiteDatabaseAccess(DatabaseAccess):
-    def __init___(self):
-        pass
+    def __init___(self, database):
+        super(SQLiteDatabaseAccess, self).__init__(database)
 
-    def connect(self, database):
-        self.database = database
+    def connect(self):
         self.conn = sqlite3.connect(self.database)
         self.cursor = self.conn.cursor()
 
@@ -440,10 +438,10 @@ class SQLiteDatabaseAccess(DatabaseAccess):
         pass
 
 class MySQLDatabaseAccess(DatabaseAccess):
-    def __init___(self):
-        pass
+    def __init___(self, database):
+        super(SQLiteDatabaseAccess, self).__init__(database)
 
-    def connect(self, database):
+    def connect(self):
         pass
 
     def add_swarm(self, swarm):