浏览代码

- init method takes a file name a parameter

oz123 12 年之前
父节点
当前提交
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):
 class SQLiteDatabaseNewForm(Database):
     """SQLite Database implementation"""
     """SQLite Database implementation"""
 
 
-    def __init__(self):
+    def __init__(self, filename=None):
         """Initialise SQLitePwmanDatabase instance."""
         """Initialise SQLitePwmanDatabase instance."""
         Database.__init__(self)
         Database.__init__(self)
-
+        if filename:
-        try:
+            self._filename = filename
-            self._filename = config.get_value('Database', 'filename')
+        else:
-        except KeyError, e:
+            try:
-            raise DatabaseException(
+                self._filename = config.get_value('Database', 'filename')
-                "SQLite: missing parameter [%s]" % (e))
+            except KeyError, e:
+                raise DatabaseException(
+                    "SQLite: missing parameter [%s]" % (e))
 
 
     def _open(self):
     def _open(self):
         try:
         try: