mbirth/1pif-to-kdbx
1
0
Fork 0
Code Issues Pull Requests Activity

Add support for writing protected properties.

Browse Source
This commit is contained in:
Markus Birth 2021-08-20 02:58:11 +02:00
parent bb408537e5
commit 90fc1e3d24
Signed by: mbirth
GPG Key ID: A9928D7A098C3A9A
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
kpwriter.py
View File

@ -36,5 +36,11 @@ class KpWriter:
if not otp_url:
otp_url = "otpauth://totp/Sample:username?secret={}&algorithm=SHA1&digits=6&period=30&issuer=Sample".format(quote_plus(init_string))
# TODO: Support multiple / don't overwrite
self.current_entry.set_custom_property("TimeOtp-Secret-Base32", init_string)
self.current_entry.set_custom_property("otp", otp_url)
self.set_prop("TimeOtp-Secret-Base32", init_string, True)
self.set_prop("otp", otp_url)
def set_prop(self, key, value, protected=False):
self.current_entry.set_custom_property(key, value)
if protected:
# https://github.com/libkeepass/pykeepass/issues/89
self.current_entry._element.xpath('String[Key[text()="{}"]]/Value'.format(key))[0].attrib["Protected"] = "True"