import logging import argparse from sleekxmpp import ClientXMPP from sleekxmpp.exceptions import IqError, IqTimeout import os import urllib import datetime import ssl from PIL import Image parser = argparse.ArgumentParser() parser.add_argument("-j", "--jid", help="jabber identifier", type=str, required=True) parser.add_argument("-p", "--password", help="password", type=str, required=True) parser.add_argument("-m", "--muc", help="destination muc", type=str, required=True) parser.add_argument("-n", "--nick", help="nickname of the bot", default="archivist", type=str) parser.add_argument("-o", "--output", help="output folder", default="files/", type=str) args = parser.parse_args() class ArchivistBot(ClientXMPP): def __init__(self, jid, password, room, nick, output): ClientXMPP.__init__(self, jid, password) self.datadir = output self.room = room self.nick = nick self.add_event_handler("session_start", self.session_start) self.add_event_handler("message", self.archive_msg) # by using 'message' instead of 'groupchat_message' every message received can be archived (also personal msgs) # self.add_event_handler("groupchat_message", self.archive_msg2) self.register_plugin('xep_0045') self.register_plugin('xep_0030') self.register_plugin('xep_0084') self.register_plugin('xep_0096') # Transfer files.. self.register_plugin('xep_0066') # Transfer files.. def session_start(self, event): self.get_roster() self.send_presence() self.plugin['xep_0045'].joinMUC(self.room, self.nick) # XEP-0084 User Avatar # Requires SleekXMPP 81b7b2c1908e0f6a5435ce67745b5f4dafb59816 with open('contact.png', 'rb') as avatar_file: avatar = avatar_file.read() avatar_id = self['xep_0084'].generate_id(avatar) info = { 'id': avatar_id, 'type': 'image/jpeg', 'bytes': len(avatar) } self['xep_0084'].publish_avatar(avatar) self['xep_0084'].publish_avatar_metadata(items=[info]) # XEP-0153: vCard-Based Avatars # Not working ATM self['xep_0153'].set_avatar(avatar=avatar, mtype='image/png') def archive_msg(self, msg): # Always check that a message is not from yourself, otherwise you will create an infinite loop responding to your own messages. if 'mucnick' in msg and msg['mucnick'] == self.nick: return if msg['oob']['url']: logging.getLogger().debug("received OOB from %s with %s" % (self.nick, msg['oob']['url'])) filename = os.path.basename(msg['oob']['url']) targetDir = self.datadir if not os.path.exists(targetDir): os.mkdir( targetDir, 0755 ) targetFile = os.path.join(targetDir, filename) #needed to disable certificate validation: ctx = ssl.create_default_context() ctx.check_hostname = False ctx.verify_mode = ssl.CERT_NONE # save image to file urllib.urlretrieve(msg['oob']['url'], targetFile, context=ctx) logging.getLogger().debug("saved to %s" % targetFile) # if msg['type'] in ('chat', 'normal'): # msg.reply("Thanks for sending\n%(body)s" % msg).send() if __name__ == '__main__': # Ideally use optparse or argparse to get JID, # password, and log level. logging.basicConfig(level=logging.DEBUG, format='%(levelname)-8s %(message)s') client = ArchivistBot(args.jid, args.password, args.muc, args.nick, args.output) if client.connect(): client.process(block=True) else: logging.getLogger().error("Can't connect.")