#!/usr/bin/env python # -*- coding: utf-8 -*- # humbug-send -- Sends a message to the specified recipients. # Copyright © 2012 Zulip, Inc. # # Permission is hereby granted, free of charge, to any person obtaining a copy # of this software and associated documentation files (the "Software"), to deal # in the Software without restriction, including without limitation the rights # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell # copies of the Software, and to permit persons to whom the Software is # furnished to do so, subject to the following conditions: # # The above copyright notice and this permission notice shall be included in # all copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN # THE SOFTWARE. import sys import os import optparse import logging logging.basicConfig() log = logging.getLogger('humbug-send') def do_send_message(client, message_data ): '''Sends a message and optionally prints status about the same.''' if message_data['type'] == 'stream': log.info('Sending message to stream "%s", subject "%s"... ' % \ (message_data['to'], message_data['subject'])) else: log.info('Sending message to %s... ' % message_data['to']) response = client.send_message(message_data) if response['result'] == 'success': log.info('Message sent.') return True else: log.error(response['msg']) return False def main(argv=None): if argv is None: argv = sys.argv usage = """%prog [options] [recipient...] Sends a message specified recipients. Examples: %prog --stream denmark --subject castle -m "Something is rotten in the state of Denmark." %prog hamlet@example.com cordelia@example.com -m "Conscience doth make cowards of us all." These examples assume you have a proper '~/.zuliprc'. You may also set your credentials with the '--user' and '--api-key' arguments. """ sys.path.append(os.path.join(os.path.dirname(__file__), '..')) import zulip parser = optparse.OptionParser(usage=usage) # Grab parser options from the API common set parser.add_option_group(zulip.generate_option_group(parser)) parser.add_option('-m', '--message', help='Specifies the message to send, prevents interactive prompting.') group = optparse.OptionGroup(parser, 'Stream parameters') group.add_option('-s', '--stream', dest='stream', action='store', help='Allows the user to specify a stream for the message.') group.add_option('-S', '--subject', dest='subject', action='store', help='Allows the user to specify a subject for the message.') parser.add_option_group(group) (options, recipients) = parser.parse_args(argv[1:]) if options.verbose: logging.getLogger().setLevel(logging.INFO) # Sanity check user data if len(recipients) != 0 and (options.stream or options.subject): parser.error('You cannot specify both a username and a stream/subject.') if len(recipients) == 0 and (bool(options.stream) != bool(options.subject)): parser.error('Stream messages must have a subject') if len(recipients) == 0 and not (options.stream and options.subject): parser.error('You must specify a stream/subject or at least one recipient.') client = zulip.init_from_options(options) if not options.message: options.message = sys.stdin.read() if options.stream: message_data = { 'type': 'stream', 'content': options.message, 'subject': options.subject, 'to': options.stream, } else: message_data = { 'type': 'private', 'content': options.message, 'to': recipients, } if not do_send_message(client, message_data): return 1 if __name__ == '__main__': sys.exit(main())