소스 검색

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.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 = ""