python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(current_state): chat or user may be None

Browse source
This commit is contained in:
mpa 2020-08-16 16:54:44 +04:00
parent 18f95d6c25
commit b0792a5905
No known key found for this signature in database
GPG key ID: BCCFBFCCC9B754A8
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
aiogram/dispatcher/dispatcher.py
View file

@ -4,7 +4,7 @@ import asyncio
import contextvars
import warnings
from asyncio import CancelledError, Future, Lock
from typing import Any, AsyncGenerator, Dict, Generic, Optional, Union
from typing import Any, AsyncGenerator, Dict, Generic, Optional, Union, cast
from .. import loggers
from ..api.client.bot import Bot
@ -42,10 +42,13 @@ class Dispatcher(Router, Generic[StorageDataT]):
if self.storage is None:
self.storage: DummyStorage = DummyStorage() # type: ignore
chat = cast(Optional[Chat], Chat.get_current())
user = cast(Optional[User], User.get_current())
return CurrentUserContext(
storage=self.storage,
chat_id=Chat.get_current().id, # type: ignore
user_id=User.get_current().id, # type: ignore
chat_id=chat.id if chat else None,
user_id=user.id if user else None,
)
@property