聊天室

李承键

[Python] 纯文本查看 复制代码

import socket
import threading

# 服务器配置
HOST = '127.0.0.1'
PORT = 12345

# 创建 TCP 套接字
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect((HOST, PORT))

# 接收服务器消息
def receive():
    while True:
        try:
            message = client.recv(1024).decode('utf-8')
            if message == 'AUTH':
                username = input("请输入用户名: ")
                password = input("请输入密码: ")
                client.send(username.encode('utf-8'))
                client.send(password.encode('utf-8'))
            elif message == 'AUTH_SUCCESS':
                print("认证成功！")
            elif message == 'AUTH_FAIL':
                print("认证失败，退出聊天室！")
                client.close()
                break
            elif message == 'NICK':
                nickname = input("请输入你的昵称: ")
                client.send(nickname.encode('utf-8'))
            else:
                print(message)
        except:
            print("发生错误，退出聊天室！")
            client.close()
            break

# 发送消息给服务器
def write():
    while True:
        message = f'{input("")}'
        client.send(message.encode('utf-8'))

# 创建接收和发送消息的线程
receive_thread = threading.Thread(target=receive)
receive_thread.start()

write_thread = threading.Thread(target=write)
write_thread.start()