Add simple networking module and test

README.md

@@ -1,4 +1,4 @@
-# 20241144010013
-
-# Repository for random Projects that run through my mind
+# Sobre
+Repositorio que contem uma versão **MELHORADA** de batalha naval.
 
+Fork do [repositório de Kaio Henrique](https://gitea.mange.ifrn.edu.br/20241144010013/20241144010013)

src/main.py

@@ -1,15 +1,29 @@
 #setup
 import turtle as tl
-import random
+import random, sys
 
-posx = [-200, -100, 0, 100, 200]
-posy = [-200, -100, 0, 100, 200]
+args = sys.argv
+# name, -arguments
+def has_argument(args, arg):
+    if len(args) > 1:
+        if args[1].startswith("-"):
+            for chr in args[1]:
+                if chr == arg:
+                    return True
+    return False
+
+def num2char(num) -> chr:
+    return chr(num + 65)
+def char2num(char) -> int:
+    return ord(char) - 65
+
+posx = [0, 1, 2, 3, 4]
+posy = [0, 1, 2, 3, 4]
 postiro = []
-for i in range(5):
-    for j in range(5):
-        postiro.append(f"{(i-2) * 100} {(j-2) * 100}")
-        print((i-2) * 100, (j-2) * 100)
-#postiro = ['-200 -200', '-200 -100', '-200 0', '-200 100', '-200 200',  '-100 -200', '-100 -100', '-100 0', '-100 100', '-100 200',  '0 -200', '0 -100', '0 0', '0 100', '0 200',  '100 -200', '100 -100', '100 0', '100 100', '100 200',  '200 -200', '200 -100', '200 0', '200 100', '200 200']
+for i in range(len(posx)):
+    for j in range(len(posy)):
+        postiro.append(f"{i-2} {j-2}")
+
 
 tl.setup(650,650,None,None)
 tl.title("Naval Battle 1943 Loading...")
@@ -39,58 +53,67 @@ linhax(-300,300)
 #linhas X
 for i in range(5):
     linhax(-300, (i-2)*100)
-    tl.write(str((i-2) * 100), False, align="right")
+    tl.write(num2char(i), False, align="right")
 
 #linhas Y
 for i in range(5):
     linhay((i-2)*100, -300)
-    tl.write(str((i-2) * 100), False, align="right")
+    tl.write(str(i), False, align="right")
+
 #Tabuleiro pronto
 tl.title("Naval Battle 1943")
 tl.hideturtle()
 boat = tl.Turtle()
+boat.shapesize(3, 3)
 boat.up(); boat.goto(0,0); boat.left(90)
 
 #Enemies
-e1 = tl.Turtle()
-e1.up()
-e1.color("red")
-e1.goto(-100,-100)
+AMOUNT_OF_ENEMIES = 3
+enemies = []
 
+for i in range(AMOUNT_OF_ENEMIES):
+    enemy = tl.Turtle()
+    enemy.color("red")
+    enemy.up()
+    if not has_argument(args, "D"):
+        enemy.hideturtle()
+    else:
+        enemy.shapesize(3, 3)
+    pos = [random.choice(posy), random.choice(posy)]
+    enemy.goto((pos[0] - 2) * 100, (pos[1] - 2) * 100)
     
-e2 = tl.Turtle()
-e2.up()
-e2.color("red")
-e2.goto(random.choice(posx), random.choice(posy))
-if e1.pos() == e2.pos():
-    e2.goto(random.choice(posx), random.choice(posy))
-
-
-e3 = tl.Turtle()
-e3.up()
-e3.color("red")
-e3.goto(random.choice(posx), random.choice(posy))
-if e1.pos() == e3.pos():
-    e3.goto(random.choice(posx), random.choice(posy))
-if e2.pos() == e3.pos():
-    e3.goto(random.choice(posx), random.choice(posy))
+    enemies.append([enemy, pos])
 
 #partidas
 win = False
 while win == False:
     tirox = tl.numinput("linha X do disparo","Em qual quadrante X devemos disparar a artilharia")
-    tiroy = tl.numinput("linha Y do disparo","Em qual quadrante Y devemos disparar a artilharia")
-    if (tirox,tiroy) == e1.pos():
-        boat.goto(tirox,tiroy)
+    tiroy = tl.textinput("linha Y do disparo","Em qual quadrante Y devemos disparar a artilharia")
+    tiroy = char2num(tiroy.upper())
+    
+    if tirox == None or tiroy == None:
+        tl.exitonclick()
+    
+    boat.goto((tirox - 2) * 100, (tiroy - 2) * 100)
+    
+    acerto = False
+    for enemy in enemies:
+        if [tirox,tiroy] == enemy[1]:
+            hit_enemy: list[tl.Turtle, list[int, int]] = enemy
+            acerto = True
+            break
+    if acerto:
         print("Acerto!")
         boat.write("U-boat Afundado\n Capitão!",False,align="center")
-        break
+        hit_enemy[0].hideturtle()
+        enemies.remove(hit_enemy)
     else:
-        boat.goto(tirox,tiroy)
         print("errou!")
         boat.write("Erramos o alvo\n Capitão!",False,align="center")
-        continue
+    acerto = False
     
-#Execução
-tl.mainloop()
+    if len(enemies) <= 0: # tamanho negativo???
+        win = True
+
+print("Você ganhou!!")
 tl.exitonclick()

src/networking/net_module.py

@@ -0,0 +1,130 @@
+import socket, threading
+
+class Networking:
+    """
+    A class that contains common networking functions for both server and client.
+    """
+    @staticmethod
+    def recv_any_size(sock):
+        """
+        Receives data from the socket without knowing the size of the data.
+
+        The sender must send the size of the data before sending the data.
+        """
+        data_size = sock.recv(1024).decode()
+        sock.send("OK".encode())
+        data = sock.recv(int(data_size)).decode()
+        return data
+
+    @staticmethod
+    def send_any_size(sock, data):
+        """
+        Sends data to the socket without knowing the size of the data.
+
+        Sends the size of the data before sending the data.
+        """
+        sock.send(str(len(data)).encode())
+        response = sock.recv(1024).decode()
+        sock.send(data.encode())
+def recv_any_size(self, client):
+    """
+    Receives data from the client without knowing the size of the data.
+
+    Client must send the size of the data before sending the data.
+    """
+    data_size = client.recv(1024).decode()
+    data = client.recv(int(data_size)).decode()
+    return data
+
+class Server:
+    """
+    A basic server class that can be inherited to create a custom server.
+
+    Contains some basic networking functions.
+    """
+    def __init__(self, host = "127.0.0.1", port = 8080):
+        """
+        Hosts a basic server on the specified host and port
+        """
+        self.host = host
+        self.port = port
+        self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.server.bind((self.host, self.port))
+    
+    def start(self, max_connections = 5):
+        """
+        Starts the server and listens for *max_connections* connections, defaulted to 5
+        """
+        self.server.listen(max_connections)
+        while True:
+            client, addr = self.server.accept()
+            print(f"Connection from {addr}")
+
+            # Start a new thread for the client
+            threading.Thread(target=self.handle_client, args=(client,)).start()
+
+    def handle_client(self, client):
+        """
+        Checks if the client sent "exit" and closes the connection if it did, else send the data back.
+
+        Overwrite with your own implementation.
+        """
+        while True:
+            data = self.recv(client)
+            if data == "exit":
+                break
+            self.send(client, data)
+        pass
+
+    def send(self, client, data):
+        """
+        Sends data to the client
+        """
+        Networking.send_any_size(client, data)
+
+    def recv(self, client):
+        """
+        Receives data from the client
+        """
+        return Networking.recv_any_size(client)
+
+    def stop(self):
+        """
+        Stops the server
+        """
+        self.server.close()
+    
+
+class Client:
+    """
+    A basic client class that can be inherited to create a custom client.
+
+    Contains some basic networking functions.
+    """
+    def __init__(self, host = "127.0.0.1", port = 8080):
+        """
+        Connects to a server on the specified host and port
+        """
+        self.host = host
+        self.port = port
+        self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.sock.connect((self.host, self.port))
+
+    def send(self, data):
+        """
+        Sends data to the server
+        """
+        Networking.send_any_size(self.sock, data)
+
+    def recv(self):
+        """
+        Receives data from the server
+        """
+        return Networking.recv_any_size(self.sock)
+    
+    def close(self):
+        """
+        Closes the connection
+        """
+        Networking.send_any_size(self.sock, "exit")
+        self.sock.close()

src/test_networking.py

@@ -0,0 +1,24 @@
+from networking import net_module
+import socket, threading
+
+class Test_server(net_module.Server):
+    def handle_client(self, client: socket.socket):
+        print(self.recv(client))
+        self.send(client, "Hello World")
+
+        print(self.recv(client))
+        client.close()
+class Test_client(net_module.Client):
+    def main(self):
+
+        self.send("Hello World")
+        
+        print(self.recv())
+        
+        self.close()
+
+server = Test_server()
+threading.Thread(target=server.start, args=(1,)).start()
+client = Test_client()
+
+client.main()