#!/usr/bin/env python3

import argparse

import zulip

usage = """update-message-flags (add | remove) <flag> <message_id1> \
[more_message_ids]

Example: update-message-flags add read 4 8 15
Example: update-message-flags remove starred 16 23 42
"""

parser = zulip.add_default_arguments(argparse.ArgumentParser(usage=usage))
parser.add_argument("op", choices=["add", "remove"])
parser.add_argument("flag")
parser.add_argument("messages", type=int, nargs="+")
options = parser.parse_args()

client = zulip.init_from_options(options)

print(
    client.update_message_flags(
        {"op": options.op, "flag": options.flag, "messages": options.messages}
    )
)
