1 importtkinter2 from tkinter importmessagebox3 importjson4 importtime5 importthreading6 importselect7 from socket import *

8 importtraceback9 from chat importclient_draw10

11

12 classClient:13

14 #配置连接

15 defconnect(self):16 #创建socket

17 self.s =socket(AF_INET, SOCK_STREAM)18 #服务器端和客户端均在同个机器上运行

19 remote_host =gethostname()20 #设置端口号

21 port = 1200

22 #发起连接

23 self.s.connect((remote_host, port))24 print("从%s成功连接到%s" %(self.s.getsockname(), self.s.getpeername()))25 returnself.s26

27 #监听(接收)消息

28 defreceive(self, s):29 #需要监控的对象列表

30 self.my =[s]31 while 1:32 print("监听中...")33

34 #实参:

35 #第1个实参 self.my：可读的对象，监听服务器端的响应消息

36 #第2个实参：可写的对象(本例不用)

37 #第3个实参：出现异常的对象(本例不用)

38 #这三个参数内容都是被操作系统监控的，即select.select()会执行系统内核代码

39 #1)当有事件发生时，立马往下执行代码；否则阻塞监控10秒

40 #2)若监控10秒了仍无事件发生，才往下执行

41 rl, wl, error = select.select(self.my, [], [], 10)42 #返回值：

43 #rl：监听某个文件描述符是否发生了读的事件(server给client发了数据)

44 #rl列表一开始为空，只有当s发生事件了(如客户端与服务器端建立了连接)，才会将s加到rl中

45 #wl：监听某个文件描述符是否发生了写的事件(如client给server发了数据)

46 #error：监听某个文件描述符是否发生了异常事件

47 #如果发生事件的对象是客户端连接对象，则代表收到服务器端数据

48 if s inrl:49 try:50 data = s.recv(1024).decode("utf-8")51 data_dict =json.loads(data)52 #根据服务器端返回的type值，执行不同逻辑

53 type = data_dict["type"]54 #登录逻辑

55 if type == "login":56 #登录成功，跳转聊天页面

57 if "000" == data_dict["code"]:58 nickname = data_dict["nickname"]59 self.chat_interface(nickname)60 #登录失败，获取失败信息

61 else:62 messagebox.showinfo(title="登录提示", message=data_dict["msg"])63 #注册逻辑

64 elif type == "register":65 #注册成功，跳转聊天页面

66 if "000" == data_dict["code"]:67 nickname = data_dict["nickname"]68 messagebox.showinfo(title="进入聊天室", message=data_dict["msg"])69 self.chat_interface(nickname)70 #注册失败

71 else:72 messagebox.showinfo(title="注册提示", message=data_dict["msg"])73 #聊天逻辑

74 elif type == "chat":75 message = data_dict["message"]76 nickname = data_dict["nickname"]77 isMy = data_dict["isMy"]78 chat_time = " " + nickname + "\t" + time.strftime("%Y/%m/%d %H:%M:%S", time.localtime()) + "\n"

79 #聊天页面，显示发送人及发送时间

80 self.txtMsgList.insert(tkinter.END, chat_time, "DimGray")81 #如果是自己发的消息，字体使用'DarkTurquoise'

82 if "yes" ==isMy:83 self.txtMsgList.insert(tkinter.END, " " + message + "\n\n", 'DarkTurquoise')84 #如果是别人发的消息，字体使用'Black'

85 else:86 self.txtMsgList.insert(tkinter.END, " " + message + "\n\n", 'Black')87 #插入消息时，自动滚动到底部

88 self.txtMsgList.see(tkinter.END)89 except(ConnectionAbortedError, ConnectionResetError):90 #将连接对象从监听列表去掉

91 self.my.remove(s)92 print("客户端发生连接异常，与服务器端断开连接")93 traceback.print_exc()94 s.close()95 exceptException as e:96 print("客户端发生了其它异常:")97 traceback.print_exc()98 s.close()99

100 #进入注册页面

101 defregister_interface(self):102 client_draw.draw_register(self)103

104 #进入聊天页面

105 defchat_interface(self, nickname):106 client_draw.draw_chat(self, nickname)107

108 #返回登录页面

109 defreturn_login_interface(self):110 #将不需要的控件先销毁

111 self.label_nickname.destroy()112 self.input_nickname.destroy()113 self.label_password.destroy()114 self.input_password.destroy()115 client_draw.draw_login(self)116

117 #获取输入框内容，进行注册验证

118 defverify_register(self):119 username =self.input_account.get()120 password =self.input_password.get()121 nickname =self.input_nickname.get()122 try:123 register_data ={}124 register_data["type"] = "register"

125 register_data["username"] =username126 register_data["password"] =password127 register_data["nickname"] =nickname128 #将dict类型转为json字符串，便于网络传输

129 data =json.dumps(register_data)130 self.s.send(data.encode("utf-8"))131 except:132 traceback.print_exc()133

134 #获取输入框内容，进行登录校验

135 defverify_login(self):136 account =self.input_account.get()137 password =self.input_password.get()138 try:139 login_data ={}140 login_data["type"] = "login"

141 login_data["username"] =account142 login_data["password"] =password143 data =json.dumps(login_data)144 self.s.send(data.encode('utf-8'))145 except:146 traceback.print_exc()147

148 #获取输入框内容，发送消息

149 defsend_msg(self):150 message = self.txtMsg.get('0.0', tkinter.END).strip()151 if notmessage:152 messagebox.showinfo(title='发送提示', message="发送内容不能为空，请重新输入")153 return

154 self.txtMsg.delete('0.0', tkinter.END)155 try:156 chat_data ={}157 chat_data["type"] = "chat"

158 chat_data["message"] =message159 data =json.dumps(chat_data)160 self.s.send(data.encode('utf-8'))161 except:162 traceback.print_exc()163

164 #发送消息事件

165 defsend_msg_event(self, event):166 #如果捕捉到键盘的回车按键，触发消息发送

167 if event.keysym == 'Return':168 self.send_msg()169

170 #聊天页面，点击右上角退出时执行

171 defon_closing(self):172 if messagebox.askokcancel("退出提示", "是否离开聊天室？"):173 self.window.destroy()174

175

176 defmain():177 chatRoom =Client()178 client =chatRoom.connect()179 t = threading.Thread(target=chatRoom.receive, args=(client,)) #创建一个线程，监听消息

180 t.start()181 #创建主窗口,用于容纳其它组件

182 chatRoom.window =tkinter.Tk()183 #登录界面控件创建、布局

184 client_draw.draw_login(chatRoom)185 #进入事件(消息)循环

186 tkinter.mainloop()187

188 if __name__ == "__main__":189 main()