最基础的python socket demo

基础部分

server

import socket


def socket_server():
    HOST = '0.0.0.0'
    PORT = 5001
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 定义socket类型，网络通信，TCP
    s.bind((HOST, PORT))
    s.listen(1)  # 开始TCP监听,监听1个请求
    print("listening ... ")
    while 1:
        conn, addr = s.accept()  # 接受TCP连接，并返回新的套接字与IP地址
        print('Connected by', addr)  # 输出客户端的IP地址
        conn.send(bytes("Connected success", encoding='utf-8'))
        while 1:
            data = str(conn.recv(1024), encoding='utf-8')  # 把接收的数据实例化
            # conn.send(bytes("has recv data: " + data, encoding='utf-8'))
            print(data)
    conn.close()  # 关闭连接


if __name__ == "__main__":
    socket_server()

client

import socket


def socket_client():
    HOST = '127.0.0.1'
    PORT = 5001
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 定义socket类型，网络通信，TCP
    s.connect((HOST, PORT))
    data = s.recv(1024)  # 把接收的数据实例化
    print(data)
    while 1:
        cmd = input("Please input cmd:")  # 与人交互，输入命令
        s.send(bytes(cmd, encoding='utf-8'))  # 把命令发送给对端
        # data = s.recv(1024)  # 把接收的数据定义为变量
    s.close()  # 关闭连接


if __name__ == "__main__":
    socket_client()

Python实现一个服务端，多个客户端接入

服务器端

服务器端每一次接收客户端连接，都要fork一个进程来完成，如果单纯使用socket模块来实现，是不容易完成多线程接入的。为了方便，可以直接使用socketserver 模块。

from socketserver import BaseRequestHandler, ThreadingTCPServer


class Handler(BaseRequestHandler):

    def handle(self) -> None:
        address, pid = self.client_address
        print(f'{address} connected!')
        while True:
            data = self.request.recv(1024)
            if len(data) <= 0:
                print("close!")
                break
            print(f'receive data: {data.decode()}')
            self.request.sendall('response'.encode())


if __name__ == '__main__':
    server = ThreadingTCPServer(('127.0.0.1', 8998), Handler)
    print("Listening")
    server.serve_forever()

客户端代码

import socket


class Client:
    def __init__(self):
        # ipv4  TCP
        self.client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    def connect(self, server_ip, server_port):
        self.client.connect((server_ip, server_port))

    def run(self):
        while True:
            print("please input:")
            message = input()
            self.client.sendall(message.encode())
            response = self.client.recv(1024)
            print(response.decode())


if __name__ == '__main__':
    client = Client()
    client.connect('127.0.0.1', 8998)
    client.run()