修复了一下不知名bug

recv/recv_net.py

@@ -28,11 +28,11 @@ from sender import sender
 #     pass
 
 class recver_net():
-    def __init__(self,port,client_host,client_port):
+    def __init__(self, port, client_host, client_port):
         self.client_host = client_host
         self.client_port = client_port
         self.letter = Letter()
-        self.port=port
+        self.port = port
         self.server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         self.server.bind(('0.0.0.0', self.port))
         self.server.listen(10)
@@ -51,18 +51,21 @@ class recver_net():
 
     def send_thread(self, conn):
         while True:
-            flag: int = int(input(""))
+            try:
-            if flag == 0:
+                flag: int = int(input())
-                data = input("[-] Enter data to send: ")
+                if flag == 0:
-                msg = {"flag": 0, "data": data}
+                    data = input("[-] Enter data to send: ")
-                msg = json.dumps(msg).encode("utf-8")
+                    msg = {"flag": 0, "data": data}
-                self.send_data(conn, msg)
+                    msg = json.dumps(msg).encode("utf-8")
-            elif flag == 1:
+                    self.send_data(conn, msg)
-                print("[+] Calling file transfer module...")
+                elif flag == 1:
-                # file_name = input("Enter file name to send: ")
+                    print("[+] Calling file transfer module...")
-                self.send_file(conn)
+                    # file_name = input("Enter file name to send: ")
-            else:
+                    self.send_file(conn)
-                print("[x] Error,plesae enter 0 to send data, or 1 to send file.")
+                else:
+                    print("[x] Error,plesae enter 0 to send data, or 1 to send file.")
+            except ValueError:
+                print("[x] Error, please enter a valid number.")
 
     def send_data(self, conn, data):
         conn.sendall(data)
@@ -105,6 +108,7 @@ class recver_net():
     def run(self):
         threading.Thread(target=self.main).start()
 
+
 def input_verify():
     while True:
         try:
@@ -128,6 +132,8 @@ def input_verify():
         else:
             print("Invalid address format. Please enter in the format 'host:port'.")
     return port, client_host, client_port
+
+
 # test
 if __name__ == '__main__':
     port, client_host, client_port = input_verify()

sender/sender_net.py

@@ -50,18 +50,22 @@ class sender_net():
 
     def send_thread(self, conn):
         while True:
-            flag: int = int(input())
+            try:
-            if flag == 0:
+                flag: int = int(input())
-                data = input("[-] Enter data to send: ")
+                if flag == 0:
-                msg = {"flag": 0, "data": data}
+                    data = input("[-] Enter data to send: ")
-                msg = json.dumps(msg).encode("utf-8")
+                    msg = {"flag": 0, "data": data}
-                self.send_data(conn, msg)
+                    msg = json.dumps(msg).encode("utf-8")
-            elif flag == 1:
+                    self.send_data(conn, msg)
-                print("[+] Calling file transfer module...")
+                elif flag == 1:
-                # file_name = input("Enter file name to send: ")
+                    print("[+] Calling file transfer module...")
-                self.send_file(conn)
+                    # file_name = input("Enter file name to send: ")
-            else:
+                    self.send_file(conn)
-                print("[x] Error,plesae enter 0 to send data, or 1 to send file.")
+                else:
+                    print("[x] Error,plesae enter 0 to send data, or 1 to send file.")
+            except ValueError:
+                print("[x] Error, please enter a valid number.")
+
 
     def send_data(self, conn, msg):
         conn.sendall(msg)
@@ -134,8 +138,6 @@ def input_verify():
 
 #test
 if __name__ == '__main__':
-
-
     port, client_host, client_port = input_verify()
     a = sender_net(port, client_host, client_port)
     a.run()