1. 程式人生 > 程式設計 >Python實現TCP通訊的示例程式碼

Python實現TCP通訊的示例程式碼

使用socket實現tcp通訊,需匯入socket模組

1、服務端

主要步驟:

(1)建立socket:socket.socket(family=AF_INET,type=SOCK_STREAM,proto=0,fileno=None)

family:AF_INET,AF_INET6,AF_UNIX,AF_CAN,AF_PACKET,AF_RDS

type:SOCK_STREAM,SOCK_DGRAM,SOCK_RAW,或者其他SOCK_*

proto:一般使用預設值0

fileno:一般使用預設值

(2)繫結IP和埠號:socket.bind(address)

address:使用(host,port)元組即可

(3)監聽:socket.listen([backlog])

backlog:就理解為最大連線數

(4)接受連線:socket.accept()

返回(conn,address)元組,分別是已連線客戶端的socket物件和地址

(5)傳送/接受資料:socket.recv(bufsize[,flags])/socket.send(bytes[,flags])

bufsize:接收的資料最大位元組數

bytes:需要傳送的資料位元組數

(6)關閉連線:socket.close()

#!/usr/bin/python
# -*- coding: UTF-8 -*-
# 檔名:tcpserver.py
 
import socket
import time
MaxBytes=1024*1024
 
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server.settimeout(60)
host = '127.0.0.1'
#host = socket.gethostname()
port = 11223
server.bind((host,port))    # 繫結埠
 
server.listen(1)           # 監聽
try:
  client,addr = server.accept()     # 等待客戶端連線
  print(addr," 連線上了")
  while True:
    data = client.recv(MaxBytes)
    if not data:
      print('資料為空,我要退出了')
      break
    localTime = time.asctime( time.localtime(time.time()))
    print(localTime,' 接收到資料位元組數:',len(data))
    print(data.decode())
    client.send(data)
except BaseException as e:
  print("出現異常:")
  print(repr(e))
finally:
  server.close()          # 關閉連線
  print("我已經退出了,後會無期")

這個例項沒有建立其他執行緒,也只接收一個客戶端的連線;為了能手動結束或自動結束accept和recv的阻塞,設定了超時60秒,並將主體部分放入到try語句中;將socket的關閉放到finally塊中,可以防止程式異常退出,而導致沒有呼叫close函式。

2、客戶端

主要步驟:

(1)建立socket:socket.socket(family=AF_INET,fileno=None)

(2)連線:socket.connect(address)

address:使用(host,port)元組即可

(3)傳送/接受資料:socket.recv(bufsize[,flags])

(4)關閉連線:socket.close()

#!/usr/bin/python
# -*- coding: UTF-8 -*-
# 檔名:tcpclient.py
 
import socket
import time
MaxBytes=1024*1024
host ='127.0.0.1'
port = 11223
client = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
client.settimeout(30)
client.connect((host,port))
while True:
  inputData=input();     #等待輸入資料
  if(inputData=="quit"):
    print("我要退出了,再見")
    break
  sendBytes = client.send(inputData.encode())
  if sendBytes<=0:
    break;
  recvData = client.recv(MaxBytes)
  if not recvData:
    print('接收資料為空,我要退出了')
    break
  localTime = time.asctime( time.localtime(time.time()))
  print(localTime,len(recvData))
  print(recvData.decode())
client.close()
print("我已經退出了,後會無期")

以上客戶端連線成功後等待使用者輸入,然後將輸入資訊傳送到服務端,服務端將資料原樣返回給客戶端,如果客戶端輸入quit就會跳出迴圈並關閉socket,這時服務端recv會返回空從而跳出迴圈並關閉socket。希望對大家的學習有所幫助,也希望大家多多支援我們。