Quellcode durchsuchen

add __str__ methods to NewNode and Tag classes

oz123 vor 11 Jahren
Ursprung
Commit
b76ce4e9c0
2 geänderte Dateien mit 17 neuen und 2 gelöschten Zeilen
  1. 13 0
      pwman/data/nodes.py
  2. 4 2
      pwman/data/tags.py

+ 13 - 0
pwman/data/nodes.py

@@ -35,6 +35,19 @@ class NewNode(object):
         self._notes = notes
         self._tags = tags
 
+    def __str__(self):
+        enc = CryptoEngine.get()
+        try:
+            tags = ', '.join([enc.decrypt(tag).strip() for tag in filter(None,
+                             self._tags)])
+        except Exception:
+            tags = ', '.join([tag.strip() for tag in filter(None, self._tags)])
+
+        user = enc.decrypt(self._username).strip()
+        url = enc.decrypt(self._url).strip()
+        return '{0}@{1}\t{2}'.format(user, url,
+                                     tags)
+
     #def __setattr__(self, name, value):
     #    if name in ['username', 'password', 'url', 'notes']:
     #        enc = CryptoEngine.get()

+ 4 - 2
pwman/data/tags.py

@@ -39,6 +39,10 @@ class Tag(object):
         else:
             return False
 
+    def __str__(self):
+        enc = CryptoEngine.get()
+        return enc.decrypt(self._name)
+
     @property
     def name(self):
         enc = CryptoEngine.get()
@@ -48,5 +52,3 @@ class Tag(object):
     def name(self, value):
         enc = CryptoEngine.get()
         self._name = enc.encrypt(value)
-
-