Python利用装饰器实现类似于flask路由

def f1():
    print(1111)
 
 
def f2():
    print(2222)
 
 
if __name__ == '__main__':
    print(33)

# -*- coding:utf-8 -*-
# @Author: 喵酱
# @time: 2023 - 02 -21
# @File: grass.py
 
from types import FunctionType
 
class Grass(object):
    # 字典，key 是 用户输入的路由
    # value，是调用对应的函数
    url_map = {}
 
    def router(self,url):
        def decorator(f: FunctionType):
            self.add_url_to_map(url,f)
            # return f
        return decorator
 
    # f 指的是一个函数
    def add_url_to_map(self,url,f):
        self.url_map[url] = f
 
    def run(self):
        while True:
            url = input("请输入URL： ")
            try:
                print(self.url_map[url]())
            except Exception as e:
                print(404)
                print(e)

# -*- coding:utf-8 -*-
# @Author: 喵酱
# @time: 2023 - 02 -21
# @File: blog.py
 
from grass import Grass
 
app = Grass()
 
@app.router("/home")
def home():
    print("欢迎来到首页")
    return "首页"
 
@app.router("/index")
def index():
    print("欢迎来到列表页")
    return "列表页"
 
 
if __name__ == '__main__':
    app.run()