python 动态获取对象文件所有对象

2019-08-13  本文已影响0人  py晓枫

上resource代码

from utils.tools import create_route

class Base(Resource):
    def __init__(self):
        pass

class FrontShop(Base):
    def get(self):
        parse = reqparse.RequestParser()
        parse.add_argument('id', type=int)
        return {'a': 'b'}

class FrontText(Base):
    def get(self):
        return {'a': 'db'}
#这个就是创建路由的函数,下面会详细展示。
ROUTES = create_route(__name__, __file__)

上tools代码

import os, re, importlib
def create_route(source_name, fsource_ile):
    #获取文件目录
    filename_source = os.path.dirname(fsource_ile)
    #获取文件名称
    name = source_name.split('.')[-1]
    #拼接文件路径
    filename = '%s\%s.py' % (filename_source, name)

    #打开文件
    with open(filename, 'r') as file:
        #定义一个返会列表
        return_data = []
        #遍历文件
        while True:
            text = file.readline()
            #正则匹配对象文件获取对象的名称
            regex = re.search(r'^class\s([a-zA-Z]*?)\(Base\):', text, re.M | re.I)
            #判断是否匹配
            if regex:
                #获取对象名称
                cls = regex.group(1)
                #添加对象到返回参数
                return_data.append(
                    {
                        'url': '/%s/' % cls,
                        'func': getattr(importlib.import_module(source_name), cls),
                        'endpoint': cls.lower()
                    }
                )
            #判断退出
            if not text:
                break

    return return_data

最后进行遍历生成route

from .resources import ROUTES
from exts import api
print(ROUTES)
for v in ROUTES:
    api.add_resource(v['func'], v['url'], endpoint=v['endpoint'])

此方法可能不完美。有好的方法麻烦分享一下。共同进步!
完结。

上一篇下一篇

猜你喜欢

热点阅读