Pārlūkot izejas kodu

better way to access node properties, __getattr__

oz123 11 gadi atpakaļ
vecāks
revīzija
e819e42cba
1 mainītis faili ar 11 papildinājumiem un 0 dzēšanām
  1. 11 0
      pwman/data/nodes.py

+ 11 - 0
pwman/data/nodes.py

@@ -35,6 +35,17 @@ class NewNode(object):
         self._tags = tags
         self.set_tags(tags)
 
+    def __getattr__(self, name):
+        enc = CryptoEngine.get()
+        if name == 'username':
+            return enc.decrypt(self._username).strip()
+        elif name == 'password':
+            return enc.decrypt(self._password).strip()
+        elif name == 'url':
+            return enc.decrypt(self._url).strip()
+        elif name == 'notes':
+            return enc.decrypt(self._notes).strip()
+
     def dump_edit_to_db(self):
         enc = CryptoEngine.get()
         dump = ""