jshiffer/slixmpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

XEP-0461: add get fallback body helper

Browse Source
This commit is contained in:
nicoco 2022-11-22 10:23:52 +01:00
parent 7300f1285e
commit 9b89401b36
2 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
slixmpp/plugins/xep_0461/stanza.py
View File

@ -18,6 +18,16 @@ class FeatureFallBack(ElementBase):
plugin_attrib = "feature_fallback" plugin_attrib = "feature_fallback"
interfaces = {"for"} interfaces = {"for"}
def get_fallback_body(self):
# only works for a single fallback_body attrib
start = self["fallback_body"]["start"]
end = self["fallback_body"]["end"]
body = self.parent()["body"]
if start < end < len(body):
return body[start:end]
else:
return ""
def get_stripped_body(self): def get_stripped_body(self):
# only works for a single fallback_body attrib # only works for a single fallback_body attrib
start = self["fallback_body"]["start"] start = self["fallback_body"]["start"]

10
tests/test_stanza_xep_0461.py
View File

@ -61,5 +61,15 @@ class TestReply(SlixTest):
""" """
) )
def testGetFallBackBody(self):
body = "Anna wrote:\nHi, how are you?"
quoted = "> Anna wrote:\n> Hi, how are you?\n"
msg = Message()
msg["body"] = "Great"
msg["feature_fallback"].add_quoted_fallback(body)
body2 = msg["feature_fallback"].get_fallback_body()
self.assertTrue(body2 == quoted, body2)
suite = unittest.TestLoader().loadTestsFromTestCase(TestReply) suite = unittest.TestLoader().loadTestsFromTestCase(TestReply)