1. 程式人生 > 程式設計 >Python socket服務常用操作程式碼例項

Python socket服務常用操作程式碼例項

套接字(socket)是一個抽象層,應用程式可以通過它傳送或接收資料,可對其進行像對檔案一樣的開啟、讀寫和關閉等操作。

1. 實現客戶端傳送字元,伺服器返回大寫的字元:

伺服器:

import socketserver

class MyTCPHandler(socketserver.BaseRequestHandler):     # 通過類的繼承,實現
  def handle(self):          # 重寫父類的handle方法,所有的操作都在此方法中
    while True:           # 迴圈,不停的接收從客戶端來的資料
      try:
        self.data = self.request.recv(1024).strip()     # 從客戶端接收資料,每次收1024位元組
        print("{} send:".format(self.client_address),self.data)
        self.request.send(self.data.upper())         # 從伺服器傳送給客戶端資料
      except ConnectionResetError as e:
        print('Error: ',e)
        break

if __name__ == '__main__':
  host,port = 'localhost',9999
  server = socketserver.ThreadingTCPServer((host,port),MyTCPHandler)   # 通過多執行緒實現多個客戶端連線,每個客戶端連線都是一個執行緒
  server.serve_forever()                        # 一直執行服務

客戶端:

import socket

client = socket.socket()         # socket物件
client.connect(('localhost',9999))   # 連線伺服器地址和埠

while True:               # 迴圈,不停的輸入傳送資料
  con = input('>>>:').strip()
  if len(con) ==0: continue      # 不能傳送空資料,否則會阻塞
  client.send(con.encode('utf-8'))  # 傳送資料,必須是二進位制的
  data = client.recv(1024)      # 接收伺服器返回的資料
  print(data.decode())        # 列印 解碼後的資料

client.close()           # 關閉

2. 通過socket執行伺服器命令:

用法:直接在客戶端輸入處輸入命令如:ipconfig

伺服器:

import socket
import os
import threading

def tcplink(sock,addr):
  print('Accept new connection from %s:%s...' % addr)
  while True:     # 和每個接入的客戶端,進行多次資料通訊
    data = sock.recv(1024) # 接收客戶端資料
    if not data or data.decode('utf-8') == 'exit': # 如果客戶端不傳送資料或者傳送了exit
      print('client disconnected.')
      break
    content = os.popen(data.decode('utf-8')).read() # 對傳送來的資料執行cmd命令,獲取結果
    if len(content) == 0:      #如果執行的命令結果為空的,就手動造一個結果。因為如果為空資料,會掛起,無法正常傳送。
      content = 'cmd not exists.'
    sock.send(str(len(content.encode('utf-8'))).encode('utf-8')) # 傳送資料的長度
    print('send length:',(len(content.encode('utf-8'))))
    # print('content,',content.encode('utf-8'))
    recv = sock.recv(1024) # 因為上下都有一個send連在一起,可能發生粘包現象,為了防止這種情況,可以讓客戶端重新應答一下
    print('Answer:',recv.decode('utf-8'))
    sock.send(content.encode('utf-8'))    # 傳送資料
    print('send finished.')
  sock.close()
  print('Connection from %s:%s closed.' % addr)


s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
# 監聽埠:
s.bind(('127.0.0.1',9999))
s.listen(3)
print('Waiting for connection...')

while True:
  # 接受一個新連線:
  sock,addr = s.accept()
  # 建立新執行緒來處理TCP連線:
  t = threading.Thread(target=tcplink,args=(sock,addr))
  t.start()

客戶端:

import socket

# AF_INET 代表ipv4,SOCK_STREAM 代表TCP
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)   # 確定網路協議,生成物件
s.connect(('127.0.0.1',9999)) # 連線伺服器的地址和埠,元組的形式。
while True:
  msg = input('>>:').strip()
  if len(msg) != 0:         # 如果訊息為空,會一直掛起,所以不能為空
    if msg =='exit':
      s.close()           # 關閉連線
      print('Connection closed.')
      break
    s.send(msg.encode('utf-8'))    # 給伺服器傳送資料,必須是二進位制的
    length = s.recv(1024)      # 首先接收伺服器返回的將要接收的資料的長度資訊。
    s.send(b'Ready to receive...')  # 傳送接收命令
    length = int(length.decode('utf-8'))
    print('receive len:',length)
    data_len = 0
    data_recv = b''
    while data_len < length:  # 已經接收的資訊的長度,如果小於總長度
      data = s.recv(1024)   # 從伺服器接收資料
      data_recv += data
      data_len += len(data)
    print(data_recv.decode('utf-8')) # 列印返回的資料。

3. 通過socket傳輸檔案:

用法:get 檔名

伺服器:

import socket
import os
import hashlib
import threading

def tcplink(sock,addr):
  print('Accept new connection from %s:%s...' % addr)
  while True:     # 和每個接入的客戶端,進行多次資料通訊
    data = sock.recv(1024) # 接收客戶端資料
    if not data or data.decode('utf-8') == 'exit': # 如果客戶端不傳送資料或者傳送了exit
      print('client disconnected.')
      break
    oper,filename = data.decode('utf-8').split() # 對接收的資料按照空格分割
    if oper == 'get':
      m = hashlib.md5()
      if os.path.isfile(filename):
        size = os.stat(filename).st_size  # 獲取檔案大小
        print('Send size:',size)
        sock.send(str(size).encode('utf-8')) # 傳送檔案大小
        recv = sock.recv(1024)       # 接收客戶端確認資訊(因為上下文兩個send是連著的,所以為了防止粘包,接收一次資訊)
        f = open(filename,'rb')
        for line in f:
          sock.send(line)  #讀取檔案,傳送給客戶端
          m.update(line)
        # print('Send finished.',m.hexdigest())  # 列印md5的值
        sock.send(m.hexdigest().encode('utf-8')) # 把md5的值傳送給客戶端
  sock.close()
  print('Connection from %s:%s closed.' % addr)
s = socket.socket(socket.AF_INET,addr))
  t.start()

客戶端:

import socket
import hashlib

# AF_INET 代表ipv4,SOCK_STREAM 代表TCP
s = socket.socket(socket.AF_INET,9999)) # 連線伺服器的地址和埠,元組的形式。
while True:
  msg = input('>>:').strip()
  if len(msg) != 0:         # 如果訊息為空,會一直掛起,所以不能為空
    if msg =='exit':
      s.close()           # 關閉連線
      print('Connection closed.')
      break
    s.send(msg.encode('utf-8'))    # 給伺服器傳送資料,必須是二進位制的
    length = s.recv(1024)      # 首先接收伺服器返回的將要接收的資料的長度資訊。
    s.send(b'Ready to receive...')  # 傳送接收確認命令
    length = int(length.decode('utf-8'))
    print('Recv size:',length)
    data_len = 0
    data_recv = b''
    # 新檔名
    fileName = msg.split()[-1].split('.')[0]
    fileExt = msg.split()[-1].split('.')[-1]
    newFile = fileName+'-1.'+fileExt
    f = open(newFile,'wb') # 開啟檔案,準備寫入伺服器發過來的檔案
    m = hashlib.md5()
    while data_len < length:  # 已經接收的資訊的長度,如果小於總長度
      size = length - data_len 
      if size > 1024:   # 如果剩下的資訊長度大於1024,即不能一次性發完。
        size = 1024
      else:    # 如果能一次性發完,就只收剩下的資訊。目的是準確的接收檔案的大小,把可能粘連的send的資料留給下一次recv
        size = length-data_len
      data = s.recv(size)   # 從伺服器接收資料
      f.write(data)
      m.update(data)
      data_len += len(data)
    f.close()
    print('recv_md5:',m.hexdigest()) # 列印返回的資料。
    recv = s.recv(1024)   # 接收下一次send的資料,即md5的值。
    print('orig_md5:',recv.decode())

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。