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

Type hints in OnepifEntry.

Browse Source
This commit is contained in:
Markus Birth 2021-08-22 17:07:48 +02:00
parent 6b3c2a157c
commit 1c5804deab
Signed by: mbirth
GPG Key ID: A9928D7A098C3A9A
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
onepif/OnepifEntry.py
View File

@ -28,20 +28,20 @@ class OnepifEntry():
self.raw = data self.raw = data
self.set_type(data["typeName"]) self.set_type(data["typeName"])
def set_type(self, new_type): def set_type(self, new_type: str):
if new_type not in TYPES: if new_type not in TYPES:
raise Exception("Unknown record type: {}".format(new_type)) raise Exception("Unknown record type: {}".format(new_type))
self.type = new_type self.type = new_type
self.type_name = TYPES[new_type] self.type_name = TYPES[new_type]
def get_tags(self): def get_tags(self) -> list:
if "openContents" not in self.raw: if "openContents" not in self.raw:
return [] return []
if "tags" not in self.raw["openContents"]: if "tags" not in self.raw["openContents"]:
return [] return []
return self.raw["openContents"]["tags"] return self.raw["openContents"]["tags"]
def get_totps(self): def get_totps(self) -> list:
totp_fields = [] totp_fields = []
if "sections" in self.raw["secureContents"]: if "sections" in self.raw["secureContents"]:
for section in self.raw["secureContents"]["sections"]: for section in self.raw["secureContents"]["sections"]:
@ -57,7 +57,7 @@ class OnepifEntry():
return None return None
return totp_fields return totp_fields
def is_trash(self): def is_trash(self) -> bool:
if "trashed" in self.raw: if "trashed" in self.raw:
return self.raw["trashed"] return self.raw["trashed"]
return False return False