本文實(shí)例為大家分享了python實(shí)現(xiàn)簡易聊天室的具體代碼,供大家參考,具體內(nèi)容如下
群聊聊天室
1.功能:類似qq群聊功能
1.有人進(jìn)入聊天室需要輸入姓名,姓名不能重復(fù)
2.有人進(jìn)入聊天室,其他人會(huì)受到通知
xxx進(jìn)入聊天室
3.一個(gè)人發(fā)消息,其他人會(huì)受到消息
xxx:xxxxxxxx
4.有人退出聊天室,其他人也會(huì)收到通知
xxx退出聊天室
5.擴(kuò)展功能:服務(wù)端消息公告,服務(wù)端發(fā)送消息所有人都能收到
管理員消息:xxxxxxxx
2.確定技術(shù)模型
1.服務(wù)端和客戶端
服務(wù)端處理請求,發(fā)送管理員消息
客戶端執(zhí)行各種功能
2.套接字選擇:udp套接字
3.消息發(fā)送模型:轉(zhuǎn)發(fā)
客戶端 ~> 服務(wù)端 ~> 其他客戶端
4.存儲(chǔ)用戶信息:{name:addr}
5.處理收發(fā)關(guān)系:多進(jìn)程分別處理收發(fā)
3.注意事項(xiàng)
1.設(shè)計(jì)封裝方案
2.寫一個(gè)功能模塊測試一個(gè)模塊
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 room env:python3.5 exc:socket and fork name:mianmabb email:[email protected] 服務(wù)端功能: 1.搭建網(wǎng)絡(luò)通信 2.處理進(jìn)入聊天室 * 接收姓名 * 判斷是否允許進(jìn)入 * 將結(jié)果反饋給客戶端 * 如果不允許則結(jié)束,允許則將用戶插入數(shù)據(jù)結(jié)構(gòu) * 給其他人發(fā)送通知 3.處理聊天 * 接收消息,判斷消息類型,分為L(輸入姓名),C(發(fā)消息),Q(退出聊天室) * 將消息轉(zhuǎn)發(fā) 4.處理退出聊天室 5.發(fā)送管理員消息 ''' from socket import * from os import * from sys import * user = {} #創(chuàng)建空字典用來存儲(chǔ)用戶的昵稱和地址 #處理登錄 def do_login(s,name,addr): if name in user: #判斷昵稱是否已經(jīng)存在 s.sendto( '該昵稱已被占用' .encode(),addr) return else : #昵稱不存在,則發(fā)送約定好的'OK' s.sendto(b 'OK' ,addr) #功能:有人進(jìn)入聊天室,其他人會(huì)收到消息 msg = '\n 歡迎 %s 進(jìn)入聊天室 ' % name for i in user: #發(fā)送該條消息給其他用戶 s.sendto(msg.encode(),user[i]) user[name] = addr #將該用戶插入數(shù)據(jù)結(jié)構(gòu)(字典) #處理聊天 def do_chat(s,name,text): msg = '%s : %s' % (name,text) #設(shè)置消息顯示格式 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: #給其他人發(fā)送該用戶退出的消息 s.sendto(msg.encode(),user[i]) else : #給該用戶客戶端發(fā)送約定好的EXIT讓父進(jìn)程退出 s.sendto(b 'EXIT' ,user[i]) del user[name] #刪除字典中該用戶 #處理請求 def do_request(s): #循環(huán)接受所有客戶請求 while True : try : data,addr = s.recvfrom( 1024 ) except KeyboardInterrupt: #捕獲父進(jìn)程直接退出錯(cuò)誤 exit( '服務(wù)端退出!' ) # 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) #創(chuàng)建進(jìn)程 pid = fork() if pid < 0 : print ( 'Error' ) elif pid = = 0 : #子進(jìn)程用來發(fā)送管理員消息 while True : try : text = input ( '管理員 : ' ) except KeyboardInterrupt: #捕獲子進(jìn)程直接退出錯(cuò)誤 exit() msg = 'C 管理員 %s' % text s.sendto(msg.encode(),ADDR) else : #父進(jìn)程用來處理請求 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.進(jìn)入聊天室 * 輸入姓名 * 發(fā)送給服務(wù)器 * 接收服務(wù)器反饋 * 不允許則重新輸入,允許則進(jìn)入聊天室 * 創(chuàng)建新的進(jìn)程用于消息收發(fā) 3.聊天 * 循環(huán)發(fā)送消息 消息類型分為L(輸入姓名),C(發(fā)消息),Q(退出聊天室) * 循環(huán)接收消息 4.退出聊天室 5.接受管理員消息 ''' from socket import * from os import * from sys import * ADDR = ( '127.0.0.1' , 8888 ) #填寫服務(wù)端地址 #循環(huán)發(fā)送消息 def send_msg(s,name): while True : try : text = input () #客戶輸入要發(fā)送的消息 except KeyboardInterrupt: #子進(jìn)程 防止用戶Ctrl+C直接退出 text = 'quit' if text.strip() = = 'quit' : #規(guī)定輸入quit退出 msg = 'Q ' + name #消息類型,姓名 s.sendto(msg.encode(),ADDR) exit( '您已退出聊天室' ) else : msg = 'C %s %s' % (name,text) #消息類型,姓名,消息 s.sendto(msg.encode(),ADDR) #循環(huán)接收信息 def recv_msg(s): while True : try : data,addr = s.recvfrom( 1024 ) except KeyboardInterrupt: #父進(jìn)程 防止用戶Ctrl+C直接退出 exit() if data.decode() = = 'EXIT' : #當(dāng)用戶退出,無需再收消息,約定EXIT讓父進(jìn)程退出 exit() #退出父進(jìn)程 print (data.decode()) #創(chuàng)建網(wǎng)絡(luò)連接 def main(): s = socket(AF_INET,SOCK_DGRAM) while True : name = input ( '請輸入昵稱:' ) #輸入姓名 if not name: return msg = 'L ' + name #發(fā)送請求 s.sendto(msg.encode(),ADDR) #等待回復(fù) data,addr = s.recvfrom( 1024 ) if data.decode() = = 'OK' : print ( '您已進(jìn)入聊天室' ) break else : #登錄失敗 print (data.decode()) #直接打印服務(wù)端的錯(cuò)誤信息 #創(chuàng)建進(jìn)程 pid = fork() if pid < 0 : print ( 'Error' ) elif pid = = 0 : #子進(jìn)程發(fā)消息 send_msg(s,name) else : #父進(jìn)程收消息 recv_msg(s) main() |
先運(yùn)行服務(wù)端,再運(yùn)行客戶端
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持服務(wù)器之家。
原文鏈接:https://blog.csdn.net/mianmabb/article/details/88596190