mirror of
https://github.com/stoatchat/python-client-sdk.git
synced 2026-07-25 00:15:23 -04:00
Zomatree
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, Optional
|
|
from collections import deque
|
|
|
|
from .user import User
|
|
from .channel import Channel, channel_factory
|
|
from .server import Server
|
|
from .message import Message
|
|
from .member import Member
|
|
|
|
if TYPE_CHECKING:
|
|
from .http import HttpClient
|
|
from .payloads import ApiInfo, User as UserPayload, Channel as ChannelPayload, Server as ServerPayload, Message as MessagePayload, Member as MemberPayload
|
|
|
|
class State:
|
|
def __init__(self, http: HttpClient, api_info: ApiInfo, max_messages: int):
|
|
self.http = http
|
|
self.api_info = api_info
|
|
self.max_messages = max_messages
|
|
|
|
self.users: dict[str, User] = {}
|
|
self.channels: dict[str, Channel] = {}
|
|
self.servers: dict[str, Server] = {}
|
|
self.messages: deque[Message] = deque()
|
|
|
|
def get_user(self, id: str) -> Optional[User]:
|
|
return self.users.get(id)
|
|
|
|
def get_member(self, server_id: str, member_id: str) -> Optional[Member]:
|
|
server = self.servers.get(server_id)
|
|
|
|
if not server:
|
|
return
|
|
|
|
return server.get_member(member_id)
|
|
|
|
def get_channel(self, id: str) -> Optional[Channel]:
|
|
return self.channels.get(id)
|
|
|
|
def get_server(self, id: str) -> Optional[Server]:
|
|
return self.servers.get(id)
|
|
|
|
def add_user(self, payload: UserPayload) -> User:
|
|
user = User(payload, self)
|
|
self.users[user.id] = user
|
|
return user
|
|
|
|
def add_member(self, server_id: str, payload: MemberPayload) -> Optional[Member]:
|
|
server = self.get_server(server_id)
|
|
|
|
if not server:
|
|
return
|
|
|
|
member = Member(payload, server, self)
|
|
|
|
server._members[member.id] = member
|
|
return member
|
|
|
|
def add_channel(self, payload: ChannelPayload) -> Channel:
|
|
cls = channel_factory(payload)
|
|
channel = cls(payload, self)
|
|
self.channels[channel.id] = channel
|
|
return channel
|
|
|
|
def add_server(self, payload: ServerPayload) -> Server:
|
|
server = Server(payload, self)
|
|
self.servers[server.id] = server
|
|
return server
|
|
|
|
def add_message(self, payload: MessagePayload) -> Message:
|
|
message = Message(payload, self)
|
|
if len(self.messages) >= self.max_messages:
|
|
self.messages.pop()
|
|
|
|
self.messages.appendleft(message)
|
|
return message
|