wxPython 创建协作白板绘图应用

在本文中，我们将介绍如何使用wxPython库创建一个协作白板绘图应用。该应用允许多个用户连接到同一个白板，并实时绘制和编辑图形。

准备工作

在开始之前，我们需要安装wxPython库。可以使用以下命令来安装：

pip install -U wxPython

创建主窗口

首先，我们需要创建一个主窗口，用于容纳白板绘图区域和工具栏等组件。我们可以使用wxPython的wx.Frame类来实现：

import wx

class MainFrame(wx.Frame):
    def __init__(self, parent=None, title="Whiteboard"):
        super().__init__(parent, title=title)

        # 创建白板绘图区域
        self.canvas = wx.Panel(self)

        # 创建工具栏
        self.toolbar = wx.ToolBar(self)
        self.add_toolbar_buttons()
        self.SetToolBar(self.toolbar)

        # 设置主窗口布局
        self.SetSizer(wx.BoxSizer(wx.VERTICAL))
        self.GetSizer().Add(self.canvas, 1, wx.EXPAND)

绘制图形

接下来，我们需要实现绘制图形的功能。我们可以使用wxPython的绘图上下文来实现不同类型的绘制操作。下面是一个简单的例子，演示如何在白板上绘制一个矩形：

import wx

class MainFrame(wx.Frame):
    ...
    def draw_rectangle(self, event):
        dc = wx.ClientDC(self.canvas)
        dc.SetBrush(wx.Brush("blue"))
        dc.DrawRectangle(50, 50, 100, 100)

这段代码在主窗口类中添加了一个名为draw_rectangle的方法，该方法在白板上绘制一个蓝色的矩形。

添加协作功能

要实现协作功能，我们需要使用一个网络通信库来处理多个用户之间的通信。这里我们选择使用Python的socket库。下面是一个简单的示例，演示如何创建一个简单的服务器和客户端，实现多用户之间的实时通信：

import socket

class Server:
    def __init__(self):
        self.host = 'localhost'
        self.port = 9999
        self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.s.bind((self.host, self.port))
        self.s.listen(5)

    def start(self):
        while True:
            conn, addr = self.s.accept()
            print(f'Connected with {addr[0]}:{addr[1]}')

            while True:
                data = conn.recv(1024).decode()
                if not data:
                    break

                # 处理客户端发送的数据

            conn.close()

class Client:
    def __init__(self):
        self.host = 'localhost'
        self.port = 9999
        self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.s.connect((self.host, self.port))

    def send_data(self, data):
        self.s.sendall(data.encode())

在这个简单的示例中，我们创建了一个Server类和一个Client类来封装服务器和客户端的功能。服务器监听端口，并在接收到客户端连接后，通过循环接收和处理数据来实现实时通信。客户端通过发送数据给服务器来进行通信。

实现协作白板

有了绘制图形和网络通信的基础，我们可以开始实现协作白板应用了。我们为每个用户分配一个唯一的标识符，当用户在白板上绘制图形时，将图形的绘制操作和用户标识符发送给服务器。服务器将接收到的数据广播给其他用户，其他用户接收到数据后，在本地绘制图形。

以下是一个简单的示例代码，演示了如何在协作白板中实现绘制直线的功能：

import wx
import socket
import threading

class MainFrame(wx.Frame):
    def __init__(self, parent=None, title="Whiteboard"):
        ...

        # 创建服务器和客户端
        self.server = Server()
        self.client = Client()

        # 启动服务器和客户端
        server_thread = threading.Thread(target=self.server.start)
        server_thread.daemon = True
        server_thread.start()

    def draw_line(self, event):
        pen_color = "black"
        pen_width = 2
        start_point = (50, 50)
        end_point = (100, 100)

        # 绘制本地图形
        dc = wx.ClientDC(self.canvas)
        dc.SetPen(wx.Pen(pen_color, pen_width))
        dc.DrawLine(start_point[0], start_point[1], end_point[0], end_point[1])

        # 发送图形数据给服务器
        data = f"line {pen_color} {pen_width} {start_point[0]} {start_point[1]} {end_point[0]} {end_point[1]}"
        self.client.send_data(data)

在这段代码中，我们在主窗口类中添加了一个名为draw_line的方法，该方法在绘制直线前先在本地绘制直线，然后将绘制的直线数据发送给服务器。