Socket套接字实现TCP协议客户端

# 导包
import socket
import threading

# 发消息的函数
def send_msg(tcp_socket):
    while True:
        send_data = input("send: ")
        if send_data == "exit":
            break
        tcp_socket.send(send_data.encode())

# 接收消息的函数
def recv_msg(tcp_socket):
    while True:
        recv_data = tcp_socket.recv(1024)
        if recv_data:
            print(recv_data.decode('utf-8'))
        else:
            break

# 主函数
def main():
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    host = input("请输入连接的服务的IP：")
    port = int(input("请输入服务的端口："))
    tcp_socket.connect((host, port))
    th_send = threading.Thread(target=send_msg, args=(tcp_socket,))
    th_recv = threading.Thread(target=recv_msg, args=(tcp_socket,))
    th_send.start()
    th_recv.start()


if __name__ == '__main__':
    main()