Sfoglia il codice sorgente

Add a class for converting the key

oz123 10 anni fa
parent
commit
fc57f9e11b
1 ha cambiato i file con 18 aggiunte e 2 eliminazioni
  1. 18 2
      pwman/data/convertdb.py

+ 18 - 2
pwman/data/convertdb.py

@@ -37,7 +37,7 @@ _NEWVERSION = 0.4
 class SQLiteDatabaseReader(Database):
     """SQLite Database implementation"""
 
-    def __init__(self):
+    def __init__(self, filename=None):
         """Initialise SQLitePwmanDatabase instance."""
         Database.__init__(self)
 
@@ -45,7 +45,7 @@ class SQLiteDatabaseReader(Database):
             self._filename = config.get_value('Database', 'filename')
         except KeyError as e:
             raise DatabaseException(
-                "SQLite: missing parameter [%s]" % (e))
+                "SQLite: missing config parameter [%s]" % (e))
 
     def _open(self):
         try:
@@ -282,3 +282,19 @@ class PwmanConvertDB(DBConverter):
         enc = CryptoEngine.get()
         self.oldkey = enc.get_cryptedkey()
         self.newdb.savekey(self.oldkey)
+
+
+class PwmanConvertKey(DBConverter):
+
+    def read_old_db(self):
+        enc = CryptoEngine.get()
+        enc.set_callback(CLICallback())
+        self.db.open()
+        self.oldnodes = self.db.listnodes()
+        self.oldnodes = self.db.getnodes(self.oldnodes)
+
+    def save_old_key(self):
+        CryptoEngine._instance = None
+        enc = CryptoEngine.get(0.5)
+        self.oldkey = enc.get_cryptedkey()
+        self.newdb.savekey(self.oldkey)