怎么用python代碼實(shí)現(xiàn)簡(jiǎn)易聊天室

這篇文章主要介紹了怎么用python代碼實(shí)現(xiàn)簡(jiǎn)易聊天室的相關(guān)知識(shí)，內(nèi)容詳細(xì)易懂，操作簡(jiǎn)單快捷，具有一定借鑒價(jià)值，相信大家閱讀完這篇怎么用python代碼實(shí)現(xiàn)簡(jiǎn)易聊天室文章都會(huì)有所收獲，下面我們一起來(lái)看看吧。

1.功能：類(lèi)似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ù)端和客戶(hù)端

服務(wù)端處理請(qǐng)求，發(fā)送管理員消息
客戶(hù)端執(zhí)行各種功能

2.套接字選擇：udp套接字

3.消息發(fā)送模型：轉(zhuǎn)發(fā)
客戶(hù)端 ~> 服務(wù)端 ~> 其他客戶(hù)端

4.存儲(chǔ)用戶(hù)信息：{name：addr}

5.處理收發(fā)關(guān)系：多進(jìn)程分別處理收發(fā)

3.注意事項(xiàng)

1.設(shè)計(jì)封裝方案

2.寫(xiě)一個(gè)功能模塊測(cè)試一個(gè)模塊

3.注意注釋的添加

#coding =utf-8
"""
chat room
env:python3.5
exc:socket and fork
name:mianmabb
email:mianmabb@163.com
服務(wù)端功能:
1.搭建網(wǎng)絡(luò)通信
2.處理進(jìn)入聊天室
    * 接收姓名
    * 判斷是否允許進(jìn)入
    * 將結(jié)果反饋給客戶(hù)端
    * 如果不允許則結(jié)束，允許則將用戶(hù)插入數(shù)據(jù)結(jié)構(gòu)
    * 給其他人發(fā)送通知
3.處理聊天
    * 接收消息，判斷消息類(lèi)型,分為L(zhǎng)(輸入姓名),C(發(fā)消息),Q(退出聊天室)
    * 將消息轉(zhuǎn)發(fā)
4.處理退出聊天室
5.發(fā)送管理員消息 
"""

from socket import *
from os import *
from sys import *

user = {}   #創(chuàng)建空字典用來(lái)存儲(chǔ)用戶(hù)的昵稱(chēng)和地址

#處理登錄
def do_login(s,name,addr):
    if name in user:    #判斷昵稱(chēng)是否已經(jīng)存在
        s.sendto("該昵稱(chēng)已被占用".encode(),addr)
        return
    else:    #昵稱(chēng)不存在,則發(fā)送約定好的"OK"
        s.sendto(b"OK",addr)

    #功能:有人進(jìn)入聊天室,其他人會(huì)收到消息
    msg = "
   歡迎 %s 進(jìn)入聊天室   "%name
    for i in user:    #發(fā)送該條消息給其他用戶(hù)
        s.sendto(msg.encode(),user[i])
    
    user[name] = addr   #將該用戶(hù)插入數(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ā)送該用戶(hù)退出的消息
            s.sendto(msg.encode(),user[i])
        else:   #給該用戶(hù)客戶(hù)端發(fā)送約定好的EXIT讓父進(jìn)程退出
            s.sendto(b"EXIT",user[i])
    del user[name]   #刪除字典中該用戶(hù)


#處理請(qǐng)求
def do_request(s):
    #循環(huán)接受所有客戶(hù)請(qǐng)求
    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":    #判斷消息類(lèi)型
            do_login(s,msgList[1],addr)

        elif msgList[0] == "C":
            text = " ".join(msgList[2:])  #將消息中可能有的空格加回來(lái)
            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)程用來(lái)發(fā)送管理員消息
        while True:
            try:
                text = input("管理員 : ")
            except KeyboardInterrupt:    #捕獲子進(jìn)程直接退出錯(cuò)誤
                exit()

            msg ="C 管理員 %s"%text
            s.sendto(msg.encode(),ADDR)

    else:   #父進(jìn)程用來(lái)處理請(qǐng)求
        do_request(s)



main()

客戶(hù)端功能：

"""
1.搭建通信
2.進(jìn)入聊天室
    * 輸入姓名
    * 發(fā)送給服務(wù)器
    * 接收服務(wù)器反饋
    * 不允許則重新輸入，允許則進(jìn)入聊天室
    * 創(chuàng)建新的進(jìn)程用于消息收發(fā)
3.聊天
    * 循環(huán)發(fā)送消息  消息類(lèi)型分為L(zhǎng)(輸入姓名),C(發(fā)消息),Q(退出聊天室)
    * 循環(huán)接收消息
4.退出聊天室
5.接受管理員消息
"""


from socket import *
from os import *
from sys import *

ADDR = ("127.0.0.1",8888)    #填寫(xiě)服務(wù)端地址

#循環(huán)發(fā)送消息
def send_msg(s,name):
    while True:
        try:
            text = input()   #客戶(hù)輸入要發(fā)送的消息
        except KeyboardInterrupt:   #子進(jìn)程 防止用戶(hù)Ctrl+C直接退出
            text = "quit"

        if text.strip() == "quit":    #規(guī)定輸入quit退出
            msg = "Q " + name     #消息類(lèi)型,姓名
            s.sendto(msg.encode(),ADDR)
            exit("您已退出聊天室")
        else:
            msg = "C %s %s"%(name,text)  #消息類(lèi)型,姓名,消息
            s.sendto(msg.encode(),ADDR)

#循環(huán)接收信息
def recv_msg(s):
    while True:
        try:
            data,addr = s.recvfrom(1024)
        except KeyboardInterrupt:    #父進(jìn)程 防止用戶(hù)Ctrl+C直接退出
            exit()
        if data.decode() =="EXIT":   #當(dāng)用戶(hù)退出,無(wú)需再收消息,約定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("請(qǐng)輸入昵稱(chēng):")     #輸入姓名
        if not name:
            return
        msg = "L " +name
        #發(fā)送請(qǐng)求
        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)行客戶(hù)端

關(guān)于“怎么用python代碼實(shí)現(xiàn)簡(jiǎn)易聊天室”這篇文章的內(nèi)容就介紹到這里，感謝各位的閱讀！相信大家對(duì)“怎么用python代碼實(shí)現(xiàn)簡(jiǎn)易聊天室”知識(shí)都有一定的了解，大家如果還想學(xué)習(xí)更多知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道。