欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

Python聊天室實(shí)例程序分享

 更新時(shí)間:2016年01月05日 11:56:32   作者:hazir  
這篇文章主要為大家分享了Python Socket編程之聊天室示例程序代碼,感興趣的小伙伴們可以參考一下

上一篇 我們學(xué)習(xí)了簡(jiǎn)單的Python TCP Socket 編程,通過(guò)分別寫(xiě)服務(wù)端和客戶端的代碼了解基本的 Python Socket 編程模型。本文再通過(guò)一個(gè)例子來(lái)加強(qiáng)一下對(duì) Socket 編程的理解。

一、聊天室程序需求
我們要實(shí)現(xiàn)的是簡(jiǎn)單的聊天室的例子,就是允許多個(gè)人同時(shí)一起聊天,每個(gè)人發(fā)送的消息所有人都能接收到,類似于 QQ 群的功能,而不是點(diǎn)對(duì)點(diǎn)的 QQ 好友之間的聊天。如下圖:

我們要實(shí)現(xiàn)的有兩部分:

  • Chat Server:聊天服務(wù)器,負(fù)責(zé)與用戶建立 Socket 連接,并將某個(gè)用戶發(fā)送的消息廣播到所有在線的用戶。
  • Telnet Client:用戶聊天客戶端,可以輸入聊天的內(nèi)容并發(fā)送,同時(shí)可以顯示其他用戶的消息記錄。

同樣,我們的消息通信采用 TCP 連接保證可靠性。在分別對(duì)服務(wù)端和客戶端進(jìn)行程序設(shè)計(jì)之前,首先要學(xué)習(xí)一下 Python 中實(shí)現(xiàn)異步 I/O 的一個(gè)函數(shù) —— select。

二、Python 異步I/O
Python 在 select 模塊中提供了異步 I/O(Asynchronous I/O),這與 Linux 下的 select 機(jī)制相似,但進(jìn)行一些簡(jiǎn)化。我首先介紹一下 select,然后告訴你在 Python 中如何使用它。

前面文章使用多線程來(lái)并行處理多路 socket I/O,這里介紹的select 方法允許你響應(yīng)不同 socket 的多個(gè)事件以及其它不同事件。例如你可以讓 select 在某個(gè) socket 有數(shù)據(jù)到達(dá)時(shí),或者當(dāng)某個(gè) socket 可以寫(xiě)數(shù)據(jù)時(shí),又或者是當(dāng)某個(gè) socket 發(fā)生錯(cuò)誤時(shí)通知你,好處是你可以同時(shí)響應(yīng)很多 socket 的多個(gè)事件。

Linux 下 C 語(yǔ)言的 select 使用到位圖來(lái)表示我們要關(guān)注哪些文件描述符的事件,Python 中使用 list 來(lái)表示我們監(jiān)控的文件描述符,當(dāng)有事件到達(dá)時(shí),返回的也是文件描述符的 list,表示這些文件有事件到達(dá)。下面的簡(jiǎn)單程序是表示等待從標(biāo)準(zhǔn)輸入中獲得輸入:

rlist, wlist, elist = select.select( [sys.stdin], [], [] )

print sys.stdin.read()

select 方法的三個(gè)參數(shù)都是 list 類型,分別代表讀事件、寫(xiě)事件、錯(cuò)誤事件,同樣方法返回值也是三個(gè) list,包含的是哪些事件(讀、寫(xiě)、異常)滿足了。上面的例子,由于參數(shù)只有一個(gè)事件 sys.stdin,表示只關(guān)心標(biāo)準(zhǔn)輸入事件,因此當(dāng) select 返回時(shí) rlist 只會(huì)是 [sys.stdin],表示可以從 stdin 中讀入數(shù)據(jù)了,我們使用 read 方法來(lái)讀入數(shù)據(jù)。

當(dāng)然 select 對(duì)于 socket 描述符也是有效的,下面的一個(gè)例子是創(chuàng)建了兩個(gè) socket 客戶端連接到遠(yuǎn)程服務(wù)器,select 用來(lái)監(jiān)控哪個(gè) socket 有數(shù)據(jù)到達(dá):

import socket
import select

sock1 = socket.socket( socket.AF_INET, socket.SOCK_STREAM )
sock2 = socket.socket( socket.AF_INET, socket.SOCK_STREAM )

sock1.connect( ('192.168.1.1', 25) )
sock2.connect( ('192.168.1.1', 25) )

while 1:

  # Await a read event
  rlist, wlist, elist = select.select( [sock1, sock2], [], [], 5 )

  # Test for timeout
  if [rlist, wlist, elist] == [ [], [], [] ]:
    print "Five seconds elapsed.\n"

  else:
    # Loop through each socket in rlist, read and print the available data
    for sock in rlist:
      print sock.recv( 100 )

好了,有了上面的基礎(chǔ),我們就可以來(lái)設(shè)計(jì)聊天室的服務(wù)器和客戶端了。

三、聊天室服務(wù)器
聊天室服務(wù)器主要完成下面兩件事:

  • 接收多個(gè)客戶端的連接
  • 從每個(gè)客戶端讀入消息病廣播到其它連接的客戶端

我們定義一個(gè) list 型變量 CONNECTION_LIST 表示監(jiān)聽(tīng)多個(gè) socket 事件的可讀事件,那么利用上面介紹的我們的服務(wù)器使用 select 來(lái)處理多路復(fù)用 I/O 的代碼如下:

# Get the list sockets which are ready to be read through select
read_sockets,write_sockets,error_sockets = select.select(CONNECTION_LIST,[],[])

當(dāng) select 返回時(shí),說(shuō)明在 read_sockets 上有可讀的數(shù)據(jù),這里又分為兩種情況:

1、如果是主 socket(即服務(wù)器開(kāi)始創(chuàng)建的 socket,一直處于監(jiān)聽(tīng)狀態(tài))有數(shù)據(jù)可讀,表示有新的連接請(qǐng)求可以接收,此時(shí)需要調(diào)用 accept 函數(shù)來(lái)接收新的客戶端連接,并將其連接信息廣播到其它客戶端。
2、如果是其它 sockets(即與客戶端已經(jīng)建立連接的 sockets)有數(shù)據(jù)可讀,那么表示客戶端發(fā)送消息到服務(wù)器端,使用 recv 函數(shù)讀消息,并將消息轉(zhuǎn)發(fā)到其它所有連接的客戶端。
上面兩種情況到涉及到廣播消息的過(guò)程,廣播也就是將從某個(gè) socket 獲得的消息通過(guò) CONNECTION_LIST 的每個(gè) socket (除了自身和主 socket)一個(gè)個(gè)發(fā)送出去:

def broadcast_data (sock, message):
  #Do not send the message to master socket and the client who has send us the message
  for socket in CONNECTION_LIST:
    if socket != server_socket and socket != sock :
      try :
        socket.send(message)
      except :
        # broken socket connection may be, chat client pressed ctrl+c for example
        socket.close()
        CONNECTION_LIST.remove(socket)

如果發(fā)送失敗,我們假設(shè)某個(gè)客戶端已經(jīng)斷開(kāi)了連接,關(guān)閉該 socket 病將其從連接列表中刪除。

完整的聊天室服務(wù)器源代碼如下:

# Tcp Chat server
 
import socket, select
 
#Function to broadcast chat messages to all connected clients
def broadcast_data (sock, message):
  #Do not send the message to master socket and the client who has send us the message
  for socket in CONNECTION_LIST:
    if socket != server_socket and socket != sock :
      try :
        socket.send(message)
      except :
        # broken socket connection may be, chat client pressed ctrl+c for example
        socket.close()
        CONNECTION_LIST.remove(socket)
 
if __name__ == "__main__":
   
  # List to keep track of socket descriptors
  CONNECTION_LIST = []
  RECV_BUFFER = 4096 # Advisable to keep it as an exponent of 2
  PORT = 5000
   
  server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  # this has no effect, why ?
  server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
  server_socket.bind(("0.0.0.0", PORT))
  server_socket.listen(10)
 
  # Add server socket to the list of readable connections
  CONNECTION_LIST.append(server_socket)
 
  print "Chat server started on port " + str(PORT)
 
  while 1:
    # Get the list sockets which are ready to be read through select
    read_sockets,write_sockets,error_sockets = select.select(CONNECTION_LIST,[],[])
 
    for sock in read_sockets:
      #New connection
      if sock == server_socket:
        # Handle the case in which there is a new connection recieved through server_socket
        sockfd, addr = server_socket.accept()
        CONNECTION_LIST.append(sockfd)
        print "Client (%s, %s) connected" % addr
         
        broadcast_data(sockfd, "[%s:%s] entered room\n" % addr)
       
      #Some incoming message from a client
      else:
        # Data recieved from client, process it
        try:
          #In Windows, sometimes when a TCP program closes abruptly,
          # a "Connection reset by peer" exception will be thrown
          data = sock.recv(RECV_BUFFER)
          if data:
            broadcast_data(sock, "\r" + '<' + str(sock.getpeername()) + '> ' + data)        
         
        except:
          broadcast_data(sock, "Client (%s, %s) is offline" % addr)
          print "Client (%s, %s) is offline" % addr
          sock.close()
          CONNECTION_LIST.remove(sock)
          continue
   
  server_socket.close()

在控制臺(tái)下運(yùn)行該程序:

$ python chat_server.py 
Chat server started on port 5000

四、聊天室客戶端
我們寫(xiě)一個(gè)客戶端程序可以連接到上面的服務(wù)器,完成發(fā)送消息和接收消息的過(guò)程。主要做下面兩件事:

  • 監(jiān)聽(tīng)服務(wù)器是否有消息發(fā)送過(guò)來(lái)
  • 檢查用戶的輸入,如果用戶輸入某條消息,需要發(fā)送到服務(wù)器

這里有兩個(gè) I/O 事件需要監(jiān)聽(tīng):連接到服務(wù)器的 socket 和標(biāo)準(zhǔn)輸入,同樣我們可以使用 select 來(lái)完成:

rlist = [sys.stdin, s]
     
# Get the list sockets which are readable
read_list, write_list, error_list = select.select(rlist , [], [])

那邏輯就很簡(jiǎn)單了,如果是 sys.stdin 有數(shù)據(jù)可讀,表示用戶從控制臺(tái)輸入數(shù)據(jù)并按下回車,那么就從標(biāo)準(zhǔn)輸入讀數(shù)據(jù),并發(fā)送到服務(wù)器;如果是與服務(wù)器連接的 socket 有數(shù)據(jù)可讀,表示服務(wù)器發(fā)送消息給該客戶端,那么就從 socket 接收數(shù)據(jù)。加上一些提示信息及異常處理的完整客戶端代碼如下:

# telnet program example
import socket, select, string, sys
 
def prompt() :
  sys.stdout.write('<You> ')
  sys.stdout.flush()
 
#main function
if __name__ == "__main__":
   
  if(len(sys.argv) < 3) :
    print 'Usage : python telnet.py hostname port'
    sys.exit()
   
  host = sys.argv[1]
  port = int(sys.argv[2])
   
  s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  s.settimeout(2)
   
  # connect to remote host
  try :
    s.connect((host, port))
  except :
    print 'Unable to connect'
    sys.exit()
   
  print 'Connected to remote host. Start sending messages'
  prompt()
   
  while 1:
    rlist = [sys.stdin, s]
     
    # Get the list sockets which are readable
    read_list, write_list, error_list = select.select(rlist , [], [])
     
    for sock in read_list:
      #incoming message from remote server
      if sock == s:
        data = sock.recv(4096)
        if not data :
          print '\nDisconnected from chat server'
          sys.exit()
        else :
          #print data
          sys.stdout.write(data)
          prompt()
       
      #user entered a message
      else :
        msg = sys.stdin.readline()
        s.send(msg)
        prompt()

可以在多個(gè)終端下運(yùn)行該代碼:

$ python telnet.py localhost 5000
Connected to remote host. Start sending messages
<You> hello
<You> I am fine
<('127.0.0.1', 38378)> ok good
<You>

在另一個(gè)終端顯示的信息:

<You> [127.0.0.1:39339] entered room
<('127.0.0.1', 39339)> hello
<('127.0.0.1', 39339)> I am fine
<You> ok good

總結(jié)
上面的代碼注意兩點(diǎn):

1、聊天室客戶端代碼不能在 windows 下運(yùn)行,因?yàn)榇a使用 select 同時(shí)監(jiān)聽(tīng) socket 和輸入流,在 Windows 下 select 函數(shù)是由 WinSock 庫(kù)提供,不能處理不是由 WinSock 定義的文件描述符。
2、客戶端代碼還有個(gè)缺陷是,當(dāng)某個(gè)客戶端在輸入消息但還未發(fā)送出去時(shí),服務(wù)器也發(fā)送消息過(guò)來(lái),這樣會(huì)沖刷掉客戶端正在輸入的消息。這目前來(lái)看沒(méi)辦法解決的,唯一的解決方法是使用像 ncurses 終端庫(kù)使用戶輸入和輸出獨(dú)立開(kāi),或者寫(xiě)一個(gè) GUI 的程序。
那么本文通過(guò)一個(gè)聊天室的范例進(jìn)一步學(xué)習(xí)了Python 下 Socket 編程。

相關(guān)文章

  • PyCharm第一次安裝及使用教程

    PyCharm第一次安裝及使用教程

    本文通過(guò)圖文并茂的形式給大家介紹PyCharm第一次安裝及使用教程,分步驟通過(guò)實(shí)際案例給大家操作演示,本文給大家介紹的非常詳細(xì),具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
    2020-01-01
  • python學(xué)習(xí)之hook鉤子的原理和使用

    python學(xué)習(xí)之hook鉤子的原理和使用

    這篇文章主要為大家詳細(xì)介紹了python學(xué)習(xí)之hook鉤子的原理和使用,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下
    2018-10-10
  • python實(shí)現(xiàn)不同數(shù)據(jù)庫(kù)間數(shù)據(jù)同步功能

    python實(shí)現(xiàn)不同數(shù)據(jù)庫(kù)間數(shù)據(jù)同步功能

    這篇文章主要介紹了python實(shí)現(xiàn)不同數(shù)據(jù)庫(kù)間數(shù)據(jù)同步功能,本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
    2021-02-02
  • 關(guān)于自動(dòng)化測(cè)試框架pytest的Fixture固件

    關(guān)于自動(dòng)化測(cè)試框架pytest的Fixture固件

    這篇文章主要介紹了關(guān)于自動(dòng)化測(cè)試框架pytest的Fixture固件,Fixture它其實(shí)就是一些函數(shù),會(huì)在執(zhí)行測(cè)試方法/測(cè)試函數(shù)前后加載運(yùn)行它們,需要的朋友可以參考下
    2023-03-03
  • python截取兩個(gè)單詞之間的內(nèi)容方法

    python截取兩個(gè)單詞之間的內(nèi)容方法

    今天小編就為大家分享一篇python截取兩個(gè)單詞之間的內(nèi)容方法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
    2018-12-12
  • Python命令行解析模塊詳解

    Python命令行解析模塊詳解

    這篇文章主要介紹了Python命令行解析模塊詳解,分享了相關(guān)代碼示例,小編覺(jué)得還是挺不錯(cuò)的,具有一定借鑒價(jià)值,需要的朋友可以參考下
    2018-02-02
  • Python機(jī)器學(xué)習(xí)入門(四)之Python選擇模型

    Python機(jī)器學(xué)習(xí)入門(四)之Python選擇模型

    這篇文章主要介紹了Python機(jī)器學(xué)習(xí)入門知識(shí),本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下
    2021-08-08
  • pandas的apply函數(shù)用法詳解

    pandas的apply函數(shù)用法詳解

    本文主要介紹了pandas的apply函數(shù)用法詳解,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
    2023-01-01
  • 利用Pytorch實(shí)現(xiàn)獲取特征圖的方法詳解

    利用Pytorch實(shí)現(xiàn)獲取特征圖的方法詳解

    這篇文章主要為大家詳細(xì)介紹了如何利用Pytorch實(shí)現(xiàn)獲取特征圖,包括提取單個(gè)特征圖和提取多個(gè)特征圖,文中的示例代碼講解詳細(xì),需要的可以參考一下
    2022-10-10
  • Python探索之創(chuàng)建二叉樹(shù)

    Python探索之創(chuàng)建二叉樹(shù)

    這篇文章主要介紹了Python探索之創(chuàng)建二叉樹(shù),Python的相關(guān)內(nèi)容,小編是初窺門徑。這里分享給大家一些簡(jiǎn)單知識(shí),供需要的朋友參考。
    2017-10-10

最新評(píng)論