#!/usr/bin/env python3

import argparse

usage = """subscribe [options] --streams=<streams>

Ensures the user is subscribed to the listed streams.

Examples: subscribe --streams=foo
          subscribe --streams='foo bar'

Specify your Zulip API credentials and server in a ~/.zuliprc file or using the options.
"""

import zulip

parser = zulip.add_default_arguments(argparse.ArgumentParser(usage=usage))
parser.add_argument("--streams", action="store", required=True)
options = parser.parse_args()

client = zulip.init_from_options(options)

print(client.add_subscriptions([{"name": stream_name} for stream_name in options.streams.split()]))
