Python Socket编程Python Socket编程

Python Socket编程

在使用Python做socket编程时,由于需要使用阻塞(默认)的方式来读取数据流,此时对于数据的结束每次都需要自己处理,太麻烦。并且网上也没找到太好的封装,所以就自己写了个简单的封装。

封装思路

  1. 客户端每次请求均发送一个 SocketRequest 对象,其中封装具体的数据,这里使用json。对于要发送的数据,会自动添加一个结束符标识(EOF = '0x00')。
  2. 服务器端接收数据时,根据结束符标识来生成完整的数据,并解包成 SocketRequest 对象。
  3. 服务器端根据 SocketRequest 的内容,来生成 SocketResponse 对象,这里使用了一个 SimpleRequestHandler 类来处理,例子中就是没有做任何处理,然后原样返回。
  4. 服务器端发送 SocketResponse 给客户端。其中也需要对包做一个封装,会自动添加一个结束符标识(EOF = '0x00')。
  5. 客户接收数据时,根据结束符标识来生成完整的数据,并解包成 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

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏任浩强的运维生涯

CentOS在线升级内核

升级内核需要使用 elrepo 的yum 源首先我们导入 elrepo 的key  rpm –import https://www.elrepo.o...

66670
来自专栏晓晨的专栏

Jenkins 集群搭建

Jenkins是当下比较流行的一款功能强大的持续集成工具,它支持搭建集群,来提高多项目的构建速度,模式为主从模式,master会将任务分配到各个从节点进行并发构...

17810
来自专栏Java成神之路

gradle_学习_01_gradle安装与基本使用

8820
来自专栏CloudZ的专栏

在Windows 10上使用Hyper-V创建VM

如果您运行的是Windows 10并且系统硬件支持Hyper-V,则可以创建一个独立的存储空间来部署自己的虚拟机并使用它。您可以同时创建一个或多个虚拟机并运行它...

27370
来自专栏贾鹏辉的技术专栏@CrazyCodeBoy

React Native应用部署/热更新-CodePush最新集成总结(新)

React Native应用部署/热更新-CodePush最新集成总结(新) ---- 更新说明: 此次博文更新适配了最新版的CodePush v1.17....

90260
来自专栏SpringBoot 核心技术

第一章:Maven环境下如何配置QueryDSL环境

24830
来自专栏抠抠空间

Django项目实战之用户头像上传与访问

1 将文件保存到服务器本地 upload.html <!DOCTYPE html> <html lang="en"> <head> <meta char...

56570
来自专栏晓晨的专栏

Jenkins 集群搭建

Jenkins是当下比较流行的一款功能强大的持续集成工具,它支持搭建集群,来提高多项目的构建速度,模式为主从模式,master会将任务分配到各个从节点进行并发构...

37050
来自专栏BIT泽清

React Native应用部署/马甲包热更新-CodePush最新集成总结(2018年最新)

React Native支持大家用React Native技术开发APP,并打包生成一个APP。在动态更新方面React Native只是提供了动态更新的基础,...

46100
来自专栏逸鹏说道

Ubuntu离线安装Sogou拼音(附老版本安装&输入法自启动)

跨平台系列汇总:http://www.cnblogs.com/dunitian/p/4822808.html#linux 离线安装的基础可以看看这篇文章的 前期...

36760

扫码关注云+社区

领取腾讯云代金券