mcvsoftware/enet-client-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Reindented all source code

Browse Source
This commit is contained in:
Manuel Cortez
2022-01-14 11:47:24 -06:00
parent 17b9161813
commit 1aa7491624
11 changed files with 436 additions and 437 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
enetcomponents/__init__.py
View File

@@ -1 +1 @@
# -*- coding: utf-8 -*-
# -*- coding: utf-8 -*-

74
enetcomponents/client.py
View File

@@ -5,46 +5,46 @@ import enet
class client(object):
def __init__(self, host=b"localhost", port=33333, peer_count=1, channel_limit=4, incoming_bandwidth=0, outgoing_bandwidth=0):
self.host = enet.Host(None, peer_count, channel_limit, incoming_bandwidth, outgoing_bandwidth)
address = enet.Address(host, port)
self.peer = self.host.connect(address, channel_limit)
def __init__(self, host=b"localhost", port=33333, peer_count=1, channel_limit=4, incoming_bandwidth=0, outgoing_bandwidth=0):
self.host = enet.Host(None, peer_count, channel_limit, incoming_bandwidth, outgoing_bandwidth)
address = enet.Address(host, port)
self.peer = self.host.connect(address, channel_limit)
def run(self):
self.running = True
while self.running:
event = self.host.service(0)
if event.type == enet.EVENT_TYPE_CONNECT:
if hasattr(self, "connected"):
self.connected(event.peer)
elif event.type == enet.EVENT_TYPE_DISCONNECT:
if hasattr(self, "disconnected"):
self.disconnected(event.peer)
elif event.type == enet.EVENT_TYPE_RECEIVE:
data = event.packet.data
data_dict = json.loads(data)
self.network(event, data_dict)
time.sleep(0.001)
def run(self):
self.running = True
while self.running:
event = self.host.service(0)
if event.type == enet.EVENT_TYPE_CONNECT:
if hasattr(self, "connected"):
self.connected(event.peer)
elif event.type == enet.EVENT_TYPE_DISCONNECT:
if hasattr(self, "disconnected"):
self.disconnected(event.peer)
elif event.type == enet.EVENT_TYPE_RECEIVE:
data = event.packet.data
data_dict = json.loads(data)
self.network(event, data_dict)
time.sleep(0.001)
def send_data(self, channel, data, reliable=True):
data_str = json.dumps(data, ensure_ascii=False)
data_bytes = bytes(data_str, "utf-8")
if reliable:
flags = enet.PACKET_FLAG_RELIABLE
else:
flags = enet.PACKET_FLAG_UNSEQUENCED
packet = enet.Packet(data_bytes, flags)
self.peer.send(channel, packet)
def send_data(self, channel, data, reliable=True):
data_str = json.dumps(data, ensure_ascii=False)
data_bytes = bytes(data_str, "utf-8")
if reliable:
flags = enet.PACKET_FLAG_RELIABLE
else:
flags = enet.PACKET_FLAG_UNSEQUENCED
packet = enet.Packet(data_bytes, flags)
self.peer.send(channel, packet)
def network(self, event, data):
print(data)
def network(self, event, data):
print(data)
def disconnect(self):
self.peer.disconnect()
def disconnect(self):
self.peer.disconnect()
def close(self):
self.running = False
def close(self):
self.running = False
def __del__(self):
del self.host
del self.peer
def __del__(self):
del self.host
del self.peer

132
enetcomponents/server.py
View File

@@ -4,82 +4,82 @@ import json
class channel(object):
def __init__(self, server, peer):
self.peer = peer
self.server = server
def __init__(self, server, peer):
self.peer = peer
self.server = server
def send_data(self, channel, data, reliable=True):
data_str = json.dumps(data, ensure_ascii=False)
data_bytes = bytes(data_str, "utf-8")
if reliable:
flags = enet.PACKET_FLAG_RELIABLE
else:
flags = enet.PACKET_FLAG_UNSEQUENCED
packet = enet.Packet(data_bytes, flags)
self.peer.send(channel, packet)
def send_data(self, channel, data, reliable=True):
data_str = json.dumps(data, ensure_ascii=False)
data_bytes = bytes(data_str, "utf-8")
if reliable:
flags = enet.PACKET_FLAG_RELIABLE
else:
flags = enet.PACKET_FLAG_UNSEQUENCED
packet = enet.Packet(data_bytes, flags)
self.peer.send(channel, packet)
def network(self, event, data):
print(data)
def network(self, event, data):
print(data)
def disconnect(self):
self.peer.disconnect()
def disconnect(self):
self.peer.disconnect()
class server(object):
def __init__(self, host=None, port=33333, peer_count=256, channel_limit=4, incoming_bandwidth=0, outgoing_bandwidth=0):
address = enet.Address(host, port)
self.host = enet.Host(address, peer_count, channel_limit, incoming_bandwidth, outgoing_bandwidth)
self.peers = {}
def __init__(self, host=None, port=33333, peer_count=256, channel_limit=4, incoming_bandwidth=0, outgoing_bandwidth=0):
address = enet.Address(host, port)
self.host = enet.Host(address, peer_count, channel_limit, incoming_bandwidth, outgoing_bandwidth)
self.peers = {}
def run(self):
self.running = True
while self.running:
event = self.host.service(0)
if event.type == enet.EVENT_TYPE_CONNECT:
self.connected(event.peer)
elif event.type == enet.EVENT_TYPE_DISCONNECT:
self.disconnected(event.peer)
elif event.type == enet.EVENT_TYPE_RECEIVE:
channel_object = self.get_channel(event.peer)
if channel_object == None:
print("Error receiving packet for invalid channel: %r" % (event.data))
return
data = event.packet.data
data_dict = json.loads(data)
channel_object.network(event, data_dict)
time.sleep(0.0001)
def run(self):
self.running = True
while self.running:
event = self.host.service(0)
if event.type == enet.EVENT_TYPE_CONNECT:
self.connected(event.peer)
elif event.type == enet.EVENT_TYPE_DISCONNECT:
self.disconnected(event.peer)
elif event.type == enet.EVENT_TYPE_RECEIVE:
channel_object = self.get_channel(event.peer)
if channel_object == None:
print("Error receiving packet for invalid channel: %r" % (event.data))
return
data = event.packet.data
data_dict = json.loads(data)
channel_object.network(event, data_dict)
time.sleep(0.0001)
def get_channel(self, peer):
for channel in self.peers:
if peer.incomingPeerID == channel.peer.incomingPeerID:
return channel
def get_channel(self, peer):
for channel in self.peers:
if peer.incomingPeerID == channel.peer.incomingPeerID:
return channel
def connected(self, peer):
p = channel(self, peer)
self.peers[p] = True
print(len(self.peers))
p.send_data(0, dict(action="connected"), False)
print("%s: CONNECTED" % peer.address)
def connected(self, peer):
p = channel(self, peer)
self.peers[p] = True
print(len(self.peers))
p.send_data(0, dict(action="connected"), False)
print("%s: CONNECTED" % peer.address)
def disconnected(self, peer):
for channel in self.peers:
if peer.incomingPeerID == channel.peer.incomingPeerID:
del self.peers[channel]
print("%s: DISCONNECT" % peer.address)
break
def disconnected(self, peer):
for channel in self.peers:
if peer.incomingPeerID == channel.peer.incomingPeerID:
del self.peers[channel]
print("%s: DISCONNECT" % peer.address)
break
def send_to_all(self, channel, data, reliable=True):
data_str = json.dumps(data, ensure_ascii=False)
data_bytes = bytes(data_str, "utf-8")
if reliable:
flags = enet.PACKET_FLAG_RELIABLE
else:
flags = enet.PACKET_FLAG_UNSEQUENCED
packet = enet.Packet(data_bytes, flags)
self.host.broadcast(channel, packet)
def send_to_all(self, channel, data, reliable=True):
data_str = json.dumps(data, ensure_ascii=False)
data_bytes = bytes(data_str, "utf-8")
if reliable:
flags = enet.PACKET_FLAG_RELIABLE
else:
flags = enet.PACKET_FLAG_UNSEQUENCED
packet = enet.Packet(data_bytes, flags)
self.host.broadcast(channel, packet)
def close(self):
self.running = False
def close(self):
self.running = False
def __del__(self):
del self.host
def __del__(self):
del self.host