practice_code/python/test/Client/Page/Chat_main.py

import csv
import os
import wx
import time
import multiprocessing

from Client.Transmission.Process_Client import ProcessClient


class ChatFrame(wx.Frame, ProcessClient):
    def __init__(self, Id):
        wx.Frame.__init__(self, None, size=(800, 600), title="账号:  " + str(Id))
        ProcessClient.__init__(self)
        self.userid = Id

        current_file_path = __file__
        current_file_name = os.path.basename(current_file_path).split('.')[0]
        self.Process_client_send("Server", "Name", current_file_name)

        ChatMain_Panel = wx.Panel(self, style=wx.BORDER_SUNKEN)
        ChatMain_box = wx.BoxSizer(wx.HORIZONTAL)
        ChatMain_Panel.SetSizer(ChatMain_box)

        Function_Panel = wx.Panel(ChatMain_Panel, style=wx.BORDER_RAISED)
        Function_box = wx.BoxSizer(wx.VERTICAL)
        Function_Panel.SetSizer(Function_box)
        Function_Panel.SetBackgroundColour(wx.Colour(240, 240, 240))

        image_chat = r'./Client/image/chat.png'
        Chat_bitmap = wx.Bitmap(image_chat, wx.BITMAP_TYPE_PNG)
        Chat_bitmap_static = wx.StaticBitmap(Function_Panel, bitmap=Chat_bitmap, style=wx.BORDER_RAISED)
        Chat_bitmap_static.SetToolTip("聊天")
        Function_box.Add(Chat_bitmap_static, 1, wx.EXPAND, 0)
        Chat_bitmap_static.Bind(wx.EVT_LEFT_DOWN, self.click_chat_button)

        Function_box.AddSpacer(20)  # 调整这个数字以改变间距大小

        image_connect = r'./Client/image/connect.png'
        connect_bitmap = wx.Bitmap(image_connect, wx.BITMAP_TYPE_PNG)
        connect_bitmap_static = wx.StaticBitmap(Function_Panel, bitmap=connect_bitmap, style=wx.BORDER_RAISED)
        connect_bitmap_static.SetToolTip("通讯录")
        Function_box.Add(connect_bitmap_static, 1, wx.EXPAND, 0)
        connect_bitmap_static.Bind(wx.EVT_LEFT_DOWN, self.click_connect_button)

        Function_box.AddSpacer(20)  # 调整这个数字以改变间距大小

        image_find = r'./Client/image/find.png'
        find_bitmap = wx.Bitmap(image_find, wx.BITMAP_TYPE_PNG)
        find_bitmap_static = wx.StaticBitmap(Function_Panel, bitmap=find_bitmap, style=wx.BORDER_RAISED)
        find_bitmap_static.SetToolTip("查找")
        Function_box.Add(find_bitmap_static, 1, wx.EXPAND, 0)
        find_bitmap_static.Bind(wx.EVT_LEFT_DOWN, self.click_find_button)

        Function_box.AddSpacer(280)  # 调整这个数字以改变间距大小

        image_site = r'./Client/image/site.png'
        site_bitmap = wx.Bitmap(image_site, wx.BITMAP_TYPE_PNG)
        site_bitmap_static = wx.StaticBitmap(Function_Panel, bitmap=site_bitmap, style=wx.BORDER_RAISED)
        site_bitmap_static.SetToolTip("设置")
        Function_box.Add(site_bitmap_static, 1, wx.EXPAND, 0)
        site_bitmap_static.Bind(wx.EVT_LEFT_DOWN, self.click_site_button)

        Function_box.Fit(Function_Panel)  # 调整面板大小以适应图像

        ChatMain_box.Add(Function_Panel, 0, wx.EXPAND | wx.ALL, 5)

        operate_box = wx.BoxSizer(wx.VERTICAL)

        self.chat_page = self.ChatPage(ChatMain_Panel)  # 将 ChatMain_Panel 作为 ChatPage 的 parent
        operate_box.Add(self.chat_page, 1, wx.EXPAND, 5)
        self.chat_page.Hide()

        self.Process_client_send("File_operate", "read_data", None)

        self.connect_page = self.ConnectPage(ChatMain_Panel)
        operate_box.Add(self.connect_page, 1, wx.EXPAND, 5)
        self.connect_page.Hide()

        self.site_page = self.SitePage(ChatMain_Panel)
        operate_box.Add(self.site_page, 1, wx.EXPAND, 5)
        self.site_page.Hide()

        self.find_page = self.FindPage(ChatMain_Panel)
        operate_box.Add(self.find_page, 1, wx.EXPAND, 5)
        self.find_page.Hide()

        ChatMain_box.Add(operate_box, 1, wx.EXPAND, 5)

        self.chat_page.Show()

        ChatMain_Panel.SetSizer(ChatMain_box)

    def click_chat_button(self, event):
        self.chat_page.Hide()
        self.connect_page.Hide()
        self.site_page.Hide()
        self.find_page.Hide()
        self.chat_page.Show()
        self.Layout()

    def click_connect_button(self, event):
        self.chat_page.Hide()
        self.connect_page.Hide()
        self.site_page.Hide()
        self.find_page.Hide()
        self.connect_page.Show()
        self.Layout()

    def click_site_button(self, event):
        self.chat_page.Hide()
        self.connect_page.Hide()
        self.site_page.Hide()
        self.find_page.Hide()
        self.site_page.Show()
        self.Layout()

    def send_Chat_history(self, Contact, send, receive, Type, data, UpDataTime):
        chat_panel = self.chat_page.chat_page_map[Contact]
        match Type:
            case 'text':
                if Contact == send:
                    contents = ['My:', data, UpDataTime]
                    for content in contents:
                        message = wx.StaticText(chat_panel.chat_receive_panel, label=content, style=wx.ALIGN_RIGHT)
                        chat_panel.chat_receive_box.Add(message, flag=wx.EXPAND | wx.RIGHT, border=10)


    def click_find_button(self, event):
        self.chat_page.Hide()
        self.connect_page.Hide()
        self.site_page.Hide()
        self.find_page.Hide()
        self.find_page.Show()
        self.Layout()

    def Process_client_pick(self, data):
        if data['target'] in ['ALL', 'Chat_main']:
            match data['function']:
                case 'ChatPage_add_Contact_person':
                    data = data['content']
                    Contact = eval(data['Contact'])
                    Remark = data['Remark']
                    Time = data['UpDataTime']
                    wx.CallAfter(self.chat_page.ChatPage_add_Contact_person, Contact, Remark)
                case 'ChatPage_add_Contact_tab':
                    Contact = None
                    data = data['content']
                    send = eval(data['send'])
                    receive = eval(data['receive'])
                    Type = data['Type']
                    content = data['content']
                    UpDataTime = data['UpDataTime']
                    if send == self.userid:
                        Contact = receive
                    elif receive == self.userid:
                        Contact = send
                    self.send_Chat_history(Contact, send, receive, Type, content, UpDataTime)

    class ChatPage(wx.Panel):
        def __init__(self, parent):
            wx.Panel.__init__(self, parent, style=wx.BORDER_SUNKEN)

            ChatPage_main_box = wx.BoxSizer(wx.HORIZONTAL)
            self.chat_page_map = {}
            self.chat_page_ids = []
            self.current_chat_page = None  # 新增的属性用于跟踪当前显示的聊天面板

            self.ChatPage_Listbook = wx.Listbook(self, style=wx.LB_LEFT)
            self.ChatPage_Listbook.Bind(wx.EVT_LISTBOOK_PAGE_CHANGED, self.on_page_changed)

            ChatPage_main_box.Add(self.ChatPage_Listbook, 1, wx.EXPAND, 0)
            self.SetSizer(ChatPage_main_box)

        def ChatPage_add_Contact_person(self, Id, Remark):
            contact_page = wx.Panel(self.ChatPage_Listbook)
            self.ChatPage_Listbook.AddPage(contact_page, Remark)
            chat_page = self.ChatPage_add_Contact_tab()
            self.chat_page_map[Id] = chat_page
            self.chat_page_ids.append(Id)
            # 隐藏当前的聊天面板
            if self.current_chat_page:
                self.current_chat_page.Hide()
            # 将新的聊天面板添加到ChatPage_main_box中
            self.GetSizer().Add(chat_page, 5, wx.EXPAND)
            # 显示新的聊天面板
            chat_page.Show()
            # 更新当前显示的聊天面板
            self.current_chat_page = chat_page
            # 重新布局
            self.Layout()

        def ChatPage_add_Contact_tab(self):
            chat_panel = wx.Panel(self)
            chat_box = wx.BoxSizer(wx.VERTICAL)

            chat_panel.chat_receive_panel = wx.ScrolledWindow(chat_panel, style=wx.BORDER_SUNKEN)
            chat_panel.chat_receive_box = wx.BoxSizer(wx.HORIZONTAL)

            chat_panel.chat_receive_panel.SetSizer(chat_panel.chat_receive_box)
            chat_box.Add(chat_panel.chat_receive_panel, 3, wx.EXPAND | wx.ALL, 0)

            toolbar = wx.ToolBar(chat_panel)
            toolbar.SetToolBitmapSize((16, 16))  # Set the size of the toolbar icons
            toolbar.AddTool(wx.ID_ANY, "Tool", wx.Bitmap(wx.Bitmap(r'./Client/image/picture.png')))
            toolbar.AddTool(wx.ID_ANY, "Tool", wx.Bitmap(wx.Bitmap(r'./Client/image/file.png')))
            toolbar.AddTool(wx.ID_ANY, "Tool", wx.Bitmap(wx.Bitmap(r'./Client/image/speech.png')))
            toolbar.AddTool(wx.ID_ANY, "Tool", wx.Bitmap(wx.Bitmap(r'./Client/image/video.png')))
            toolbar.Realize()
            toolbar.SetSize((-1, 30))  # Set the size of the toolbar
            chat_box.Add(toolbar, 0, wx.EXPAND)

            chat_send_box = wx.BoxSizer(wx.HORIZONTAL)
            chat_panel.chat_send_text = wx.TextCtrl(chat_panel, style=wx.TE_MULTILINE)
            chat_send_box.Add(chat_panel.chat_send_text, 1, wx.EXPAND, 0)
            chat_box.Add(chat_send_box, 1, wx.EXPAND, 0)

            send_button = wx.Button(chat_panel, label='发送')  # Create a send button
            chat_send_box.Add(send_button, 0, wx.EXPAND | wx.LEFT, 5)  # Add the button to the send box

            chat_panel.SetSizer(chat_box)

            return chat_panel

        def on_page_changed(self, event):
            # 获取当前选定的页面的索引
            current_page_index = self.ChatPage_Listbook.GetSelection()
            # 通过索引从chat_page_ids列表中获取对应的Id
            current_page_id = self.chat_page_ids[current_page_index]
            # 从chat_page_map中找到对应的聊天面板
            new_chat_page = self.chat_page_map[current_page_id]
            # 隐藏旧的聊天面板
            if self.current_chat_page:
                self.current_chat_page.Hide()
            # 显示新的聊天面板
            new_chat_page.Show()
            # 更新当前显示的聊天面板
            self.current_chat_page = new_chat_page
            # 重新布局
            self.Layout()
            # 处理完事件后，不要忘记调用event.Skip()，以便事件可以继续传播到其他处理器
            event.Skip()

    class ConnectPage(wx.Panel):
        def __init__(self, parent):
            wx.Panel.__init__(self, parent)
            print("联系人")

    class SitePage(wx.Panel):
        def __init__(self, parent):
            wx.Panel.__init__(self, parent)
            print("设置")

    class FindPage(wx.Panel):
        def __init__(self, parent):
            wx.Panel.__init__(self, parent)
            print("查找")