#!/usr/bin/env python3

import argparse

import zulip

usage = """mute-topic (mute | unmute) <stream> <topic> [options]

Example: mute-topic mute Verona dinner
Example: mute-topic unmute Denmark party
"""

parser = zulip.add_default_arguments(argparse.ArgumentParser(usage=usage))
parser.add_argument("op", choices=["mute", "unmute"])
parser.add_argument("stream")
parser.add_argument("topic")
options = parser.parse_args()

client = zulip.init_from_options(options)

OPERATIONS = {"mute": "add", "unmute": "remove"}

print(
    client.mute_topic(
        {"op": OPERATIONS[options.op], "stream": options.stream, "topic": options.topic}
    )
)
