#!/usr/bin/env python
#*-* coding:utf-8 *-*
import socket
import select
import sys
import signal
class ChatServer():
  def __init__(self,host,port,timeout=10,backlog=5):
    #記錄連接的客戶(hù)端數(shù)量
    self.clients =0
    #存儲(chǔ)連接的客戶(hù)端socket和地址對(duì)應(yīng)的字典
    self.clientmap={}
    #存儲(chǔ)連接的客戶(hù)端socket
    self.outputs = []
    #建立socket
    self.server=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
    self.server.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
    self.server.bind((host,port))
    self.server.listen(backlog)
    #增加信號(hào)處理
    signal.signal(signal.SIGINT,self.sighandler) 
  def sighandler(self):
    sys.stdout.write("Shutdown Server......\n")
    #向已經(jīng)連接客戶(hù)端發(fā)送關(guān)系信息，并主動(dòng)關(guān)閉socket
    for output in self.outputs:
      output.send("Shutdown Server")
      output.close()
    #關(guān)閉listen
    self.server.close()
    sys.stdout.flush()
  #主函數(shù)，用來(lái)啟動(dòng)服務(wù)器
  def run(self):
    #需要監(jiān)聽(tīng)的可讀對(duì)象
    inputs=[self.server]
    
    runing=True
    #添加監(jiān)聽(tīng)主循環(huán)
    while runing:
      try:
        readable,writeable,exceptional = select.select(inputs,self.outputs,[])
        #此處會(huì)被select模塊阻塞，只有當(dāng)監(jiān)聽(tīng)的三個(gè)參數(shù)發(fā)生變化時(shí)，select才會(huì)返回
      except select.error,e:
        break
      #當(dāng)返回的readable中含有本地socket的信息時(shí)，表示有客戶(hù)端正在請(qǐng)求連接
      if self.server in readable:
        #接受客戶(hù)端連接請(qǐng)求
        client,addr=self.server.accept()
        sys.stdout.write("New Connection from %s\n"%str(addr))
        sys.stdout.flush()
        #更新服務(wù)器上客戶(hù)端連接情況
        #1，數(shù)量加1
        #2，self.outputs增加一列
        #3，self.clientmap增加一對(duì)
        #4, 給input添加可讀監(jiān)控
        self.clients += 1
        self.outputs.append(client)
        self.clientmap[client]=addr
        inputs.append(client)
      
      #readable中含有已經(jīng)添加的客戶(hù)端socket，并且可讀
      #說(shuō)明 1,客戶(hù)端有數(shù)據(jù)發(fā)送過(guò)來(lái)或者 2,客戶(hù)端請(qǐng)求關(guān)閉
      elif len(readable) != 0:
        #1, 取出這個(gè)列表中的socket
        csock=readable[0]
        #2, 根據(jù)這個(gè)socket，在事先存放的clientmap中，去除客戶(hù)端的地址，端口的詳細(xì)信息
        host,port = self.clientmap[csock]
        #3,取數(shù)據(jù), 或接受關(guān)閉請(qǐng)求，并處理
        #注意，這個(gè)操作是阻塞的，但是由于數(shù)據(jù)是在本地緩存之后，所以速度會(huì)非常快
        try:
          data = csock.recv(1024).strip()
          for cs in self.outputs:
            if cs != csock:
              cs.send("%s\n"%data)
        except socket.error,e:
          self.clients -= 1
          inputs.remove(csock)
          self.outputs.remove(csock)
          del self.clientmap[csock]
      #print self.outputs
    self.server.close()
        
if __name__ == "__main__":
  chat=ChatServer("",8008)
  chat.run()