46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from dataclasses import dataclass, field
|
||
|
|
|
||
|
|
from .file_consent import FileConsentStore
|
||
|
|
from .graph import MSTeamsGraphClient
|
||
|
|
from .polls import PollStore
|
||
|
|
from .sdk import BotFrameworkAdapter
|
||
|
|
from .sent_message_cache import SentMessageCache
|
||
|
|
from .sso import SSOTokenStore
|
||
|
|
from .user_agent import GraphRateLimiter
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class MSTeamsRuntime:
|
||
|
|
adapter: BotFrameworkAdapter | None = None
|
||
|
|
conversation_store: object | None = None
|
||
|
|
poll_store: PollStore | None = None
|
||
|
|
sent_message_cache: SentMessageCache = field(default_factory=SentMessageCache)
|
||
|
|
file_consent_store: FileConsentStore = field(default_factory=FileConsentStore)
|
||
|
|
sso_token_store: SSOTokenStore = field(default_factory=SSOTokenStore)
|
||
|
|
graph_client: MSTeamsGraphClient | None = None
|
||
|
|
graph_rate_limiter: GraphRateLimiter = field(default_factory=GraphRateLimiter)
|
||
|
|
is_running: bool = False
|
||
|
|
|
||
|
|
@property
|
||
|
|
def has_graph_client(self) -> bool:
|
||
|
|
return self.graph_client is not None
|
||
|
|
|
||
|
|
def get_graph_client(self) -> MSTeamsGraphClient | None:
|
||
|
|
if self.graph_client and self.graph_rate_limiter.can_proceed():
|
||
|
|
self.graph_rate_limiter.record()
|
||
|
|
return self.graph_client
|
||
|
|
if self.graph_client:
|
||
|
|
logger.warning("Graph API rate limit reached (%d remaining)", self.graph_rate_limiter.remaining)
|
||
|
|
return None
|
||
|
|
|
||
|
|
def shutdown(self) -> None:
|
||
|
|
self.is_running = False
|
||
|
|
if self.poll_store:
|
||
|
|
self.poll_store.save()
|
||
|
|
self.sent_message_cache.cleanup()
|