global: use defusedxml if it is available

This commit is contained in:
mathieui 2021-02-11 20:23:11 +01:00
parent bfbc433c8d
commit 53dc9847e2
2 changed files with 8 additions and 0 deletions

View file

@ -87,6 +87,7 @@ setup(
extras_require={ extras_require={
'XEP-0363': ['aiohttp'], 'XEP-0363': ['aiohttp'],
'XEP-0444 compliance': ['emoji'], 'XEP-0444 compliance': ['emoji'],
'Safer XML parsing': ['defusedxml'],
}, },
classifiers=CLASSIFIERS, classifiers=CLASSIFIERS,
cmdclass={'test': TestCommand} cmdclass={'test': TestCommand}

View file

@ -6,6 +6,13 @@
import logging import logging
logging.getLogger(__name__).addHandler(logging.NullHandler()) logging.getLogger(__name__).addHandler(logging.NullHandler())
# Use defusedxml if available
try:
import defusedxml
defusedxml.defuse_stdlib()
except ImportError:
pass
from slixmpp.stanza import Message, Presence, Iq from slixmpp.stanza import Message, Presence, Iq
from slixmpp.jid import JID, InvalidJID from slixmpp.jid import JID, InvalidJID
from slixmpp.xmlstream.stanzabase import ET, ElementBase, register_stanza_plugin from slixmpp.xmlstream.stanzabase import ET, ElementBase, register_stanza_plugin