Python網(wǎng)絡編程之TCP套接字簡單用法示例

本文實例講述了Python網(wǎng)絡編程之TCP套接字簡單用法。分享給大家供大家參考，具體如下：
上學期學的計算機網(wǎng)絡，因為之前還未學習python，而java則一知半解，C寫起來又麻煩，所以一直都沒有真正實現(xiàn)過TCP套接字編程。
最近學習了python，而用它來寫套接字又十分方便簡單，所以當然要試一試咯。

下面根據(jù)代碼來介紹一下最簡單的tcp程序，由客戶端輸入數(shù)據(jù)，發(fā)送給服務器，服務器加上時間后返回給客戶端    
#!/usr/bin/python
'test TCP server'
from socket import *  #套接字所需要的模塊
from time import ctime  #時間模塊
HOST = ''  #主機名，為空，表示可以綁定在任何有效的地址上。
PORT = 21567  #隨機選的可用端口
BUFSIZ = 1024  #發(fā)送數(shù)據(jù)的緩沖區(qū)
ADDR = (HOST, PORT)
tcpSerSock = socket(AF_INET, SOCK_STREAM)  #第一個參數(shù)是socket_family,此處用常用的AF_INET,第二個參數(shù)是流套接字，即創(chuàng)建的是tcp
tcpSerSock.bind(ADDR)  #套接字與地址綁定
tcpSerSock.listen(5)  #開始監(jiān)聽，參數(shù)表示允許的最大連接數(shù)
while True:
  print 'waiting for connection...'
  tcpCliSock, addr = tcpSerSock.accept()  #打開大門等待客人，accept返回一個元組(connection,address)。第一個元素是新的socket對象，服務器必須通過它與客戶通信；第二個是客戶的地址。
  print '...connected from:',addr
  while True:
    data = tcpCliSock.recv(BUFSIZ)  #洗耳恭聽，把客人的話記錄到data中
    if not data:  #客人話說完了，送客
      break
    tcpCliSock.send('[%s] %s' %(ctime(), data))  #發(fā)送當前時間，以及數(shù)據(jù)給客人
  tcpCliSock.close()  #送客，關閉與該客人通信的套接字
tcpSerSock.close()  #因為上面是死循環(huán)，此處不是必須的，只是提醒大家離開的時候務必要關門！

上面用到了兩個套接字，第一個tcpSerSock稱為歡迎套接字，第二個tcpCliSock才是真正負責與客戶端通信的套接字，這樣做的好處是，一個服務器可以同時接待多個客人。

下面則是客戶端的代碼，非常簡單：    
#!/usr/bin/python
'test tcp client'
from socket import *
HOST = 'localhost'  #因為我是在同一臺機器上運行，所以是localhost，不同主機的話，此處需要改成服務器地址
PORT = 21567
BUFSIZ = 1024
ADDR = (HOST, PORT)
tcpCliSock = socket(AF_INET, SOCK_STREAM)  #客戶端只需要一個套接字
tcpCliSock.connect(ADDR)  #發(fā)起連接
while True:
  data = raw_input('>')
  tcpCliSock.send(data)  #發(fā)送數(shù)據(jù)
  if not data:
    break
  data = tcpCliSock.recv(BUFSIZ)  #接收數(shù)據(jù)
  print data
tcpCliSock.close()