浏览代码

better way to access node properties, __getattr__

oz123 11 年之前
父节点
当前提交
e819e42cba
共有 1 个文件被更改,包括 11 次插入0 次删除
  1. 11 0
      pwman/data/nodes.py

+ 11 - 0
pwman/data/nodes.py

@@ -35,6 +35,17 @@ class NewNode(object):
         self._tags = tags
         self._tags = tags
         self.set_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):
     def dump_edit_to_db(self):
         enc = CryptoEngine.get()
         enc = CryptoEngine.get()
         dump = ""
         dump = ""