12 Commits
Author SHA1 Message Date
vladimir 6254de74e4 feat(ui): wip session_view.py 2026-08-11 04:08:55 +03:00
vladimir 1da622cd7c feat(ui): wip session_view.py 2026-08-10 03:29:23 +03:00
vladimir 63a51c7ae3 feat(ui): работа с новыми классами состояний приложения 2026-08-09 02:10:02 +03:00
vladimir c07bf63b8b fix(net): аннотация payload: dict[str, str] в классе NetwrokMessage 2026-08-09 02:09:01 +03:00
vladimir f80a9b4098 feat(core): отдельный класс для каждого состояния приложения 2026-08-09 02:05:53 +03:00
vladimir f38539f3c8 feat(ui): добавление работы с сетью через графический интерфейс 2026-08-07 03:07:04 +03:00
vladimir 715853c4b9 feat(net): добавление функционала для работы с сетью 2026-08-07 03:05:51 +03:00
vladimir e5bfbee528 docs: добавление описания формата сетевых сообщений 2026-08-07 03:04:55 +03:00
vladimir 5fcbf5ac06 feat(ui): заменены кнопки выбора ролей на переключаемые 2026-08-07 00:48:21 +03:00
vladimir 3add50cbbc feat(ui): добавление поддержки лобби в основное окно 2026-08-06 03:29:16 +03:00
vladimir e533acd9f1 feat(core): добавление класса для хранения внутреннего состояния 2026-08-06 03:28:07 +03:00
vladimir a25997992e feat(ui): добавление лобби и возможности выбора роли 2026-08-06 03:27:08 +03:00
10 changed files with 910 additions and 10 deletions
+12
View File
@@ -0,0 +1,12 @@
# Описание протокола
## Формат сообщений
```json
{
"version": "1.0",
"id": 0,
"event": "role_changed",
"payload": {
"role": "NONE"
}
}
```
+7
View File
@@ -0,0 +1,7 @@
from .state import AppState, LobbyState, SessionState
__all__ = [
"AppState",
"LobbyState",
"SessionState",
]
+172
View File
@@ -0,0 +1,172 @@
from typing import Callable, Optional
from wx.lib.pubsub import pub
from ..network import NetworkController, NetworkMesssage
OnNetworkEventCallback = Callable[[dict[str, str]], None]
class BaseState:
def __init__(self, app_state: "AppState"):
self.app_state = app_state
def handle_network_message(self, message: NetworkMesssage) -> None:
pass
class LobbyState(BaseState):
def __init__(self, app_state: "AppState", local_player_name: str):
super().__init__(app_state)
self.local_player_name: str = local_player_name
self.remote_player_name: Optional[str] = None
self.players: dict[str, str] = {self.local_player_name: "NONE"}
def handle_network_message(self, message: NetworkMesssage) -> None:
callbacks: dict[str, OnNetworkEventCallback] = {
"add_player": self.handle_add_player,
"get_player": self.handle_get_player,
"set_role": self.handle_set_player_role,
"disconnect": self.handle_disconnect,
"server_stop": self.handle_server_stop,
}
callback = callbacks.get(message.event, None)
if callback:
callback(message.payload)
pub.sendMessage("lobby.refresh", state=self)
def send_local_player(self) -> None:
player_name = self.local_player_name
player_role = self.players.get(player_name, "NONE")
self.app_state.network.send(
"add_player",
{
"player_name": player_name,
"player_role": player_role,
},
)
def get_remote_player(self) -> None:
self.app_state.network.send("get_player")
def set_player_role(self, role: str) -> bool:
current_role = self.players[self.local_player_name]
result = False
if current_role == role:
result = False
if role in ["NONE", "SETTER", "GUESSER"]:
self.players[self.local_player_name] = role
result = True
if result:
if self.app_state.network.conn:
self.app_state.network.send(
"set_role",
{
"player_name": self.local_player_name,
"player_role": role,
},
)
return result
def send_client_disconnect(self) -> None:
self.app_state.network.send(
"disconnect", {"player_name": self.local_player_name}
)
def send_server_stop(self) -> None:
self.app_state.network.send("server_stop")
def handle_add_player(self, payload: dict[str, str]) -> None:
player_name = payload["player_name"]
player_role = payload["player_role"]
self.remote_player_name = player_name
self.players[self.remote_player_name] = player_role
def handle_get_player(self, payload: dict[str, str]) -> None:
self.send_local_player()
def handle_set_player_role(self, payload: dict[str, str]) -> None:
player_name = payload["player_name"]
player_role = payload["player_role"]
self.players[player_name] = player_role
def handle_disconnect(self, payload: dict[str, str]) -> None:
player_name = payload["player_name"]
self.players.pop(player_name)
self.remote_player_name = None
def handle_server_stop(self, payload: dict[str, str]) -> None:
pub.sendMessage("mainframe.remote_server_stop", state=self.app_state)
@property
def local_player_role(self) -> str:
return self.players[self.local_player_name]
def is_ready(self) -> bool:
roles = list(self.players.values())
return ("SETTER" in roles) and ("GUESSER" in roles)
class SessionState(BaseState):
def __init__(self, app_state: "AppState"):
super().__init__(app_state)
class AppState:
def __init__(self):
self.network = NetworkController(
self.handle_network_message,
self.handle_network_notice,
)
self.lobby: Optional[LobbyState] = None
self.session: Optional[SessionState] = None
def handle_network_message(self, message: NetworkMesssage) -> None:
if self.lobby:
self.lobby.handle_network_message(message)
if self.session:
self.session.handle_network_message(message)
def handle_network_notice(self, notice: str) -> None:
pub.sendMessage("mainframe.statusbar", notice=notice)
def recreate_network_controller(self) -> None:
self.network.stop()
self.network = NetworkController(
self.handle_network_message,
self.handle_network_notice,
)
def connect(self, player_name: str, host: str, port: int = 8994) -> bool:
if self.network.connect(host, port):
self.lobby = LobbyState(self, player_name)
self.lobby.send_local_player()
self.lobby.get_remote_player()
return True
else:
self.recreate_network_controller()
return False
def serve(self, player_name: str, host: str, port: int = 8994) -> None:
self.network.serve(host, port)
self.lobby = LobbyState(self, player_name)
def disconnect(self) -> None:
if self.lobby:
if self.network.is_server:
self.lobby.send_server_stop()
else:
self.lobby.send_client_disconnect()
self.lobby = None
self.recreate_network_controller()
+7
View File
@@ -0,0 +1,7 @@
from .controller import NetworkController
from .protocol import NetworkMesssage
__all__ = [
"NetworkController",
"NetworkMesssage",
]
+150
View File
@@ -0,0 +1,150 @@
import socket
import struct
import threading
from typing import Callable, Optional
import wx
from .protocol import NetworkMesssage
OnMessageCallback = Callable[[NetworkMesssage], None]
OnNoticeCallback = Callable[[str], None]
def receive_exact(sock: socket.socket, n: int) -> Optional[bytes]:
data = bytearray()
while len(data) < n:
packet = sock.recv(n - len(data))
if not packet:
return None
data.extend(packet)
return bytes(data)
class NetworkController(threading.Thread):
def __init__(
self,
on_message_callback: OnMessageCallback,
on_notice_callback: Optional[OnNoticeCallback] = None,
):
super().__init__(daemon=True)
self.on_message_callback = on_message_callback
self.on_notice_callback = on_notice_callback
self.sock: Optional[socket.socket] = None
self.conn: Optional[socket.socket] = None
self.is_running: bool = False
self.is_server: bool = False
def serve(self, host: str = "0.0.0.0", port: int = 8994) -> None:
self.is_server = True
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.sock.bind((host, port))
self.sock.listen(1)
self.is_running = True
self.start()
def connect(self, host: str, port: int = 8994) -> bool:
self.is_server = False
try:
self.conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.conn.connect((host, port))
self.is_running = True
self._notify("Connected to host")
self.start()
return True
except Exception as e:
self._notify(f"Connection failed: {e}")
return False
def send(self, event: str, payload: dict = {}) -> None:
if not self.conn or not self.is_running:
return
m = NetworkMesssage(event, payload)
try:
self.conn.sendall(m.encode())
except Exception as e:
self._notify(f"Send error: {e}")
self._close_active_connection()
def run(self) -> None:
HEADER_SIZE = 4
while self.is_running:
if self.is_server and self.sock and not self.conn:
self._notify("Waiting for peer to connect...")
try:
self.conn, addr = self.sock.accept()
self._notify(f"Peer connected from {addr[0]}:{addr[1]}")
except Exception:
break
while self.is_running and self.conn:
try:
header_bytes = receive_exact(self.conn, HEADER_SIZE)
if not header_bytes:
self._notify("Peer disconnected")
break
body_size = struct.unpack(">I", header_bytes)[0]
body_bytes = receive_exact(self.conn, body_size)
if not body_bytes:
self._notify("Peer disconnected unexpectedly")
break
json_str = body_bytes.decode("utf-8")
m = NetworkMesssage.from_json(json_str)
wx.CallAfter(self.on_message_callback, m)
except Exception as e:
self._notify(f"Read error: {e}")
break
self._close_active_connection()
if not self.is_server:
break
self.stop()
def _close_active_connection(self) -> None:
if self.conn:
try:
self.conn.shutdown(socket.SHUT_RDWR)
self.conn.close()
except Exception:
pass
finally:
self.conn = None
def _notify(self, notice: str) -> None:
if self.on_notice_callback:
wx.CallAfter(self.on_notice_callback, notice)
def stop(self) -> None:
self.is_running = False
self._close_active_connection()
if self.sock:
try:
self.sock.close()
except Exception:
pass
finally:
self.sock = None
self._notify("Connection closed")
+46
View File
@@ -0,0 +1,46 @@
import itertools
import json
import struct
from dataclasses import dataclass, field
_id_generator = itertools.count(0)
@dataclass
class NetworkMesssage:
event: str
payload: dict[str, str] = field(default_factory=dict)
id: int = field(default_factory=lambda: next(_id_generator))
version: str = "1.0"
def to_json(self) -> str:
return json.dumps(
{
"version": self.version,
"id": self.id,
"event": self.event,
"payload": self.payload,
},
ensure_ascii=False,
)
@classmethod
def from_json(cls, json_str: str) -> "NetworkMesssage":
data = json.loads(json_str)
return cls(
event=data.get("event", "unknown"),
payload=data.get("payload", {}),
id=data.get("id", -1),
version=data.get("version", "unknown"),
)
def encode(self) -> bytes:
raw_bytes = self.to_json().encode("utf-8")
length_prefix = struct.pack(">I", len(raw_bytes))
return length_prefix + raw_bytes
if __name__ == "__main__":
for i in range(10):
message = NetworkMesssage("none")
print(message.encode())
+51 -10
View File
@@ -1,7 +1,12 @@
import wx
from typing import Optional
from mind_reader.ui.dialogs import ClientConnectDialog, ServerStartDialog
from mind_reader.ui.menu_bar import AppMenuBar
import wx
from wx.lib.pubsub import pub
from ..core import AppState
from ..ui.dialogs import ClientConnectDialog, ServerStartDialog
from ..ui.menu_bar import AppMenuBar
from ..ui.views import LobbyPanel
class MainFrame(wx.Frame):
@@ -13,13 +18,21 @@ class MainFrame(wx.Frame):
size=wx.Size(700, 500),
style=wx.DEFAULT_FRAME_STYLE,
)
self.state: AppState = AppState()
self.lobby_panel: Optional[LobbyPanel] = None
self.Center()
self.main_sizer = wx.BoxSizer(wx.VERTICAL)
self.SetSizer(self.main_sizer)
self.menu_bar = AppMenuBar(self)
self.SetMenuBar(self.menu_bar)
self.CreateStatusBar(1)
self.SetStatusText("Подключение отсутствует")
pub.subscribe(self.set_statusbar_text, "mainframe.statusbar")
pub.subscribe(self.remote_server_stop, "mainframe.remote_server_stop")
self._bind_menu_events()
@@ -32,22 +45,50 @@ class MainFrame(wx.Frame):
wx.EVT_MENU, self.on_disconnect, self.menu_bar.item_disconnect
)
def set_statusbar_text(self, notice: str):
self.SetStatusText(notice)
def remote_server_stop(self, state: AppState):
self.on_disconnect(None)
def _show_lobby(self):
self._destroy_lobby()
if self.state.lobby:
self.lobby_panel = LobbyPanel(self, self.state.lobby)
self.main_sizer.Add(self.lobby_panel, 1, wx.EXPAND)
self.Layout()
def _destroy_lobby(self):
if self.lobby_panel:
self.main_sizer.Detach(self.lobby_panel)
self.lobby_panel.Destroy()
self.lobby_panel = None
self.Layout()
def on_client_connect(self, event):
d = ClientConnectDialog(self)
if d.ShowModal() == wx.ID_OK:
name, ip = d.get_data()
self.menu_bar.set_connected_state(True)
self.SetStatusText(f"Подключено к {ip} | Ваше имя {name}")
player_name, ip = d.get_data()
if self.state.connect(player_name, ip):
self.menu_bar.set_connected_state(True)
self._show_lobby()
d.Destroy()
def on_server_start(self, event):
d = ServerStartDialog(self)
if d.ShowModal() == wx.ID_OK:
name, ip = d.get_data()
player_name, ip = d.get_data()
self.state.serve(player_name, ip)
self.menu_bar.set_connected_state(True)
self.SetStatusText(f"Сервер запущен на {ip} | Ваше имя {name}")
self._show_lobby()
d.Destroy()
def on_disconnect(self, event):
self._destroy_lobby()
self.state.disconnect()
self.menu_bar.set_connected_state(False)
self.SetStatusText("Подключение отсутствует")
+5
View File
@@ -0,0 +1,5 @@
from mind_reader.ui.views.lobby_view import LobbyPanel
__all__ = [
"LobbyPanel",
]
+128
View File
@@ -0,0 +1,128 @@
from typing import Callable, Optional
import wx
from wx.lib.pubsub import pub
from ...core import LobbyState
class RoleColumn(wx.Panel):
def __init__(
self,
parent,
role_key: str,
button_label: str,
on_role_selected: Optional[Callable[[str], None]] = None,
):
super().__init__(parent)
self.role_key = role_key
self.on_role_selected = on_role_selected
self.is_active = False
sizer = wx.BoxSizer(wx.VERTICAL)
self.button_select = wx.ToggleButton(self, label=button_label)
self.button_select.Bind(wx.EVT_TOGGLEBUTTON, self._on_button_click)
sizer.Add(self.button_select, 0, wx.EXPAND | wx.ALL, 5)
self.list_players = wx.ListBox(self)
sizer.Add(self.list_players, 1, wx.EXPAND | wx.ALL, 5)
self.SetSizer(sizer)
def _on_button_click(self, event):
if self.is_active:
self.button_select.SetValue(True)
return
if self.on_role_selected:
self.on_role_selected(self.role_key)
def set_players(self, players: list[str]):
self.list_players.Clear()
if players:
self.list_players.AppendItems(players)
def set_role_state(self, role: str):
self.is_active = self.role_key == role
self.button_select.SetValue(self.is_active)
class LobbyPanel(wx.Panel):
def __init__(self, parent, state: LobbyState):
super().__init__(parent)
self.state = state
self._init_ui()
self.Bind(wx.EVT_WINDOW_DESTROY, self._on_destroy)
def _init_ui(self):
main_sizer = wx.BoxSizer(wx.VERTICAL)
columns_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.column_none = RoleColumn(
self, "NONE", "👤 Без роли", self._on_role_click
)
self.column_setter = RoleColumn(
self, "SETTER", "🧠 Загадывает", self._on_role_click
)
self.column_guesser = RoleColumn(
self, "GUESSER", "🕵️ Угадывает", self._on_role_click
)
columns_sizer.Add(self.column_none, 1, wx.EXPAND | wx.ALL, 5)
columns_sizer.Add(self.column_setter, 1, wx.EXPAND | wx.ALL, 5)
columns_sizer.Add(self.column_guesser, 1, wx.EXPAND | wx.ALL, 5)
self.button_start = wx.Button(self, label="➔ Начать сессию")
self.button_start.Bind(wx.EVT_BUTTON, self._on_start_click)
main_sizer.Add(columns_sizer, 1, wx.EXPAND | wx.ALL, 5)
main_sizer.Add(
self.button_start, 0, wx.EXPAND | wx.LEFT | wx.RIGHT | wx.BOTTOM, 10
)
self.SetSizer(main_sizer)
pub.subscribe(self.refresh, "lobby.refresh")
self.refresh(self.state)
def _on_role_click(self, target_role: str):
success = self.state.set_player_role(target_role)
if not success:
wx.MessageBox(
f"Не удалось занять роль «{target_role}».",
"Внимание",
wx.OK | wx.ICON_WARNING,
)
return
self.refresh(self.state)
def _on_start_click(self, event):
wx.MessageBox("Сессия запущена!", "Старт", wx.OK | wx.ICON_INFORMATION)
def _on_destroy(self, event):
pub.unsubscribe(self.refresh, "lobby.refresh")
event.Skip()
def refresh(self, state: LobbyState):
players_by_role = {"NONE": [], "SETTER": [], "GUESSER": []}
for player, role in state.players.items():
if role in players_by_role:
players_by_role[role].append(player)
self.column_none.set_players(players_by_role["NONE"])
self.column_setter.set_players(players_by_role["SETTER"])
self.column_guesser.set_players(players_by_role["GUESSER"])
local_player_role = state.local_player_role
self.column_none.set_role_state(local_player_role)
self.column_setter.set_role_state(local_player_role)
self.column_guesser.set_role_state(local_player_role)
self.button_start.Enable(state.is_ready())
+332
View File
@@ -0,0 +1,332 @@
import sys
from dataclasses import dataclass
from enum import Enum, auto
from typing import Optional
import wx
from mind_reader.core import SessionState
def enable_high_dpi_support():
if sys.platform == "win32":
try:
import ctypes
ctypes.windll.shcore.SetProcessDpiAwareness(2)
except Exception:
try:
import ctypes
ctypes.windll.user32.SetProcessDPIAware()
except Exception:
pass
class CardState(Enum):
SETTER_NOT_SET = auto()
SETTER_HIDDEN = auto()
SETTER_REVEALED = auto()
GUESSER_DISABLED = auto()
GUESSER_WAITING = auto()
GUESSER_CORRECT = auto()
GUESSER_WRONG = auto()
class ArrowDirection(Enum):
LEFT = auto()
RIGHT = auto()
@dataclass
class SessionStats:
total: int
setted: int
correct: int
wrong: int
taken_time: float
@property
def accuracy(self) -> float:
accuracy: float
if self.total == 0:
accuracy = 0.0
accuracy = (self.correct / self.total) * 100
return round(accuracy, 1)
@dataclass(frozen=True)
class SetterCardColour:
NOT_SET = wx.Colour(82, 97, 107)
HIDDEN = wx.Colour(201, 214, 223)
REVEALED = wx.Colour(240, 245, 249)
@dataclass(frozen=True)
class GuesserCardColour:
DISABLED = wx.Colour(82, 97, 107)
WAITING = wx.Colour(201, 214, 223)
CORRECT = wx.Colour(76, 175, 80)
WRONG = wx.Colour(229, 115, 115)
@dataclass(frozen=True)
class ArrowSymbol:
LEFT: str = ""
RIGHT: str = ""
class Card(wx.Panel):
def __init__(self, parent, state: CardState):
super().__init__(parent, style=wx.BORDER_SIMPLE)
self._direction: Optional[ArrowDirection] = None
self._state = state
self._setup()
self._update()
def _setup(self):
self.SetMinSize(wx.Size(45, 75))
self.label = wx.StaticText(self, style=wx.ALIGN_CENTER)
font = self.label.GetFont()
font.SetPointSize(20)
self.label.SetFont(font)
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(self.label, 0, wx.EXPAND)
self.SetSizer(sizer)
def _update(self):
match self._state:
case CardState.SETTER_NOT_SET:
colour = SetterCardColour.NOT_SET
show_arrow = False
case CardState.SETTER_HIDDEN:
colour = SetterCardColour.HIDDEN
show_arrow = False
case CardState.SETTER_REVEALED:
colour = SetterCardColour.REVEALED
show_arrow = True
case CardState.GUESSER_DISABLED:
colour = GuesserCardColour.DISABLED
show_arrow = False
case CardState.GUESSER_WAITING:
colour = GuesserCardColour.WAITING
show_arrow = False
case CardState.GUESSER_CORRECT:
colour = GuesserCardColour.CORRECT
show_arrow = True
case CardState.GUESSER_WRONG:
colour = GuesserCardColour.WRONG
show_arrow = True
if show_arrow and self._direction:
match self._direction:
case ArrowDirection.LEFT:
symbol = ArrowSymbol.LEFT
case ArrowDirection.RIGHT:
symbol = ArrowSymbol.RIGHT
self.label.SetLabel(symbol)
else:
self.label.SetLabel("")
self.SetBackgroundColour(colour)
self.Layout()
self.Refresh()
def set(
self,
state: CardState,
direction: Optional[ArrowDirection] = None,
) -> None:
self._state = state
self._direction = direction
self._update()
class SessionPanel(wx.Panel):
def __init__(
self,
parent,
setter_name: str,
guesser_name: str,
sequence_length: int = 20,
):
super().__init__(parent)
self._setter_name = setter_name
self._guesser_name = guesser_name
self._sequence_length = sequence_length
self._setter_sequence: list[Card] = []
self._guesser_sequence: list[Card] = []
self._statusbar = self._get_statusbar()
self._setup()
self._fit_parent_frame()
def _setup(self) -> None:
main_sizer = wx.BoxSizer(wx.VERTICAL)
setter_title = wx.StaticText(
self, label=f"Загадывает: {self._setter_name}"
)
setter_sizer = wx.BoxSizer(wx.HORIZONTAL)
for _ in range(self._sequence_length):
card = Card(self, CardState.SETTER_NOT_SET)
setter_sizer.Add(card, 0, wx.ALL, 4)
self._setter_sequence.append(card)
guesser_title = wx.StaticText(
self, label=f"Отгадывает: {self._guesser_name}"
)
guesser_sizer = wx.BoxSizer(wx.HORIZONTAL)
for _ in range(self._sequence_length):
card = Card(self, CardState.GUESSER_DISABLED)
guesser_sizer.Add(card, 0, wx.ALL, 4)
self._guesser_sequence.append(card)
controls_sizer = self._create_controls_sizer()
secondary_sizer = wx.BoxSizer(wx.VERTICAL)
secondary_sizer.Add(setter_title, 0, wx.LEFT | wx.TOP, 9)
secondary_sizer.Add(setter_sizer, 0, wx.ALL, 5)
secondary_sizer.Add(guesser_title, 0, wx.LEFT | wx.TOP, 9)
secondary_sizer.Add(guesser_sizer, 0, wx.ALL, 5)
secondary_sizer.Add(
controls_sizer, 0, wx.ALIGN_CENTER | wx.TOP | wx.BOTTOM, 25
)
main_sizer.Add(secondary_sizer, 0, wx.LEFT | wx.ALL, 0)
self.SetSizer(main_sizer)
def _create_controls_sizer(self) -> wx.Sizer:
left = wx.Button(self, label=ArrowSymbol.LEFT, size=wx.Size(145, 55))
right = wx.Button(self, label=ArrowSymbol.RIGHT, size=wx.Size(145, 55))
font = left.GetFont()
font.SetPointSize(22)
left.SetFont(font)
right.SetFont(font)
controls_sizer = wx.BoxSizer(wx.HORIZONTAL)
controls_sizer.Add(left, 0, wx.RIGHT, 10)
controls_sizer.Add(right, 0, wx.LEFT, 10)
return controls_sizer
def _get_statusbar(self) -> Optional[wx.StatusBar]:
top_frame = self.GetTopLevelParent()
if isinstance(top_frame, wx.Frame):
statusbar = top_frame.GetStatusBar()
if not statusbar:
statusbar = top_frame.CreateStatusBar(1)
statusbar.SetFieldsCount(4)
return statusbar
return None
def _autofit_statusbar(self, texts: list[str]) -> None:
if not self._statusbar:
return
dc = wx.ClientDC(self._statusbar)
width_1 = dc.GetTextExtent(texts[0]).width
width_2 = dc.GetTextExtent(texts[1]).width
width_3 = dc.GetTextExtent(texts[2]).width
self._statusbar.SetStatusWidths([-1, width_1, width_2, width_3])
def _fit_parent_frame(self) -> None:
top_frame = self.GetTopLevelParent()
if isinstance(top_frame, wx.Frame):
self.Layout()
best_size = self.GetBestSize()
min_size = top_frame.ClientToWindowSize(best_size)
top_frame.SetMinSize(min_size)
top_frame.Fit()
def update_setter_card(
self,
index: int,
state: CardState,
direction: Optional[ArrowDirection] = None,
) -> None:
self._setter_sequence[index].set(state, direction)
def update_guesser_card(
self,
index: int,
state: CardState,
direction: Optional[ArrowDirection] = None,
) -> None:
self._guesser_sequence[index].set(state, direction)
def update_statusbar(
self,
statusbar_text: Optional[str],
stats: SessionStats,
):
if not self._statusbar:
return
if statusbar_text is not None:
self._statusbar.SetStatusText(statusbar_text, 0)
progress = f" Прогресс {stats.setted}/{stats.total} "
self._statusbar.SetStatusText(progress, 1)
accuracy = (
f" Точность {stats.accuracy}% ({stats.correct}/{stats.wrong}) "
)
self._statusbar.SetStatusText(accuracy, 2)
minutes = int(stats.taken_time) // 60
seconds = int(stats.taken_time) % 60
taken_time = f" Время {minutes:02d}:{seconds:02d} "
self._statusbar.SetStatusText(taken_time, 3)
self._autofit_statusbar([progress, accuracy, taken_time])
class TestFrame(wx.Frame):
def __init__(self):
super().__init__(None, title="TestFrame")
main_panel = SessionPanel(self, "Бывалый", "Балбес")
main_sizer = wx.BoxSizer(wx.VERTICAL)
main_sizer.Add(main_panel, 1, wx.EXPAND)
self.SetSizer(main_sizer)
main_panel.update_setter_card(
0, CardState.SETTER_REVEALED, ArrowDirection.LEFT
)
main_panel.update_guesser_card(
0, CardState.GUESSER_WRONG, ArrowDirection.RIGHT
)
s = SessionStats(20, 1, 0, 1, 5.1)
main_panel.update_statusbar(None, s)
if __name__ == "__main__":
enable_high_dpi_support()
app = wx.App()
frame = TestFrame()
frame.Show()
app.MainLoop()