python socket创建server和client

1、server端

支持多客户端连接。

#!/usr/bin/python3

import socket
from threading import Thread
import time
from multiprocessing import Process
import subprocess


host = '192.168.10.40'
port = 9999
ADDRESS = (host,port)

g_socket_server = None # 负责监听的socket
g_conn_pool = {} # 连接池
def init():
    """
    初始化服务端
    """
    global g_socket_server
    g_socket_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    g_socket_server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
    g_socket_server.bind(ADDRESS)
    g_socket_server.listen(5) # 最大等待数（有很多人误以为最大连接数）
    print("Server start, wait for client connecting...")

def accept_client():
    """
    接收新连接
    """
    while True:
        client, address = g_socket_server.accept() # 阻塞，等待客户端连接
        # 给每个客户端创建一个独立的线程进行管理
        thread = Thread(target=message_handle, args=(client,address))
        # 设置成守护线程        
        thread.setDaemon(True)
        thread.start()

def message_handle(client, address):
    """
    消息处理
    """
    address = str(address)
    print("[+] Connected with: " , address)
    client.sendall("Connect server successfully!".encode())
    g_conn_pool[address] = client

    while True:
        try:
            bytes = client.recv(1024)
            data = bytes.decode()
            if "" == data: # client.close()
                print("客户端主动退出连接")
                remove_client(address)
                break
                
            print('[Received]', data)
            result = run_command(data)
            print('[Result]',result)
            client.sendall(result.encode())
        except Exception as e: # close cmd terminal
            print("客户端意外退出连接")
            print(e)
            remove_client(address)
            break
            

def run_command(command):
    result = None
    result = subprocess.getoutput(command)
    return result

def remove_client(address):

    address = str(address)
    client = g_conn_pool[address]
    if None != client:
        client.close()
        g_conn_pool.pop(address)
        print("client offline: " + address)

if __name__ == '__main__':
    init()
    # 新开一个线程，用于接受新连接
    thread = Thread(target=accept_client)
    thread.setDaemon(True)
    thread.start()
    # 主线程逻辑
    while True:
        time.sleep(0.1)        

2、client端

客户端输入#exit后，主动退出连接

#!/usr/bin/python3
# 文件名：client.py

# 导入 socket、sys 模块
import socket
import sys




def new_client():
    # 创建 socket 对象
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 

    # 获取本地主机名
    host = socket.gethostname() 
    host = '192.168.10.40'

    # 设置端口号
    port = 9999
    try:
        client.connect((host, port)) # 尝试进行连接
        print(client.recv(1024).decode())
    except Exception:
        print('[!]Server not found or not open')
        sys.exit()
    while True:
        trigger = input('Input:')
        if trigger == '#exit': # 自定义结束字符串
            break
        client.sendall(trigger.encode())
        data = client.recv(1024) # 接收小于 1024 字节的数据
        data = data.decode()
        print('[Received]',data)

    client.close()

if __name__ == "__main__":
    new_client()