从 0 到 1,如何徒手撸个 Python 插件系统?

简介:

号主从事算法服务开发多年,临近三月的尾巴,输出一个插件化部署算法服务的解决方案。篇幅内容都经过生产实践,如果对各位号友有所帮助,请拽个三连:点赞、关注和转发

插件机制背景

插件化机制使框架业务模块的实现相解耦,框架服务抽象出统一的交互接口,业务模块只要符合交互标准即可做到插件替换。

微内核架构是插件架构模式的一种典型实现,常常把微内核架构也叫做插件式架构。
当前微内核架构也被应用在许多我们熟知的产品,比如:操作系统、Chrome浏览器、Eclipse编辑器等

微内核架构包含两个组件:核心系统(core system)和插件模块(plugin component)

  • 核心系统只包含让系统可以运作的最小功能
  • 插件模块则包含一些特殊处理逻辑、额外的功能,用于提供更多的业务能力

具体如下图:

Python 插件核心

__import__()函数是我们本篇的核心,具体实现都围绕这个函数定制展开。

作用:用于动态加载类和函数。如果一个模块经常变化就可以使用 __import__() 来动态载入。
语法:

__import__(name, globals=None, locals=None, fromlist=(), level=0)
    name[必须]:模块名称
    globals - 全局变量集合,默认为None
    locals - 局部变量集合,默认为None
    fromlist - 是否导入子模块,看上去是导入模块的列表。但实际是一个判断条件,只要设置为非空的值,且模块名称是带有子模块的,将导入子模块。例如:sys.path。当不设置时,返回sys,如果设置为非空值,则返回ntpath(path模块)
    level - 绝对或者相对导入

此处我们做个简单的验证,通过__import__实现和import一样的导入能力。分别验证导入os.path和导入其对应的子模块,具体如下:

>>> __import__('os.path')
<module 'os' from '/opt/anaconda3/envs/python38/lib/python3.8/os.py'>
>>> __import__('os.path', fromlist=['None'])
<module 'posixpath' from '/opt/anaconda3/envs/python38/lib/python3.8/posixpath.py'>
>>> __import__('os.path', fromlist=['path'])
<module 'posixpath' from '/opt/anaconda3/envs/python38/lib/python3.8/posixpath.py'>
>>>

以上只是在 shell 里面进行验证,那 Python 代码里面该如何写呢?有两种方式:
直接引用加载

p = __import__('widget.demo.infer', fromlist='infer')  # 导入
print(p.Demo)      # 取出类变量
# 打印结果
<class 'widget.demo.infer.Demo'>

Importlib包方式

import importlib    # 导入工具包
demo = importlib.import_module('.demo.infer', package='widget')
print(demo)
# 打印结果
<module 'widget.demo.infer' from '/Users/**/language-bootcamp/python/plugin_system/widget/demo/infer.py'>
print(demo.Demo)
# 打印结果
<class 'widget.demo.infer.Demo'>

插件服务基本组成

PluginCore:通过Plugin Manager调用算法,负责业务逻辑的实现
PluginManger: 通过读取配置文件,负责各种插件的加载、管理、甚至热更新

此处,Python 插件相关的知识点已经讲完了,下面进行算法插件化部署相关内容,Go Go Go!

算法服务基本组成

如上,一个完整的算法服务包含三部分:API逻辑算法逻辑模型文件

API逻辑:服务相关逻辑,比如:HTTP 相关的请求/响应设置或 gRPC 远程交互约定等。
算法逻辑:算法相关逻辑,比如:数据的前置预处理、后置标签映射等。
模型文件:算法运行依赖项。

可能某些号友会发问:一定是三个部分,难道不能两个,甚至一个?
此处只是从纯业务功能划分,便于理解。具体代码实现,都可以。

插件管理器定义

定义了插件管理的方法,插件功能加载、插件获取方法实现

class BaseManager(object):
    """插件管理基类"""
    
    def __init__(self) -> None:
        """加载插件配置文件"""
        pass

    def get_plugin(self, name):
        """根据调用,返回插件实例"""
        raise RuntimeError('not implemented yet!')

    def get_total_plugin(self):
        """返回所有插件"""
        raise RuntimeError('not implemented yet!')

算法插件定义

每个插件需要实现公共接口方法。

class ApiBase(object):
    """API服务基类"""
    
    def __init__(self) -> None:
        "模型初始化加载"
    
    def predict(self, req_dict={}):
        """
        预测方法,推理算法入口
        :param req_dict:        请求参数,词典类型,eg:{"name":"value"}
        :return:                返回结果,词典类型,eg:{"name":"value"}
        """
        raise RuntimeError('not implemented \'predict\' method')

    def version(self):
        "版本方法"
        return '20220330'

配置文件定义

PluginManager通过配置文件进行可选插件指定加载配置,包括:插件名 、插件路径等信息。

{
    "demo":{
        "infer_dir":"widget.demo.infer"
    },
    "digit_recognition":{
        "infer_dir":"widget.digit_recognition.infer"
    }
}

插件扩展

有些时候为了解耦,需要调用 C 代码,以 linux 平台为例,我们讲讲 python 如何调用 .so 文件进行扩展。

# -*- coding: utf-8 -*-
import json
import ctypes
from ctypes import c_float
from common.logger import LOG

class C_Type(object):
    def __init__(self) -> None:
        self.so = ctypes.CDLL('widget/ctype/lib/sum.so')
    
    def predict(self, req_dict={}):
        LOG.info("this is ctype infer, req_dict=%s" % json.dumps(req_dict))
        return req_dict

    def version(self):
        return '20220326'

def get_plugin_class():
    return C_Type

总结一下

高可扩展:通过插件机制,我们可以很方便进行服务扩展
部署灵活:也能通过配置发布单独服务或者多个服务
代码解耦:最主要能实现各个功能模块的开发解耦,方便进行业务定制

但插件的使用得注意,尽量选择依赖环境和处理性能基本一致的业务进行插件构建,否则会有性能问题,慢插件影响快插件的推理速度。

好了,以上就是今天的全部内容,以上所有代码都汇总到github.com/codetodo-io/language-bootcamp,号友们可以下载演示。

❤️❤️❤️读者每一份热爱都是笔者前进的动力!
我是三十一,感谢各位朋友:求点赞、求评论、求转发,大家下期见!

相关文章
|
18天前
|
机器学习/深度学习 数据采集 供应链
使用Python实现智能食品安全追溯系统的深度学习模型
使用Python实现智能食品安全追溯系统的深度学习模型
41 4
|
7天前
|
机器学习/深度学习 人工智能 算法
【宠物识别系统】Python+卷积神经网络算法+深度学习+人工智能+TensorFlow+图像识别
宠物识别系统,本系统使用Python作为主要开发语言,基于TensorFlow搭建卷积神经网络算法,并收集了37种常见的猫狗宠物种类数据集【'阿比西尼亚猫(Abyssinian)', '孟加拉猫(Bengal)', '暹罗猫(Birman)', '孟买猫(Bombay)', '英国短毛猫(British Shorthair)', '埃及猫(Egyptian Mau)', '缅因猫(Maine Coon)', '波斯猫(Persian)', '布偶猫(Ragdoll)', '俄罗斯蓝猫(Russian Blue)', '暹罗猫(Siamese)', '斯芬克斯猫(Sphynx)', '美国斗牛犬
64 29
【宠物识别系统】Python+卷积神经网络算法+深度学习+人工智能+TensorFlow+图像识别
|
9天前
|
机器学习/深度学习 算法 前端开发
基于Python深度学习的果蔬识别系统实现
果蔬识别系统,主要开发语言为Python,基于TensorFlow搭建ResNet卷积神经网络算法模型,通过对12种常见的果蔬('土豆', '圣女果', '大白菜', '大葱', '梨', '胡萝卜', '芒果', '苹果', '西红柿', '韭菜', '香蕉', '黄瓜')图像数据集进行训练,最后得到一个识别精度较高的模型文件。再基于Django框架搭建Web网页端可视化操作界面,以下为项目实现介绍。
19 4
基于Python深度学习的果蔬识别系统实现
|
2月前
|
机器学习/深度学习 传感器 存储
使用 Python 实现智能地震预警系统
使用 Python 实现智能地震预警系统
131 61
|
1月前
|
弹性计算 数据管理 数据库
从零开始构建员工管理系统:Python与SQLite3的完美结合
本文介绍如何使用Python和Tkinter构建一个图形界面的员工管理系统(EMS)。系统包括数据库设计、核心功能实现和图形用户界面创建。主要功能有查询、添加、删除员工信息及统计员工数量。通过本文,你将学会如何结合SQLite数据库进行数据管理,并使用Tkinter创建友好的用户界面。
52 2
从零开始构建员工管理系统:Python与SQLite3的完美结合
|
21天前
|
Python
Python之音乐专辑管理系统
音乐专辑管理系统是一款用于管理和维护音乐专辑信息的应用程序,支持添加、删除、修改和查询专辑详情(如专辑名、艺术家、发行日期及曲目列表)。系统运行需Python 3.x环境,硬件要求较低,适合个人及小型团队使用。
41 4
|
23天前
|
Python
Python实现摇号系统
本文介绍了如何使用Python构建一个简单的摇号系统,包括用户输入、随机抽取、结果展示和日志记录等功能。通过导入`random`、`datetime`和`logging`模块,实现了从参与者名单中随机抽取中奖者,并记录每次摇号的结果,方便后续查看和审计。完整代码示例提供了从功能实现到主程序调用的全过程。
33 2
|
25天前
|
机器学习/深度学习 人工智能 算法
基于Python深度学习的【垃圾识别系统】实现~TensorFlow+人工智能+算法网络
垃圾识别分类系统。本系统采用Python作为主要编程语言,通过收集了5种常见的垃圾数据集('塑料', '玻璃', '纸张', '纸板', '金属'),然后基于TensorFlow搭建卷积神经网络算法模型,通过对图像数据集进行多轮迭代训练,最后得到一个识别精度较高的模型文件。然后使用Django搭建Web网页端可视化操作界面,实现用户在网页端上传一张垃圾图片识别其名称。
75 0
基于Python深度学习的【垃圾识别系统】实现~TensorFlow+人工智能+算法网络
|
25天前
|
机器学习/深度学习 人工智能 算法
基于深度学习的【蔬菜识别】系统实现~Python+人工智能+TensorFlow+算法模型
蔬菜识别系统,本系统使用Python作为主要编程语言,通过收集了8种常见的蔬菜图像数据集('土豆', '大白菜', '大葱', '莲藕', '菠菜', '西红柿', '韭菜', '黄瓜'),然后基于TensorFlow搭建卷积神经网络算法模型,通过多轮迭代训练最后得到一个识别精度较高的模型文件。在使用Django开发web网页端操作界面,实现用户上传一张蔬菜图片识别其名称。
73 0
基于深度学习的【蔬菜识别】系统实现~Python+人工智能+TensorFlow+算法模型
|
1月前
|
机器学习/深度学习 数据采集 存储
使用Python实现智能农业灌溉系统的深度学习模型
使用Python实现智能农业灌溉系统的深度学习模型
160 6