本文實(shí)例講述了python基于socket實(shí)現(xiàn)的UDP及TCP通訊功能。分享給大家供大家參考，具體如下：

Server：

import socket
address = ('127.0.0.1', 31500)
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.bind(address)
while True:
  data, addr = s.recvfrom(2048)
  if not data:
    print "client has exist"
    break
  print "received:", data, "from", addr
s.close()

Client：

import socket
address = ('127.0.0.1', 31500)
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
while True:
  msg = raw_input()
  if not msg:
    break
  s.sendto(msg, address)
s.close()

運(yùn)行結(jié)果：

server

[work@db-testing python]$ python udp_ss.py
received: aaaaaa from ('127.0.0.1', 10169)
received: 123456 from ('127.0.0.1', 10169)
received: abcdef from ('127.0.0.1', 10169)

client

[work@db-testing python]$ python udp_cc.py
aaaaaa
123456
abcdef
[work@db-testing python]$

前言

python實(shí)在是太方便，太易用了，三方庫(kù)的支持又全，語(yǔ)法又簡(jiǎn)單，對(duì)開(kāi)發(fā)人員來(lái)說(shuō)，實(shí)在是居家旅行，XXXX的必備利器。周末研究了下python的網(wǎng)絡(luò)支持，做一下記錄，以備隨后翻閱。

python支持BSD的socket進(jìn)行網(wǎng)絡(luò)編程，其API跟C中的大同小異，先看看TCP方式的，說(shuō)到網(wǎng)絡(luò)編程，肯定會(huì)設(shè)計(jì)到server和client，分別來(lái)看下

TCP方式

server端

server端的socket一般流程是這樣：

1. 建立一個(gè)socket(可以選擇socket類(lèi)型INET,UNIX等，以及連接方式TCP/UDP)
2. 使用bind公開(kāi)一個(gè)端口，使得client可以方便連接
3. 設(shè)置一個(gè)listen隊(duì)列的大小
4. 進(jìn)入一個(gè)無(wú)限循環(huán)，在此無(wú)限循環(huán)中，使用accept函數(shù)來(lái)等待客戶連接，此函數(shù)返回一個(gè)新的socket, 對(duì)應(yīng)于客戶端的socket，建立通信信道。對(duì)socket的處理一般放在外部單獨(dú)的函數(shù)中(并發(fā))
5. 通過(guò)send()/recv()來(lái)對(duì)socket進(jìn)行讀寫(xiě)操作

好了，下面看一下例子：

netserver.py

#!/usr/bin/env python
# -*- coding:utf8 -*-
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
import socket
class NetServer(object):
  def tcpServer(self):
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.bind(('', 9527))    # 綁定同一個(gè)域名下的所有機(jī)器
    sock.listen(5)
    while True:
      clientSock, (remoteHost, remotePort) = sock.accept()
      print("[%s:%s] connect" % (remoteHost, remotePort))   # 接收客戶端的ip, port
      revcData = clientSock.recv(1024)
      sendDataLen = clientSock.send("this is send data from server")
      print "revcData: ", revcData
      print "sendDataLen: ", sendDataLen
      clientSock.close()
if __name__ == "__main__":
  netServer = NetServer()
  netServer.tcpServer()

client端

1. 建立一個(gè)新的socket
2. 使用connect函數(shù)與遠(yuǎn)程主機(jī)獲得連接
3. 在此socket上進(jìn)行I/O操作

netclient.py

#!/usr/bin/env python
# -*- coding:utf8 -*-
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
import socket
class NetClient(object):
  def tcpclient(self):
    clientSock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    clientSock.connect(('localhost', 9527))
    sendDataLen = clientSock.send("this is send data from client")
    recvData = clientSock.recv(1024)
    print "sendDataLen: ", sendDataLen
    print "recvData: ", recvData
    clientSock.close()
if __name__ == "__main__":
  netClient = NetClient()
  netClient.tcpclient()

運(yùn)行結(jié)果：

1） 先啟動(dòng) netserver.py， 然后啟動(dòng) netclient.py，注意client先發(fā)送后接收，server是先接收后發(fā)送，順序要交錯(cuò)

2） server 打印結(jié)果：

[127.0.0.1:55279] connect
revcData:  this is send data from client
sendDataLen:  30

3） client 打印結(jié)果：

sendDataLen:  29
recvData:  this is send  data from server

由代碼量可見(jiàn)python之簡(jiǎn)單，想要將這個(gè)例子改造成一個(gè)文件傳輸器，估計(jì)只需要增加不到20行代碼即可。

UDP方式

UDP號(hào)稱(chēng)無(wú)連接傳輸，全然沒(méi)有TCP那么復(fù)雜，三次握手，錯(cuò)誤重傳之類(lèi)的機(jī)制都沒(méi)有，發(fā)的只管發(fā)，收得只管收，收到?jīng)]有？不知道，順序不對(duì)怎么辦？不管！就是這樣，但是速度就要比TCP高得多了。在對(duì)數(shù)據(jù)幀要求不是很高的地方，這確實(shí)是很好用的，比如網(wǎng)絡(luò)上的視頻傳輸，音頻傳輸?shù)取?/p>

server端

1. 建立數(shù)據(jù)報(bào)形式的socket
2. 公開(kāi)一個(gè)端口，一邊客戶端連接
3. 開(kāi)始接收數(shù)據(jù)

udpserver.py

#!/usr/bin/env python
# -*- coding:utf8 -*-
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
import socket
class UdpServer(object):
  def tcpServer(self):
    sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sock.bind(('', 9527))    # 綁定同一個(gè)域名下的所有機(jī)器
    while True:
      revcData, (remoteHost, remotePort) = sock.recvfrom(1024)
      print("[%s:%s] connect" % (remoteHost, remotePort))   # 接收客戶端的ip, port
      sendDataLen = sock.sendto("this is send data from server", (remoteHost, remotePort))
      print "revcData: ", revcData
      print "sendDataLen: ", sendDataLen
    sock.close()
if __name__ == "__main__":
  udpServer = UdpServer()
  udpServer.tcpServer()

需要注意的是，server中address元組中的引號(hào)表示可以接受任何地址來(lái)的數(shù)據(jù)報(bào)，TCP例子中的則表示可以接受任意地址發(fā)起的連接。

client端

1. 新建一個(gè)數(shù)據(jù)報(bào)socket
2. 收發(fā)數(shù)據(jù)

udpclient.py

#!/usr/bin/env python
# -*- coding:utf8 -*-
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
import socket
class UdpClient(object):
  def tcpclient(self):
    clientSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    sendDataLen = clientSock.sendto("this is send data from client", ('localhost', 9527))
    recvData = clientSock.recvfrom(1024)
    print "sendDataLen: ", sendDataLen
    print "recvData: ", recvData
    clientSock.close()
if __name__ == "__main__":
  udpClient = UdpClient()
  udpClient.tcpclient()

運(yùn)行結(jié)果：

1） 先啟動(dòng) udpserver.py，然后啟動(dòng) udpclient.py，并且保證client先發(fā)后收，server先收后發(fā)

2） server 結(jié)果（client 連續(xù)發(fā)送了三次請(qǐng)求）：

[127.0.0.1:37219] connect
revcData:  this is send data from client
sendDataLen:  30
[127.0.0.1:57622] connect
revcData:  this is send data from client
sendDataLen:  30
[127.0.0.1:52290] connect
revcData:  this is send data from client
sendDataLen:  30

3） client 結(jié)果：

sendDataLen:  29
recvData:  ('this is send  data from server', ('127.0.0.1', 9527))

文中給出的例子已經(jīng)具有網(wǎng)絡(luò)程序的基本結(jié)構(gòu)，在實(shí)際應(yīng)用中，應(yīng)該自行擴(kuò)展對(duì)socket的處理，socket說(shuō)到底跟文件句柄，文件描述符，管道描述符等都是一個(gè)概念，可以對(duì)其進(jìn)行I/O處理，事實(shí)上，在UNIX類(lèi)系統(tǒng)中，對(duì)socket的操作跟對(duì)文件句柄的操作沒(méi)有任何形式上的差異。

更多關(guān)于Python相關(guān)內(nèi)容可查看本站專(zhuān)題：《Python Socket編程技巧總結(jié)》、《Python數(shù)據(jù)結(jié)構(gòu)與算法教程》、《Python函數(shù)使用技巧總結(jié)》、《Python字符串操作技巧匯總》、《Python入門(mén)與進(jìn)階經(jīng)典教程》及《Python文件與目錄操作技巧匯總》

希望本文所述對(duì)大家Python程序設(shè)計(jì)有所幫助。