102 lines
3.3 KiB
Python
Executable file
102 lines
3.3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
# zulip-send -- Sends a message to the specified recipients.
|
|
|
|
import sys
|
|
import argparse
|
|
import logging
|
|
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
import zulip
|
|
|
|
logging.basicConfig()
|
|
|
|
log = logging.getLogger('zulip-send')
|
|
|
|
def do_send_message(client, message_data):
|
|
# type: (zulip.Client, Dict[str, Any]) -> bool
|
|
'''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():
|
|
# type: () -> int
|
|
usage = """zulip-send [options] [recipient...]
|
|
|
|
Sends a message to specified recipients.
|
|
|
|
Examples: zulip-send --stream denmark --subject castle -m "Something is rotten in the state of Denmark."
|
|
zulip-send hamlet@example.com cordelia@example.com -m "Conscience doth make cowards of us all."
|
|
|
|
Specify your Zulip API credentials and server in a ~/.zuliprc file or using the options.
|
|
"""
|
|
|
|
parser = zulip.add_default_arguments(argparse.ArgumentParser(usage=usage))
|
|
|
|
parser.add_argument('recipients',
|
|
nargs='*',
|
|
help='email addresses of the recipients of the message')
|
|
|
|
parser.add_argument('-m', '--message',
|
|
help='Specifies the message to send, prevents interactive prompting.')
|
|
|
|
group = parser.add_argument_group('Stream parameters')
|
|
group.add_argument('-s', '--stream',
|
|
dest='stream',
|
|
action='store',
|
|
help='Allows the user to specify a stream for the message.')
|
|
group.add_argument('-S', '--subject',
|
|
dest='subject',
|
|
action='store',
|
|
help='Allows the user to specify a subject for the message.')
|
|
|
|
options = parser.parse_args()
|
|
|
|
if options.verbose:
|
|
logging.getLogger().setLevel(logging.INFO)
|
|
# Sanity check user data
|
|
if len(options.recipients) != 0 and (options.stream or options.subject):
|
|
parser.error('You cannot specify both a username and a stream/subject.')
|
|
if len(options.recipients) == 0 and (bool(options.stream) != bool(options.subject)):
|
|
parser.error('Stream messages must have a subject')
|
|
if len(options.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': options.recipients,
|
|
}
|
|
|
|
if not do_send_message(client, message_data):
|
|
return 1
|
|
return 0
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|