wxpython笔记

 客户端：

# pip install wxpython

import wx
from socket import *
import threading
# 随机生成名字,地址,省份...
# 测试使用
from faker import Faker
class Client(wx.Frame):
    # 构造函数
    def __init__(self):
        self.name=Faker('zh_CN').name()    #名字
        self.isConnected=False  #连接成功与否
        self.client_socket=None

        # 页面布局
        # 构造wx的窗口的初始化,他的pos相对于windows的窗口
        wx.Frame.__init__(self,None,title=self.name+"聊天室客户端",size=(450,680),pos=(100,50))
        # 创建面板
        self.pl=wx.Panel(self)

        # 布局(不清楚直接实例化+self)
        # 1.创建按钮（wx.Button）
        # 加入聊天室
        self.conn_btn=wx.Button(self.pl,label="加入聊天室",pos=(10,10),size=(200,40))
        # 离开聊天室
        self.dis_conn_btn=wx.Button(self.pl,label="离开聊天室",pos=(220,10),size=(200,40))
        # 清空按钮
        self.clear_btn=wx.Button(self.pl,label="清空",pos=(10,580),size=(200,40))
        # 发送按钮
        self.send_btn=wx.Button(self.pl,label="发送",pos=(220,580),size=(200,40))
        # 2.创建聊天阅读文本框(wx.Textctrl) TE=控制所有文本的
        self.text=wx.TextCtrl(self.pl,size=(400,400),pos=(10,60),style=wx.TE_READONLY|wx.TE_MULTILINE)
        # 3.创建聊天输入文本框
        self.input_text=wx.TextCtrl(self.pl,size=(400,100),pos=(10,470),style=wx.TE_MULTILINE)
        # 4.button绑定事件
        self.Bind(wx.EVT_BUTTON,self.conn,self.conn_btn)
        self.Bind(wx.EVT_BUTTON,self.dis_conn,self.dis_conn_btn)
        self.Bind(wx.EVT_BUTTON,self.clear,self.clear_btn)
        self.Bind(wx.EVT_BUTTON,self.send,self.send_btn)

    # 点击按钮触发函数(event=鼠标活动)
    def conn(self,event):
        print("加入")
        if self.isConnected==False:
            self.isConnected=True
            self.client_socket=socket()
            self.client_socket.connect(('127.0.0.1',8999))
            # 连接成功发送客户名
            self.client_socket.send(self.name.encode('utf-8'))
            main_thread=threading.Thread(target=self.recv_data)
            main_thread.daemon=True
            main_thread.start()
    def recv_data(self):
        while self.isConnected:
            text=self.client_socket.recv(1024).decode('utf-8')
            self.text.AppendText(text+'\n')
    def dis_conn(self,event):
        print("dis_conn方法")
        self.client_socket.send("断开连接".encode('utf-8'))
        self.isConnected=False
    def clear(self,event):
        print("清空")
        # 调用清空函数
        self.input_text.Clear()
    def send(self,event):
        print("发送")
        if self.isConnected:
            text=self.input_text.GetValue()
            if text!='':
                self.client_socket.send(text.encode('utf-8'))
                self.input_text.Clear()


if __name__ == '__main__':
    # 创建应用程序对象
    app = wx.App()

    # 客户端窗口
    client=Client()
    # 显示客户端窗口（.Show）
    client.Show()
    # app一直循环显示
    app.MainLoop()

服务端

import wx
from socket import *
import threading
# 线程池管理
from concurrent.futures import ThreadPoolExecutor


class Server(wx.Frame):
    def __init__(self):
        # 实例属性
        self.isOn = False  # 服务器的启动状态

        # 创建socket对象
        self.server_socket = socket()
        self.server_socket.bind(("0.0.0.0", 8999))
        self.server_socket.listen(5)

        # 保存所有的客户端
        self.client_thread_dict = {}
        # 创建线程池
        self.pool = ThreadPoolExecutor(max_workers=10)

        # 界面布局
        # 调用父类的init方法
        wx.Frame.__init__(self, None, title="多人聊天室", pos=(0, 50), size=(450, 600))
        self.panel = wx.Panel(self)

        # 创建按钮
        # 1.启动服务
        self.start_server_btn = wx.Button(self.panel, pos=(10, 10), size=(200, 40), label="启动服务器")
        # 保存聊天呢记录
        self.save_text_btn = wx.Button(self.panel, pos=(220, 10), size=(200, 40), label="保存聊天记录")
        # 2.创建聊天阅读文本框(wx.Textctrl) TE=控制所有文本的
        self.text = wx.TextCtrl(self.panel, size=(400, 400), pos=(10, 60), style=wx.TE_READONLY | wx.TE_MULTILINE)
        # 3.button绑定事件
        self.Bind(wx.EVT_BUTTON, self.start_server, self.start_server_btn)
        self.Bind(wx.EVT_BUTTON, self.save_text, self.save_text_btn)

    def start_server(self, event):
        print("start_server函数")
        if self.isOn == False:
            self.isOn = True
            # 注意：如果不用多线程，会导致阻塞
            # self.server_socket.accept()
            main_thread = threading.Thread(target=self.main_thread_fun)
            # 设置为守护进程
            main_thread.daemon = True
            main_thread.start()

    def main_thread_fun(self):
        while self.isOn:
            client_socket, client_addr = self.server_socket.accept()
            print(client_addr)
            client_name = client_socket.recv(1024).decode('utf-8')
            print(client_name)
            client_thread = ClientThread(client_socket, client_name, self)
            # 保存到客户端字典
            self.client_thread_dict[client_name] = client_thread
            # 线程池的submit=运行线程池中的一个线程的一个成员方法，客户线程（线程子类）的run函数
            self.pool.submit(client_thread.run)
            self.send("【服务器通知】欢迎%s进入聊天室" % client_name)

    # 群发消息
    def send(self, text):
        self.text.AppendText(text + '\n')
        for client in self.client_thread_dict.values():
            if client.isOn:
                client.client_socket.send(text.encode('utf-8'))

    def save_text(self, event):
        print("save_text")
        record = self.text.GetValue()
        with open('record.log', "a+", encoding="GBK") as f:
            f.write(record)


class ClientThread(threading.Thread):
    def __init__(self, socket, name, server):
        threading.Thread.__init__(self)
        self.client_socket = socket
        self.client_name = name
        self.server = server
        self.isOn = True

    def run(self):
        while self.isOn:
            text = self.client_socket.recv(1024).decode('utf-8')
            if text == "断开连接":
                self.isOn = False
                self.server.send("【服务器消息】%s离开群聊" % self.client_name)
            else:
                self.server.send("【%s】%s" % (self.client_name, text))
        self.client_socket.close()


if __name__ == '__main__':
    # 创建应用程序对象
    app = wx.App()
    server = Server()
    server = server.Show()
    # app循环显示
    app.MainLoop()