Files
zulip/zerver/webhooks/beeminder/view.py
Anders Kaseorg df001db1a9 black: Reformat with Black 23.
Black 23 enforces some slightly more specific rules about empty line
counts and redundant parenthesis removal, but the result is still
compatible with Black 22.

(This does not actually upgrade our Python environment to Black 23
yet.)

Signed-off-by: Anders Kaseorg <anders@zulip.com>
2023-02-02 10:40:13 -08:00

60 lines
1.8 KiB
Python

# Webhooks for external integrations.
import time
from django.http import HttpRequest, HttpResponse
from zerver.decorator import webhook_view
from zerver.lib.request import REQ, has_request_variables
from zerver.lib.response import json_success
from zerver.lib.validator import (
WildValue,
check_float,
check_int,
check_string,
check_union,
to_wild_value,
)
from zerver.lib.webhooks.common import check_send_webhook_message
from zerver.models import UserProfile
MESSAGE_TEMPLATE = (
"You are going to derail from goal **{goal_name}** in **{time:0.1f} hours**. "
"You need **{limsum}** to avoid derailing.\n"
"* Pledge: **{pledge}$** {expression}\n"
)
def get_time(payload: WildValue) -> float:
losedate = payload["goal"]["losedate"].tame(check_int)
time_remaining = (losedate - time.time()) / 3600
return time_remaining
@webhook_view("Beeminder")
@has_request_variables
def api_beeminder_webhook(
request: HttpRequest,
user_profile: UserProfile,
payload: WildValue = REQ(argument_type="body", converter=to_wild_value),
) -> HttpResponse:
goal_name = payload["goal"]["slug"].tame(check_string)
limsum = payload["goal"]["limsum"].tame(check_string)
pledge = payload["goal"]["pledge"].tame(check_union([check_int, check_float]))
time_remain = get_time(payload) # time in hours
# To show user's probable reaction by looking at pledge amount
if pledge > 0:
expression = ":worried:"
else:
expression = ":relieved:"
topic = "beekeeper"
body = MESSAGE_TEMPLATE.format(
goal_name=goal_name,
time=time_remain,
limsum=limsum,
pledge=pledge,
expression=expression,
)
check_send_webhook_message(request, user_profile, topic, body)
return json_success(request)