Python 计算机通信

一、服务端

import socket
import random
import threading

def creat_chatroom():
    s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    s.bind(('127.0.0.1',9999))
    s.listen(5)
    print('聊天室已经创建')
    return s

#给其他用户发送消息
def send_message(data):
    for conn in client:
        conn.send(data)

#生成随机昵称
def random_name():
    name=random.choice(['香蕉','苹果','梨','橘子','桃子','西瓜','草莓','菠萝'])
    for i in range(3):
        name+=random.choice('123456789')
    return name

def fun(conn,addr):
    name=random_name()
    users[addr]=conn
    client[conn]=name
    send_message(bytes(f'欢迎{name}加入聊天室',encoding='utf-8'))
    while True:
        try:
            data=conn.recv(1024)
            data=f'{client[conn]}说:{data.decode("utf-8")}'
            send_message(bytes(data,encoding='utf-8'))
        except:
            print(f'{users[addr]}已经断开连接')
            data=bytes(f'{users[addr]}已经断开连接',encoding='utf-8')
            del users[addr]
            del client[conn]
            send_message(data)
            break
def main():
    s=creat_chatroom()
    global users,client
    users={}
    client={}
    while True:
        conn,addr=s.accept()
        threading.Thread(target=fun,args=(conn,addr)).start()
    s.close()

if __name__ == '__main__':
    main()

二、客户端

import socket
import threading

def creat_thread(s):
    while True:
        data=input('请输入消息：')
        if data!='exit':
            s.send(bytes(data,encoding='utf-8'))
        else:
            s.close()
            break

def main():
    s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    # creat_thread(s)
    s.connect(('127.0.0.1',9999))
    threading.Thread(target=creat_thread,args=(s,)).start()
    while True:
        try:
            data=s.recv(1024)
            print(data.decode('utf-8')+'\n')
        except:
            print('已经断开连接\n')
            break
    s.close()

if __name__ == '__main__':
    main()