zulip/zerver/views/alert_words.py

42 lines
1.4 KiB
Python

from typing import Annotated
from django.http import HttpRequest, HttpResponse
from pydantic import Json, StringConstraints
from zerver.actions.alert_words import do_add_alert_words, do_remove_alert_words
from zerver.lib.alert_words import user_alert_words
from zerver.lib.response import json_success
from zerver.lib.typed_endpoint import typed_endpoint
from zerver.models import UserProfile
def list_alert_words(request: HttpRequest, user_profile: UserProfile) -> HttpResponse:
return json_success(request, data={"alert_words": user_alert_words(user_profile)})
def clean_alert_words(alert_words: list[str]) -> list[str]:
alert_words = [w.strip() for w in alert_words]
return [w for w in alert_words if w != ""]
@typed_endpoint
def add_alert_words(
request: HttpRequest,
user_profile: UserProfile,
*,
alert_words: Json[list[Annotated[str, StringConstraints(max_length=100)]]],
) -> HttpResponse:
do_add_alert_words(user_profile, clean_alert_words(alert_words))
return json_success(request, data={"alert_words": user_alert_words(user_profile)})
@typed_endpoint
def remove_alert_words(
request: HttpRequest,
user_profile: UserProfile,
*,
alert_words: Json[list[str]],
) -> HttpResponse:
do_remove_alert_words(user_profile, alert_words)
return json_success(request, data={"alert_words": user_alert_words(user_profile)})