From 18c9fd5a0cc115dedaa201288c1d7e4bbba265aa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20H=C3=B6rist?= <philipp@hoerist.com>
Date: Fri, 16 Aug 2019 17:19:37 +0200
Subject: DiscoInfo: Add timestamp attribute

---
 nbxmpp/modules/discovery.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'nbxmpp/modules')

diff --git a/nbxmpp/modules/discovery.py b/nbxmpp/modules/discovery.py
index 47510b6..c849afd 100644
--- a/nbxmpp/modules/discovery.py
+++ b/nbxmpp/modules/discovery.py
@@ -15,6 +15,7 @@
 # You should have received a copy of the GNU General Public License
 # along with this program; If not, see <http://www.gnu.org/licenses/>.
 
+import time
 import logging
 
 from nbxmpp.protocol import Iq
@@ -63,11 +64,14 @@ class Discovery:
         return parse_disco_items(stanza)
 
 
-def parse_disco_info(stanza):
+def parse_disco_info(stanza, timestamp=None):
     idenities = []
     features = []
     dataforms = []
 
+    if timestamp is None:
+        timestamp = time.time()
+
     query = stanza.getQuery()
     for node in query.getTags('identity'):
         attrs = node.getAttrs()
@@ -92,7 +96,8 @@ def parse_disco_info(stanza):
     return DiscoInfo(stanza=stanza,
                      identities=idenities,
                      features=features,
-                     dataforms=dataforms)
+                     dataforms=dataforms,
+                     timestamp=timestamp)
 
 
 def parse_disco_items(stanza):
-- 
cgit v1.2.3