python實現簡易聊天室(Linux終端)

#coding =utf-8
'''
chat room
env:python3.5
exc:socket and fork
name:mianmabb
email:[email protected]
服務端功能:
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.聊天
    * 循環發送消息  消息類型分為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()