From 27279d3306165f4f3704ac8130eb3d4c3ba0eb11 Mon Sep 17 00:00:00 2001 From: Thomas Sileo Date: Tue, 16 Apr 2019 22:40:45 +0200 Subject: [PATCH] Add new get_in_reply_to helper --- little_boxes/activitypub.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/little_boxes/activitypub.py b/little_boxes/activitypub.py index 91462f3..5ec5803 100644 --- a/little_boxes/activitypub.py +++ b/little_boxes/activitypub.py @@ -914,6 +914,16 @@ class Note(BaseActivity): return False + def get_in_reply_to(self) -> Optional[str]: + if self.inReplyTo is None: + return None + elif isinstance(self.inReplyTo, str): + return self.inReplyTo + elif isinstance(self.inReplyTo, dict): + return self.inReplyTo["id"] + else: + raise ValueError(f"unexpected inReplyTo: {self.inReplyTo}") + class Question(Note): ACTIVITY_TYPE = ActivityType.QUESTION