Compare commits

..

No commits in common. "ea433a93c8b5feab266c62f86a599ab6de44f972" and "80cd8cc4c451dc211e3e228c0f7bec5c6181b089" have entirely different histories.

View file

@ -44,12 +44,6 @@ def warn_if_tty(you_can_type="the message objects", write_message_to=sys.stderr)
print(f"Reading from tty. (You can type {you_can_type} below. Or maybe you meant to redirect something to stdin.)", file=write_message_to)
def get_user_id(mm_api: mattermost.MMApi):
if mm_api._my_user_id is None:
mm_api._my_user_id = mm_api.get_user()["id"]
return mm_api._my_user_id
def get_posts_for_channel(self, channel_id: str, progress=lambda x: None, after=None, **kwargs) -> Iterable[Dict]:
"""
@raises ApiException: Passed on from lower layers.
@ -73,15 +67,13 @@ def get_posts_for_channel(self, channel_id: str, progress=lambda x: None, after=
progress(total)
if len(order) < per_page:
break
after = order[-1]
page += 1
sleep(0.1)
else:
# All posts in channel: API gives pages with NEWEST messages first, so reverse the order in
# the end (and don't reverse the order of each page separately)
posts = []
# To avoid race conditions when new messages are posted while the channel is being fetched
post_ids = set()
while True:
data_page = self._get(f"/v4/channels/{channel_id}/posts", params={"page": page, "per_page": per_page, **kwargs})
order = data_page["order"]
@ -89,9 +81,7 @@ def get_posts_for_channel(self, channel_id: str, progress=lambda x: None, after=
posts.extend(
data_page["posts"][post_id]
for post_id in order
if post_id not in post_ids
)
post_ids |= set(order)
progress(len(posts))
if len(order) < per_page:
break
@ -424,24 +414,16 @@ def edit(mm_api: mattermost.MMApi, cmdline_args):
mm_api.patch_post(cmdline_args.msgid, message=new_text)
def addreaction(mm_api: mattermost.MMApi, cmdline_args):
# "me" does not work here, explicit user ID required. (sigh)
mm_api.create_reaction(get_user_id(mm_api), cmdline_args.msgid, cmdline_args.emoji_name)
def removereaction(mm_api: mattermost.MMApi, cmdline_args):
# pylint: disable=protected-access # library recommends this in docs
mm_api._delete(f"/v4/users/me/posts/{cmdline_args.msgid}/reactions/{cmdline_args.emoji_name}")
def status(mm_api: mattermost.MMApi, cmdline_args):
if not cmdline_args.status:
raise ValueError("No status selected")
mm_api._put(f"/v4/users/me/status", data={
# This API endpoint requires the user ID to be passed explicitly in the request body,
# duplicating the info in the URL. But "me" does not work here here. (sigh)
"user_id": get_user_id(mm_api),
# duplicating the info in the URL. But "me" does not suffice here.
my_user_id = mm_api.get_user()["id"]
mm_api._put(f"/v4/users/me/status", data={
"user_id": my_user_id,
"status": cmdline_args.status
})
@ -519,8 +501,6 @@ ACTIONS = {
"send": {"function": send},
"rm": {"function": rm},
"edit": {"function": edit},
"addreaction": {"function": addreaction},
"removereaction": {"function": removereaction},
"status": {"function": status},
"customstatus": {"function": customstatus},
"lastread": {"function": lastread},
@ -622,16 +602,6 @@ The input format accepted on stdin is one JSON object per line. The possible fie
parser_edit.add_argument(
"--message", help="message; if not provided, message will be expected on stdin")
parser_addreaction = subparsers.add_parser(
"addreaction", help="add emoji reaction to a message")
parser_addreaction.add_argument("msgid", help="ID of message")
parser_addreaction.add_argument("emoji_name", help="name of the emoji to add (without colons)")
parser_removereaction = subparsers.add_parser(
"removereaction", help="remove emoji reaction from a message")
parser_removereaction.add_argument("msgid", help="ID of message")
parser_removereaction.add_argument("emoji_name", help="name of the emoji to remove (without colons)")
parser_status = subparsers.add_parser("status", help="update user status")
parser_status.add_argument("--online", dest="status", action="store_const", const="online", help="Set status to online")
parser_status.add_argument("--away", dest="status", action="store_const", const="away", help="Set status to away")