Python利用裝飾器實現類似於flask路由

2023-02-28 18:01:32

1.例子1

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

列印結果：

33

在例子1中，f1() 與f2() 都沒有被呼叫，只執行了print(33)

f1與f2，是沒有被呼叫的，但是如果f1 和 f2 上面有註解，就會被呼叫執行。

2.python 利用裝飾器實現類似於flask路由

註釋類 Grass

# -*- 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()

執行app.run()

然後輸入：