Add option to write to kdbx databases

libkeepass · Jul 28, 2017 · a1c9e14 · a1c9e14
1 parent 812f225
commit a1c9e14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libkeepass/__init__.py b/libkeepass/__init__.py
@@ -24,7 +24,7 @@ def getbuffer(self):
 }
 
 @contextmanager
-def open(filename, **credentials):
+def open(filename, readwrite=False, **credentials):
     """
     A contextmanager to open the KeePass file with `filename`. Use a `password`
     and/or `keyfile` named argument for decryption.
@@ -36,7 +36,7 @@ def open(filename, **credentials):
     """
     kdb = None
     try:
-        with io.open(filename, 'rb') as stream:
+        with io.open(filename, 'rb' if readwrite == False else 'r+b') as stream:
             signature = common.read_signature(stream)
             cls = get_kdb_reader(signature)
             kdb = cls(stream, **credentials)