266 lines
11 KiB
Python
266 lines
11 KiB
Python
import wx.richtext as rt
|
||
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)
|
||
|
||
self.root_path = rf'.\{Id}'
|
||
self.Account_path = self.root_path + r'\Account.csv'
|
||
self.Contacts_path = self.root_path + r'\Contacts.csv'
|
||
self.History_path = self.root_path + r'\History.csv'
|
||
|
||
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]
|
||
if Type == 'text':
|
||
contents = ['My' if Contact == send else 'Ta']
|
||
contents.append('') # 添加一个空字符串作为消息的一部分
|
||
for i in range(0, len(data), 30):
|
||
contents.append(data[i:i + 30])
|
||
contents.append(UpDataTime)
|
||
|
||
|
||
for content in contents:
|
||
chat_panel.chat_receive_text.AppendText(content+'\n')
|
||
|
||
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 = 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 = data['send']
|
||
receive = data['receive']
|
||
Type = data['Type']
|
||
content = data['content']
|
||
UpDataTime = data['UpDataTime']
|
||
if send == str(self.userid):
|
||
Contact = receive
|
||
elif receive == str(self.userid):
|
||
Contact = send
|
||
wx.CallAfter(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
|
||
chat_page.Show()
|
||
# 重新布局
|
||
self.Layout()
|
||
|
||
def ChatPage_add_Contact_tab(self):
|
||
chat_panel = wx.Panel(self)
|
||
chat_box = wx.BoxSizer(wx.VERTICAL)
|
||
|
||
chat_receive_box = wx.BoxSizer(wx.HORIZONTAL)
|
||
chat_panel.chat_receive_text = rt.RichTextCtrl(chat_panel, style=wx.TE_MULTILINE | wx.TE_READONLY)
|
||
chat_receive_box.Add(chat_panel.chat_receive_text, 1, wx.EXPAND, 0)
|
||
chat_box.Add(chat_receive_box, 2, wx.EXPAND, 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("查找")
|