L3-Bataille-Navale/source/gui/scene/RoomHost.py

94 lines
2.5 KiB
Python

from typing import TYPE_CHECKING
import pyglet
import requests
from source import network
from source.gui.scene.abc import Scene
from source.gui import widget, texture
from source.utils.thread import in_pyglet_context, StoppableThread
if TYPE_CHECKING:
from source.gui.window import Window
from source.network.packet import PacketSettings
class RoomHost(Scene):
def __init__(self, window: "Window", settings: "PacketSettings", **kwargs):
super().__init__(window, **kwargs)
self.ip_address: str = "127.0.0.1"
self.port: int = 52321
self.batch_button_background = pyglet.graphics.Batch()
self.batch_label = pyglet.graphics.Batch()
self.back = self.add_widget(
widget.Button,
x=20, y=20, width=0.2, height=0.1,
label_text="Retour",
style=texture.Button.Style1,
background_batch=self.batch_button_background,
label_batch=self.batch_label
)
self.back.add_listener("on_click_release", self.button_back_callback)
self.label_ip = self.add_widget(
widget.Text,
x=0.5, y=0.55,
anchor_x="center", anchor_y="center",
font_size=20,
batch=self.batch_label
)
self.description = self.add_widget(
widget.Text,
x=0.5, y=0.45,
anchor_x="center", anchor_y="center",
text="En attente d'un second joueur...",
batch=self.batch_label
)
self.thread_network = network.Host(window=self.window, daemon=True, settings=settings)
self.thread_network.start()
self._refresh_ip_text()
self.thread_ip = StoppableThread(target=self._refresh_ip) # NOQA
self.thread_ip.start()
def _refresh_ip(self):
while True:
try:
response = requests.get('https://api.ipify.org')
response.raise_for_status()
break
except requests.HTTPError:
if self.thread_ip.stopped: return
self.ip_address: str = response.content.decode('utf8')
in_pyglet_context(self._refresh_ip_text)
def _refresh_ip_text(self):
self.label_ip.text = f"{self.ip_address}:{self.port}"
def button_back_callback(self, *_):
self.thread_network.stop()
self.thread_ip.stop()
from source.gui.scene import MainMenu
self.window.set_scene(MainMenu)
def on_draw(self):
self.batch_button_background.draw()
self.batch_label.draw()