From 2658a4b432110e5f3c9db290282e75bade06a17f Mon Sep 17 00:00:00 2001
From: kali <kalikaneko@riseup.net>
Date: Sat, 2 May 2020 21:27:58 +0200
Subject: add group info to message

---
 signald/main.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'signald/main.py')

diff --git a/signald/main.py b/signald/main.py
index e82f3e7..d4ece1d 100644
--- a/signald/main.py
+++ b/signald/main.py
@@ -107,7 +107,7 @@ class Signal:
                 continue
 
             message = message["data"]
-            data_message = message["dataMessage"] if message.get("dataMessage") else {}
+            data_message = message.get("dataMessage", {})
 
             yield Message(
                 username=message["username"],
@@ -118,6 +118,7 @@ class Signal:
                 timestamp_iso=message["timestampISO"],
                 expiration_secs=data_message.get("expiresInSeconds"),
                 is_receipt=message["isReceipt"],
+                group_info=data_message.get("groupInfo", {}),
                 attachments=[
                     Attachment(
                         content_type=attachment["contentType"],
-- 
cgit v1.2.3-54-g00ecf