本文實例為大家分享了python實現簡易聊天室的具體代碼，供大家參考，具體內容如下
群聊聊天室
1.功能：類似qq群聊功能
1.有人進入聊天室需要輸入姓名，姓名不能重復
2.有人進入聊天室，其他人會受到通知
xxx進入聊天室
3.一個人發消息，其他人會受到消息
xxx：xxxxxxxx
4.有人退出聊天室，其他人也會收到通知
xxx退出聊天室
5.擴展功能：服務端消息公告，服務端發送消息所有人都能收到
管理員消息：xxxxxxxx
2.確定技術模型
1.服務端和客戶端
服務端處理請求，發送管理員消息
客戶端執行各種功能
2.套接字選擇：udp套接字
3.消息發送模型：轉發
客戶端 ~> 服務端 ~> 其他客戶端
4.存儲用戶信息：{name：addr}
5.處理收發關系：多進程分別處理收發
3.注意事項
1.設計封裝方案
2.寫一個功能模塊測試一個模塊
3.注意注釋的添加
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#coding =utf-8'''chat roomenv:python3.5exc:socket and forkname:mianmabbemail:mianmabb@163.com服務端功能:1.搭建網絡通信2.處理進入聊天室 * 接收姓名 * 判斷是否允許進入 * 將結果反饋給客戶端 * 如果不允許則結束，允許則將用戶插入數據結構 * 給其他人發送通知3.處理聊天 * 接收消息，判斷消息類型,分為L(輸入姓名),C(發消息),Q(退出聊天室) * 將消息轉發4.處理退出聊天室5.發送管理員消息 '''from socket import *from os import *from sys import *user = {} #創建空字典用來存儲用戶的昵稱和地址#處理登錄def do_login(s,name,addr): if name in user: #判斷昵稱是否已經存在 s.sendto('該昵稱已被占用'.encode(),addr) return else: #昵稱不存在,則發送約定好的'OK' s.sendto(b'OK',addr) #功能:有人進入聊天室,其他人會收到消息 msg = '\n 歡迎 %s 進入聊天室 '%name for i in user: #發送該條消息給其他用戶 s.sendto(msg.encode(),user[i]) user[name] = addr #將該用戶插入數據結構(字典)#處理聊天def do_chat(s,name,text): msg = '%s : %s'%(name,text) #設置消息顯示格式 for i in user: s.sendto(msg.encode(),user[i])#處理退出def do_quit(s,name): msg = '%s 退出了聊天室'%name for i in user: if i != name: #給其他人發送該用戶退出的消息 s.sendto(msg.encode(),user[i]) else: #給該用戶客戶端發送約定好的EXIT讓父進程退出 s.sendto(b'EXIT',user[i]) del user[name] #刪除字典中該用戶#處理請求def do_request(s): #循環接受所有客戶請求 while True: try: data,addr = s.recvfrom(1024) except KeyboardInterrupt: #捕獲父進程直接退出錯誤 exit('服務端退出!') # print(data.decode()) msgList = data.decode().split() #按空格拆分為列表,方便索引 if msgList[0] == 'L': #判斷消息類型 do_login(s,msgList[1],addr) elif msgList[0] == 'C': text = ' '.join(msgList[2:]) #將消息中可能有的空格加回來 do_chat(s,msgList[1],text) elif msgList[0] == 'Q': do_quit(s,msgList[1])def main(): s = socket(AF_INET,SOCK_DGRAM) ADDR = ('0.0.0.0',8888) s.bind(ADDR) #創建進程 pid = fork() if pid < 0: print('Error') elif pid == 0: #子進程用來發送管理員消息 while True: try: text = input('管理員 : ') except KeyboardInterrupt: #捕獲子進程直接退出錯誤 exit() msg ='C 管理員 %s'%text s.sendto(msg.encode(),ADDR) else: #父進程用來處理請求 do_request(s)main() |
客戶端功能：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
'''1.搭建通信2.進入聊天室 * 輸入姓名 * 發送給服務器 * 接收服務器反饋 * 不允許則重新輸入，允許則進入聊天室 * 創建新的進程用于消息收發3.聊天 * 循環發送消息 消息類型分為L(輸入姓名),C(發消息),Q(退出聊天室) * 循環接收消息4.退出聊天室5.接受管理員消息'''from socket import *from os import *from sys import *ADDR = ('127.0.0.1',8888) #填寫服務端地址#循環發送消息def send_msg(s,name): while True: try: text = input() #客戶輸入要發送的消息 except KeyboardInterrupt: #子進程 防止用戶Ctrl+C直接退出 text = 'quit' if text.strip() == 'quit': #規定輸入quit退出 msg = 'Q ' + name #消息類型,姓名 s.sendto(msg.encode(),ADDR) exit('您已退出聊天室') else: msg = 'C %s %s'%(name,text) #消息類型,姓名,消息 s.sendto(msg.encode(),ADDR)#循環接收信息def recv_msg(s): while True: try: data,addr = s.recvfrom(1024) except KeyboardInterrupt: #父進程 防止用戶Ctrl+C直接退出 exit() if data.decode() =='EXIT': #當用戶退出,無需再收消息,約定EXIT讓父進程退出 exit() #退出父進程 print(data.decode())#創建網絡連接def main(): s = socket(AF_INET,SOCK_DGRAM) while True: name = input('請輸入昵稱:') #輸入姓名 if not name: return msg = 'L ' +name #發送請求 s.sendto(msg.encode(),ADDR) #等待回復 data,addr = s.recvfrom(1024) if data.decode() == 'OK': print('您已進入聊天室') break else: #登錄失敗 print(data.decode()) #直接打印服務端的錯誤信息 #創建進程 pid = fork() if pid < 0: print('Error') elif pid == 0: #子進程發消息 send_msg(s,name) else: #父進程收消息 recv_msg(s)main() |
先運行服務端,再運行客戶端
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持服務器之家。
原文鏈接：https://blog.csdn.net/mianmabb/article/details/88596190
