|
@@ -55,7 +55,8 @@ class CSVImporter(BaseImporter):
|
|
def _create_node(self, row):
|
|
def _create_node(self, row):
|
|
"""create a node object with encrypted properties"""
|
|
"""create a node object with encrypted properties"""
|
|
n = {'clear_text': True,
|
|
n = {'clear_text': True,
|
|
- 'username': row[0], 'password': row[2], 'url': row[1], 'notes': row[3],
|
|
|
|
|
|
+ 'username': row[0], 'password': row[2], 'url': row[1],
|
|
|
|
+ 'notes': row[3],
|
|
'tags': row[4].split(',')}
|
|
'tags': row[4].split(',')}
|
|
node = Node(**n)
|
|
node = Node(**n)
|
|
return node
|
|
return node
|
|
@@ -76,13 +77,14 @@ class CSVImporter(BaseImporter):
|
|
self._db._con.commit()
|
|
self._db._con.commit()
|
|
self._db.open()
|
|
self._db.open()
|
|
|
|
|
|
- def runner(self):
|
|
|
|
|
|
+ def run(self):
|
|
self._open_db()
|
|
self._open_db()
|
|
|
|
|
|
for row in self._read_file():
|
|
for row in self._read_file():
|
|
node = self._create_node(row)
|
|
node = self._create_node(row)
|
|
self._insert_node(node)
|
|
self._insert_node(node)
|
|
|
|
|
|
|
|
+
|
|
class Importer(object):
|
|
class Importer(object):
|
|
|
|
|
|
"""
|
|
"""
|
|
@@ -90,8 +92,8 @@ class Importer(object):
|
|
This could be changes by calling other instance which for example import
|
|
This could be changes by calling other instance which for example import
|
|
from KeePass XML or other formats.
|
|
from KeePass XML or other formats.
|
|
"""
|
|
"""
|
|
- def __init__(self, invoke=CSVImporter):
|
|
|
|
- self.runner = invoke()
|
|
|
|
|
|
+ def __init__(self, args, invoke=CSVImporter):
|
|
|
|
+ self.importer = invoke(*args)
|
|
|
|
|
|
def run(self):
|
|
def run(self):
|
|
- self.runner()
|
|
|
|
|
|
+ self.importer.run()
|