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 = "\n欢迎 %s 进入聊天室" % name for i in user: s.sendto(msg.encode(), user[i]) # 插入user user[name] = addr def do_chat(s, user, name, data): msg = "\n{} 说: {}".format(name, data) for i in user: if i != name: s.sendto(msg.encode(), user[i]) def do_quit(s, user, name): msg = "\n%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() + "\n发言(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()