瀏覽代碼

- init method takes a file name a parameter

oz123 11 年之前
父節點
當前提交
ede2430810
共有 1 個文件被更改,包括 9 次插入7 次删除
  1. 9 7
      pwman/data/drivers/sqlite.py

+ 9 - 7
pwman/data/drivers/sqlite.py

@@ -63,15 +63,17 @@ def check_db_version():
 class SQLiteDatabaseNewForm(Database):
     """SQLite Database implementation"""
 
-    def __init__(self):
+    def __init__(self, filename=None):
         """Initialise SQLitePwmanDatabase instance."""
         Database.__init__(self)
-
-        try:
-            self._filename = config.get_value('Database', 'filename')
-        except KeyError, e:
-            raise DatabaseException(
-                "SQLite: missing parameter [%s]" % (e))
+        if filename:
+            self._filename = filename
+        else:
+            try:
+                self._filename = config.get_value('Database', 'filename')
+            except KeyError, e:
+                raise DatabaseException(
+                    "SQLite: missing parameter [%s]" % (e))
 
     def _open(self):
         try: