Python網路程式設計（子程式的建立與處理、簡單群聊工具）

• 父程式建立子程式等待程式退出
• 子程式建立下一個程式，然後立即退出
• 二級子程式成為孤兒程式  處理具體工作

• 使用multiprocessing建立程式子程式同樣複製父程式的全部記憶體空間
• 之後有自己的獨立空間 執行上互不干擾 子程式也有自己的PID特有資源等
• 使用multiprocessing建立子程式，一般父程式功能就是建立子程式
• 回收子程式，返回事件交給子程式完成

from socket import * 
import os, sys


# 傳送管理員訊息
def do_child(s, addr):
    while True:
        msg = input("管理員訊息:")
        msg = "C 管理員 " + msg
        s.sendto(msg.encode(), addr)


# 使用者登入
def do_login(s, user, name, addr):
    if (name in user) or name == "管理員":
        s.sendto("該使用者已存在".encode(), addr)
        return
    s.sendto(b`OK`, addr)
    # 通知所有人
    msg = "
歡迎 %s 進入聊天室" % name
    for i in user:
        s.sendto(msg.encode(), user[i])
    # 插入user
    user[name] = addr


def do_chat(s, user, name, data):
    msg = "
{} 說: {}".format(name, data)
    for i in user:
        if i != name:
            s.sendto(msg.encode(), user[i])


def do_quit(s, user, name):
    msg = "
%s 離開了聊天室" % name
    for i in user:
        if i == name:
            s.sendto(b`EXIT`, user[i])
        else:
            s.sendto(msg.encode(), user[i])
    del user[name]  # 刪除離開的使用者


# 接收客戶端請求並處理
def do_parent(s):
    # 用於儲存使用者 {`Alex`:(`127.0.0.1`,8888)}
    user = {}
    while True:
        msg, addr = s.recvfrom(1024)
        msgList = msg.decode().split(` `)
        if msgList[0] == `L`:
            do_login(s, user, msgList[1], addr)
        elif msgList[0] == `C`:
            # "C Levi [I miss you]"
            data = ` `.join(msgList[2:])
            do_chat(s, user, msgList[1], data)
        elif msgList[0] == `Q`:
            do_quit(s, user, msgList[1])


# 建立套接字，網路連線，建立父子程式
def main():
    # server address
    ADDR = (`0.0.0.0`, 8888)
    # 建立套接字
    s = socket(AF_INET, SOCK_DGRAM)
    s.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
    s.bind(ADDR)

    # 建立父子程式
    pid = os.fork()
    if pid < 0:
        sys.exit("建立程式失敗")
    elif pid == 0:
        do_child(s, ADDR)
    else:
        do_parent(s)

if __name__ == "__main__":
    main()

from socket import * 
import sys, os


def login(s, ADDR):
    while True:
        name = input("請輸入使用者名稱:")
        msg = "L " + name
        s.sendto(msg.encode(), ADDR)
        # 接收登入結果
        data, addr = s.recvfrom(1024)
        if data.decode() == `OK`:
            print("@進入聊天室@")
            return name
        else:
            print(data.decode())


# 傳送訊息
def do_child(s, name, addr):
    while True:
        text = input("發言(quit退出):")
        # 退出
        if text.strip() == "quit":
            msg = "Q " + name
            s.sendto(msg.encode(), addr)
            sys.exit("退出聊天室")

        msg = "C %s %s" % (name, text)
        s.sendto(msg.encode(), addr)


# 接收訊息
def do_parent(s):
    while True:
        msg, addr = s.recvfrom(1024)
        if msg.decode() == `EXIT`:
            sys.exit(0)
        print(msg.decode() + "
發言(quit退出):",end="")


# main控制套接字的建立
def main():
    if len(sys.argv) < 3:
        print("argv is error")
        return
    HOST = sys.argv[1]
    PORT = int(sys.argv[2])
    ADDR = (HOST, PORT)

    s = socket(AF_INET, SOCK_DGRAM)

    name = login(s, ADDR)
    if name:
        pid = os.fork()
        if pid < 0:
            sys.exit("建立子程式失敗")
        elif pid == 0:
            do_child(s, name, ADDR)
        else:
            do_parent(s)
    else:
        return


if __name__ == "__main__":
    main()