L3-Bataille-Navale/source/network/packet/PacketBombState.py

33 lines
787 B
Python

import socket
from dataclasses import dataclass, field
from source.core.enums import BombState
from source.network.packet.abc import Packet
from source.type import Point2D
@dataclass
class PacketBombState(Packet):
position: Point2D = field()
bomb_state: BombState = field()
packet_size: int = 3
def to_bytes(self):
x, y = self.position
return (
x.to_bytes(1, "big") +
y.to_bytes(1, "big") +
self.bomb_state.value.to_bytes()
)
@classmethod
def from_bytes(cls, data: bytes):
return cls(
position=(
int.from_bytes(data[0:1], "big"),
int.from_bytes(data[1:2], "big"),
),
bomb_state=BombState.from_bytes(data[2:3])
)