typing: Add none-checks for stream.recipient_id.

Signed-off-by: Zixuan James Li <p359101898@gmail.com>
This commit is contained in:
Zixuan James Li
2022-06-14 22:59:36 -04:00
committed by Tim Abbott
parent 944e602788
commit 058dab5818
7 changed files with 9 additions and 0 deletions

View File

@@ -271,6 +271,7 @@ def most_recent_message(user_profile: UserProfile) -> Message:
def get_subscription(stream_name: str, user_profile: UserProfile) -> Subscription:
stream = get_stream(stream_name, user_profile.realm)
recipient_id = stream.recipient_id
assert recipient_id is not None
return Subscription.objects.get(
user_profile=user_profile, recipient_id=recipient_id, active=True
)