#!/usr/bin/env python3 # Slixmpp: The Slick XMPP Library # Copyright (C) 2018 Emmanuel Gil Peyrot # This file is part of Slixmpp. # See the file LICENSE for copying permission. import logging from getpass import getpass from argparse import ArgumentParser import slixmpp from slixmpp.exceptions import IqTimeout log = logging.getLogger(__name__) class HttpUpload(slixmpp.ClientXMPP): """ A basic client asking an entity if they confirm the access to an HTTP URL. """ def __init__(self, jid, password, recipient, filename, domain=None): slixmpp.ClientXMPP.__init__(self, jid, password) self.recipient = recipient self.filename = filename self.domain = domain self.add_event_handler("session_start", self.start) async def start(self, event): log.info('Uploading file %s...', self.filename) try: url = await self['xep_0363'].upload_file( self.filename, domain=self.domain, timeout=10 ) except IqTimeout: raise TimeoutError('Could not send message in time') log.info('Upload success!') log.info('Sending file to %s', self.recipient) html = ( f'' f'{url}' ) message = self.make_message(mto=self.recipient, mbody=url, mhtml=html) message['oob']['url'] = url message.send() self.disconnect() if __name__ == '__main__': # Setup the command line arguments. parser = ArgumentParser() parser.add_argument("-q","--quiet", help="set logging to ERROR", action="store_const", dest="loglevel", const=logging.ERROR, default=logging.INFO) parser.add_argument("-d","--debug", help="set logging to DEBUG", action="store_const", dest="loglevel", const=logging.DEBUG, default=logging.INFO) # JID and password options. parser.add_argument("-j", "--jid", dest="jid", help="JID to use") parser.add_argument("-p", "--password", dest="password", help="password to use") # Other options. parser.add_argument("-r", "--recipient", required=True, help="Recipient JID") parser.add_argument("-f", "--file", required=True, help="File to send") parser.add_argument("--domain", help="Domain to use for HTTP File Upload (leave out for your own server’s)") args = parser.parse_args() # Setup logging. logging.basicConfig(level=args.loglevel, format='%(levelname)-8s %(message)s') if args.jid is None: args.jid = input("Username: ") if args.password is None: args.password = getpass("Password: ") xmpp = HttpUpload(args.jid, args.password, args.recipient, args.file, args.domain) xmpp.register_plugin('xep_0066') xmpp.register_plugin('xep_0071') xmpp.register_plugin('xep_0128') xmpp.register_plugin('xep_0363') # Connect to the XMPP server and start processing XMPP stanzas. xmpp.connect() xmpp.process(forever=False)