From 825167b8aa69ddcf3261b8a81c6e09c2d01baa71 Mon Sep 17 00:00:00 2001 From: nico Date: Wed, 3 Oct 2018 23:24:36 +0200 Subject: Squashed commit of the following: commit 91cf4b74105f69f16a80825581a0d2d34d15d155 Merge: 30ceaea 90b42ed Author: nico Date: Wed Oct 3 22:34:42 2018 +0200 Merge remote-tracking branch 'origin/xep' into xep commit 30ceaea56a77ed95deba30c4fe65e238ea5960ac Author: nico Date: Wed Oct 3 22:34:33 2018 +0200 Initial Version XEP query + added initial version of xep query class Init Implementation + added xep plugin to bot class * reworked validation function * updated .gitignore file + added xep plugin commit 90b42edb9b8e92eba3bb67030d5f919b1e71d0bc Author: nico Date: Wed Oct 3 22:34:21 2018 +0200 * reworked validation function * updated .gitignore file + added xep plugin commit 25c78807731417867840d6fe4abf598e64aded28 Author: nico Date: Wed Oct 3 10:54:02 2018 +0200 Init Implementation + added xep plugin to bot class commit fe711f44d40671d927e9b946fb66679b297272c8 Author: nico Date: Tue Oct 2 21:20:08 2018 +0200 Initial Version XEP query + added initial version of xep query class --- classes/xep.py | 88 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 88 insertions(+) create mode 100644 classes/xep.py (limited to 'classes/xep.py') diff --git a/classes/xep.py b/classes/xep.py new file mode 100644 index 0000000..6293c27 --- /dev/null +++ b/classes/xep.py @@ -0,0 +1,88 @@ +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- +import requests +import xml.etree.ElementTree as ET + + +class XEPRequest: + def __init__(self, msg, xepnumber): + """ + class which requests the header of the referenced xep + :param xepnumber: number int or str to request the xep for + """ + self.message_type = msg['type'] + self.muc_nick = msg['mucnick'] + + self.reqxep = str(xepnumber) + self.xeplist = None + self.acceptedxeps = list() + + def req_xeplist(self): + """ + query and save the current xep list to reduce network bandwidth + """ + try: + with open(".etag") as file: + local_etag = file.read() + except FileExistsError: + local_etag = "" + pass + + with requests.Session() as s: + s.headers.update({'Accept': 'application/xml'}) + head = s.head("https://xmpp.org/extensions/xeplist.xml") + etag = head.headers['etag'] + + if local_etag == etag: + with open("xeplist.xml", "r") as file: + self.xeplist = ET.fromstring(file.read()) + else: + r = s.get("https://xmpp.org/extensions/xeplist.xml") + r.encoding = 'utf-8' + local_etag = head.headers['etag'] + + with open("xeplist.xml", "w") as file: + file.write(r.content.decode()) + self.xeplist = ET.fromstring(r.content.decode()) + + with open('.etag', 'w') as string: + string.write(local_etag) + + # populate xep comparison list + for xep in self.xeplist.findall(".//*[@accepted='true']/number"): + self.acceptedxeps.append(xep.text) + + def get(self): + """ + function to query the xep entry if xepnumber is present in xeplist + :return: nicely formatted xep header information + """ + # check if xeplist is accurate + self.req_xeplist() + + result = list() + # if requested number is inside acceptedxeps continou + if self.reqxep in self.acceptedxeps: + searchstring = ".//*[@accepted='true']/[number='%s']" % self.reqxep + + for item in self.xeplist.findall(searchstring): + for x in range(1,5): + result.append(item[x].tag + " : " + item[x].text) + + else: + if self.message_type == "groupchat": + result.append(self.muc_nick + " : " + "XEP-" + str(self.reqxep) + " : is not available.") + else: + result.append("XEP-" + str(self.reqxep) + " : is not available.") + + return result + + def format(self): + reply = self.get() + if self.message_type == "groupchat": + text = "%s: " % self.muc_nick + reply[0] = text + reply[0] + + text = '\n'.join(reply) + + return text -- cgit v1.2.3-54-g00ecf From 6c6c14b55fdbf3f79e21f961354952190b443a6b Mon Sep 17 00:00:00 2001 From: nico Date: Wed, 3 Oct 2018 23:36:59 +0200 Subject: * switched xml.etreeElementTree for defusexml - removed unnecessary pass statement --- classes/xep.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'classes/xep.py') diff --git a/classes/xep.py b/classes/xep.py index 6293c27..9e4f61f 100644 --- a/classes/xep.py +++ b/classes/xep.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 # -*- coding: utf-8 -*- import requests -import xml.etree.ElementTree as ET +import defusedxml.ElementTree as ET class XEPRequest: @@ -24,9 +24,8 @@ class XEPRequest: try: with open(".etag") as file: local_etag = file.read() - except FileExistsError: + except FileNotFoundError: local_etag = "" - pass with requests.Session() as s: s.headers.update({'Accept': 'application/xml'}) -- cgit v1.2.3-54-g00ecf