#!usr/bin/env python
# -*- coding:utf-8 _*-
"""
@author:zhengxin
@file: test_socket3.py
@time: 2023年2月1日 17:19
# @describe: 练习二:使用makefile改写群聊类
"""
import datetime
import socket
import threading
import logging

FORMAT = "%(asctime)s %(threadName)s %(thread)d %(message)s"
logging.basicConfig(format=FORMAT, level=logging.INFO)


class ChatServer():
 def __init__(self, ip='0.0.0.0', port=9999):
 # def __init__(self, ip='127.0.0.1', port=9999):
 self.addr = (ip, port)
 self.sock = socket.socket()
 self.clients = {}
 self.event = threading.Event()

 def start(self):
 self.sock.bind(self.addr)
 # 服务启动
 self.sock.listen()
 # 启动线程
 threading.Thread(target=self.accept, name='accept').start()

 def accept(self):
 while not self.event.is_set():
 # 阻塞
 s, raddr = self.sock.accept()
 # makefile
 f = s.makefile(mode='rw')
 logging.info(s)
 logging.info(f)
 logging.info(raddr)
 self.clients[raddr] = f
 # 启动线程
 threading.Thread(target=self.recv, name='recv', args=(f, raddr)).start()

 def recv(self, f, addr): # 很多线程
 while not self.event.is_set():
 try:
 # 阻塞,bytes
 # data = sock.recv(1024)
 data = f.readline() # string, 换行符\n
 logging.info(data)
 except Exception as e:
 logging.error(e)
 data = 'quit'
 if data == 'quit':
 self.clients.pop(addr)
 f.close()
 break

 msg = "ack{}. {} {}".format(
 addr,
 datetime.datetime.now().strftime("%Y/%m/%d-%H:%M:%s"),
 data)
 # 单个发送
 # sock.send('ack {}'.format(data.decode()).encode())
 # 多个发送
 for s in self.clients.values():
 # s.send(msg)
 f.write(msg)
 f.flush()

 def stop(self):
 """ 停止服务 """
 for s in self.clients.values():
 s.close()
 self.sock.close()
 self.event.set()


def main():
 cs = ChatServer()
 cs.start()

 while True:
 cmd = input(">>>>>>")
 # 退出
 if cmd.strip() == 'quit':
 cs.stop()
 threading.Event.wait(3)
 break
 logging.info(threading.enumerate())


if __name__ == '__main__':
 main()