import os
import importlib
import inspect
from stat import *
class Route():
# 构造函数,传入app,项目路径,应用名称
def __init__(self, app, project_path, route_dir, file_dir):
self.__app = app # flask 的app
self.__project_path = project_path # 当前的项目路径(通过在main中传入__file__)
self.__route_dir = route_dir # 存放route的目录(指定存放route的目录)
self.__file_dir = file_dir # 需要生成路由的目录(指定route目录下需要生成路由的目录)
self.route_start_path = self.__project_path + "/{0}/".format(self.__route_dir) + self.__file_dir
def build(self):
self.__make(self.route_start_path) # 需要生成路由的起始目录
def __make(self,top):
for f in os.listdir(top):
pathname = os.path.join(top, f)
try:
mode = os.stat(pathname).st_mode # python获取文件的状态,用于进一步判断是否文件
except PermissionError:
continue
else:
if S_ISDIR(mode) and not f.startswith("__"): # 如果是文件夹且不以"__"开头,则继续获取下层内容
self.__make(pathname)
elif S_ISREG(mode): # 如果是文件,则生成路由配置
self.__gen_route(pathname,f.replace(".py",""))
else:
pass
def __gen_route(self,pathname,name):
moduleStr = pathname.replace(self.__project_path+"/","").replace(".py","").replace("/",".").replace("\\",".")
moduleObj = importlib.import_module(moduleStr)
for classStr in dir(moduleObj):
if not classStr.startswith("__") and classStr.lower()==name.lower():
classObj = getattr(moduleObj, classStr)
obj = classObj()
urlPath = pathname.replace(self.__project_path,"").replace("{0}/".format(self.__route_dir),"").replace(".py","").replace("\\","/").lower()
for methodName in dir(classObj):
if not methodName.startswith("__"):
print('产生路由:', urlPath+"/"+methodName.lower())
self.__app.add_url_rule(urlPath+"/"+methodName.lower(),
endpoint=urlPath.replace("/","-")+"-"+methodName,
view_func=getattr(obj, methodName),
methods=["GET","POST"])