python使用flask实现自动根据url寻找对应目录/文件/方法,实现动态路由

例如访问:/user/index/index_config

则自动访问/user 目录里 index.py文件里的 index_config 方法


实现代码

python 复制代码
from flask import Flask,jsonify

app = Flask(__name__)

def reg_func(code, data, msg, debug=None, show=False):
    return jsonify({
        'code': code,
        'data': data,
        'msg': msg,
        'time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
        'debug': debug,
        'show': show

    }), code



@app.route('/<path:path>', methods=['GET', 'POST'])
def dynamic_route(path):
    try:
        # 将路径分割成模块路径和函数名
        parts = path.split('/')
        module_path = '.'.join(parts[:-1])
        if len(parts) < 3:
            return reg_func(404, None, '真的没有这个接口啊')
        user_or_admin = parts[-3]
        py_name = parts[-2]
        function_name = parts[-1]
        if not user_or_admin or not py_name or not function_name:
            return reg_func(404, None, '真的没有这个接口啊')

        # 动态导入模块
        module = importlib.import_module(module_path)
        # 获取函数
        func = getattr(module, function_name)
        # 调用函数
        return func()
    except (ModuleNotFoundError, AttributeError) as e:
        return reg_func(404, None, '可能,这是个错误')

if __name__ == '__main__':
    # 确保Python能找到你的模块
    os.environ['PYTHONPATH'] = os.path.dirname(os.path.abspath(__file__))
    app.run(debug=True, port=8880, host='0.0.0.0')

代码详解

chatgpt协助整理

代码含义

这段代码实现了一个动态路由的 Flask 应用,它根据 URL 路径动态加载和调用模块中的函数,并返回相应的 JSON 响应。

  1. 导入模块:

    python 复制代码
    from flask import Flask, jsonify
    import importlib
    import os
    from datetime import datetime
    • Flask:Flask 框架的核心类。
    • jsonify:用于将 Python 数据结构转换为 JSON 响应。
    • importlib:用于动态导入模块。
    • os:用于设置环境变量。
    • datetime:用于获取当前时间。
  2. 创建 Flask 应用实例:

    python 复制代码
    app = Flask(__name__)
  3. 定义 reg_func 函数:

    python 复制代码
    def reg_func(code, data, msg, debug=None, show=False):
        return jsonify({
            'code': code,
            'data': data,
            'msg': msg,
            'time': datetime.now().strftime('%Y-%m-%d %H:%M:%S'),
            'debug': debug,
            'show': show
        }), code
    • 该函数生成一个标准的 JSON 响应,其中包含状态码、数据、消息、时间戳、调试信息和展示标志。
    • 返回的响应和状态码用于向客户端提供统一的接口响应格式。
  4. 定义动态路由 dynamic_route:

    python 复制代码
    @app.route('/<path:path>', methods=['GET', 'POST'])
    def dynamic_route(path):
        try:
            parts = path.split('/')
            module_path = '.'.join(parts[:-1])
            if len(parts) < 3:
                return reg_func(404, None, '真的没有这个接口啊')
            user_or_admin = parts[-3]
            py_name = parts[-2]
            function_name = parts[-1]
            if not user_or_admin or not py_name or not function_name:
                return reg_func(404, None, '真的没有这个接口啊')
    
            module = importlib.import_module(module_path)
            func = getattr(module, function_name)
            return func()
        except (ModuleNotFoundError, AttributeError) as e:
            return reg_func(404, None, '可能,这是个错误')
    • 解析 URL 路径,将其拆分成模块路径、模块名称和函数名称。
    • 动态导入指定的模块,并获取模块中的函数。
    • 调用函数并返回其结果。如果模块或函数不存在,返回 404 错误。
  5. 运行应用:

    python 复制代码
    if __name__ == '__main__':
        os.environ['PYTHONPATH'] = os.path.dirname(os.path.abspath(__file__))
        app.run(debug=True, port=8880, host='0.0.0.0')
    • 设置 PYTHONPATH 确保 Python 可以找到模块。
    • 启动 Flask 应用,监听所有 IP 地址的 8880 端口,并启用调试模式。

总结

这段代码定义了一个 Flask 应用,它能够根据请求的 URL 路径动态导入和调用 Python 模块中的函数。通过 reg_func 函数,应用提供了一个统一的 JSON 响应格式,包含了状态码、响应数据、消息和调试信息。动态路由允许根据路径自动处理不同的模块和函数调用,实现灵活的接口处理。

温馨提示

使用这种方式 有一定的安全隐患,

可以考虑在被调用的地方加入拦截

或者在调用之前 进行白名单管理

相关推荐
星夜离尘5613 小时前
MATLAB 工程实验:不等间隔流量积分,先把缺口与覆盖率说清楚
开发语言·matlab·时间戳·数据质量·数值积分
happylifetree13 小时前
Python70:核心语法-函数进阶-案例2
python
Ai-_Man13 小时前
Meta Muse还是不可以导出左侧所有对话,只能点进去单条内容导出,方法不对吗?
开发语言·人工智能·ai·小程序·c#·1024程序员节
梓豪csd13 小时前
消息订阅中心:谁该收到哪条通知
java·开发语言·人工智能·低代码·自动化·1024程序员节
王中阳Go13 小时前
两个入口答案不一致先别调模型:6 行入口口径对账表(附脚本)
人工智能·后端·python·大语言模型·1024程序员节
BK贝壳13 小时前
bool-hybrid-array 落盘我只认 .bha:用 Create_BHA / Ask_BHA 做布尔数组序列化
数据结构·python·numpy·1024程序员节
学逆向的14 小时前
C 枚举变量
c语言·开发语言·网络安全
Sakura_j_h_14 小时前
Python 在定时任务里输出 emoji,UnicodeEncodeError 从哪来
开发语言·字符编码·windows·python·定时任务
supabc12315 小时前
欧盟电子发票(XRechnung/Peppol/ZUGFeRD)踩坑记:用 Python 不装 Java 跑官方校验规则
xml·python·开源·电子发票·1024程序员节
ZhengEnCi15 小时前
AI Agent工具调用防护层怎么写-一份能直接跑的Python四层拦截实现
python·llm·agent