Python Socket編程詳細介紹

更新時間：2017年03月23日 17:10:11 作者：kongxx

這篇文章主要介紹了Python Socket編程詳細介紹，socket可以建立連接，傳遞數(shù)據(jù)，具有一定的參考價值，感興趣的小伙伴們可以參考一下。

在使用Python做socket編程時，由于需要使用阻塞（默認）的方式來讀取數(shù)據(jù)流，此時對于數(shù)據(jù)的結(jié)束每次都需要自己處理，太麻煩。并且網(wǎng)上也沒找到太好的封裝，所以就自己寫了個簡單的封裝。

封裝思路

1. 客戶端每次請求均發(fā)送一個 SocketRequest 對象，其中封裝具體的數(shù)據(jù)，這里使用json。對于要發(fā)送的數(shù)據(jù)，會自動添加一個結(jié)束符標識（EOF = ‘0x00'）。

2. 服務器端接收數(shù)據(jù)時，根據(jù)結(jié)束符標識來生成完整的數(shù)據(jù)，并解包成 SocketRequest 對象。

3. 服務器端根據(jù) SocketRequest 的內(nèi)容，來生成 SocketResponse 對象，這里使用了一個 SimpleRequestHandler 類來處理，例子中就是沒有做任何處理，然后原樣返回。

4. 服務器端發(fā)送 SocketResponse 給客戶端。其中也需要對包做一個封裝，會自動添加一個結(jié)束符標識（EOF = ‘0x00'）。

5. 客戶接收數(shù)據(jù)時，根據(jù)結(jié)束符標識來生成完整的數(shù)據(jù)，并解包成 SocketResponse 對象，然后返回。

封裝類

sockets.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import socket
import pickle
import thread


PORT = 12345
EOF = '0x00'


class SocketServer(object):

  def __init__(self, port=None):
    self.port = port

  def startup(self):
    sock_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock_server.bind(('0.0.0.0', self.port))
    sock_server.listen(0)

    while True:
      sock, address = sock_server.accept()
      thread.start_new_thread(self.__invoke, (sock, address))

  def shutdown(self):
    pass

  def __invoke(self, sock, address):
    try:
      full_data = ''
      while True:
        data = sock.recv(1024)
        if data is None:
          return

        full_data += data
        if full_data.endswith(EOF):
          full_data = full_data[0:len(full_data) - len(EOF)]
          request = pickle.loads(full_data)
          response = SimpleRequestHandler().handle(request)
          sock.sendall(pickle.dumps(response) + EOF)
          return
    except Exception as e:
      print e
    finally:
      sock.close()


class SocketClient(object):

  def __init__(self, host, port):
    self.host = host
    self.port = port

  def execute(self, request):
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    sock.connect((self.host, self.port))

    try:
      sock.sendall(pickle.dumps(request) + EOF)
      full_data = ''
      while True:
        data = sock.recv(1024)
        if data:
          full_data += data
          if full_data.endswith(EOF):
            full_data = full_data[0:len(full_data) - len(EOF)]
            response = pickle.loads(full_data)
            return response
        else:
          return None
    except Exception as e:
      print e
      return None
    finally:
      sock.close()


class SocketRequest(object):
  def __init__(self, data):
    self.data = data

  def __repr__(self):
    return repr(self.__dict__)


class SocketResponse(object):
  def __init__(self, data):
    self.data = data

  def __repr__(self):
    return repr(self.__dict__)


class SimpleRequestHandler(object):
  def __init__(self):
    pass

  def __repr__(self):
    return repr(self.__dict__)

  def handle(self, request):
    return SocketResponse(request.data)

測試

socket_server.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from agent.sockets import *

ss = SocketServer(PORT)
ss.startup()

socket_client.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import pickle
from agent.sockets import *


sc = SocketClient('localhost', PORT)
request = SocketRequest('abc')
response = sc.execute(request)
print request
print response

運行測試

首先，運行 socket_server.py

然后，運行 socket_client.py

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: