python實現(xiàn)簡易聊天室(Linux終端)
本文實例為大家分享了python實現(xiàn)簡易聊天室的具體代碼,供大家參考,具體內(nèi)容如下
群聊聊天室
1.功能:類似qq群聊功能1.有人進入聊天室需要輸入姓名,姓名不能重復
2.有人進入聊天室,其他人會受到通知xxx進入聊天室
3.一個人發(fā)消息,其他人會受到消息xxx:xxxxxxxx
4.有人退出聊天室,其他人也會收到通知xxx退出聊天室
5.擴展功能:服務端消息公告,服務端發(fā)送消息所有人都能收到管理員消息:xxxxxxxx
2.確定技術(shù)模型1.服務端和客戶端
服務端處理請求,發(fā)送管理員消息客戶端執(zhí)行各種功能
2.套接字選擇:udp套接字
3.消息發(fā)送模型:轉(zhuǎn)發(fā)客戶端 ~> 服務端 ~> 其他客戶端
4.存儲用戶信息:{name:addr}
5.處理收發(fā)關(guān)系:多進程分別處理收發(fā)
3.注意事項1.設計封裝方案
2.寫一個功能模塊測試一個模塊
3.注意注釋的添加
#coding =utf-8’’’chat roomenv:python3.5exc:socket and forkname:mianmabbemail:mianmabb@163.com服務端功能:1.搭建網(wǎng)絡通信2.處理進入聊天室 * 接收姓名 * 判斷是否允許進入 * 將結(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)建空字典用來存儲用戶的昵稱和地址#處理登錄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) #功能:有人進入聊天室,其他人會收到消息 msg = ’n 歡迎 %s 進入聊天室 ’%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) #設置消息顯示格式 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讓父進程退出 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: #捕獲父進程直接退出錯誤 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) #創(chuàng)建進程 pid = fork() if pid < 0:print(’Error’) elif pid == 0: #子進程用來發(fā)送管理員消息while True: try:text = input(’管理員 : ’) except KeyboardInterrupt: #捕獲子進程直接退出錯誤exit() msg =’C 管理員 %s’%text s.sendto(msg.encode(),ADDR) else: #父進程用來處理請求do_request(s)main()
客戶端功能:
’’’1.搭建通信2.進入聊天室 * 輸入姓名 * 發(fā)送給服務器 * 接收服務器反饋 * 不允許則重新輸入,允許則進入聊天室 * 創(chuàng)建新的進程用于消息收發(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) #填寫服務端地址#循環(huán)發(fā)送消息def send_msg(s,name): while True:try: text = input() #客戶輸入要發(fā)送的消息except KeyboardInterrupt: #子進程 防止用戶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: #父進程 防止用戶Ctrl+C直接退出 exit()if data.decode() ==’EXIT’: #當用戶退出,無需再收消息,約定EXIT讓父進程退出 exit() #退出父進程print(data.decode())#創(chuàng)建網(wǎng)絡連接def main(): s = socket(AF_INET,SOCK_DGRAM) while True:name = input(’請輸入昵稱:’) #輸入姓名if not name: returnmsg = ’L ’ +name#發(fā)送請求s.sendto(msg.encode(),ADDR)#等待回復data,addr = s.recvfrom(1024)if data.decode() == ’OK’: print(’您已進入聊天室’) breakelse: #登錄失敗 print(data.decode()) #直接打印服務端的錯誤信息 #創(chuàng)建進程 pid = fork() if pid < 0:print(’Error’) elif pid == 0: #子進程發(fā)消息send_msg(s,name) else: #父進程收消息recv_msg(s)main()
先運行服務端,再運行客戶端
以上就是本文的全部內(nèi)容,希望對大家的學習有所幫助,也希望大家多多支持好吧啦網(wǎng)。
相關(guān)文章: