接上一篇

終于知道python的裝飾器是怎么回事，那在工作中，到底能干嗎用呢？
尤其對我這個只會寫寫腳本又不做python開發(fā)的小測試/手動無辜臉。

先說結(jié)論，肯定是有用處滴。

一、自動化測試中使用

就拿寫的自動化測試來說吧，如果我想統(tǒng)一的輸出點東西，比如：case的運行時長，case名稱等等，那就可以用起來。

首先，看下最簡單的case，沒有裝飾器：

import pytest

def test_01():
 a = 1
 b = 2
 assert a < b

def test_02():
 a = 1
 b = 1
 assert a-b == 0

if __name__ == '__main__':
 pytest.main(['demo_test.py'])

這個用例文件里有2個case，運行一下：

============================= test session starts =============================
platform win32 -- Python 3.8.5, pytest-6.0.1, py-1.9.0, pluggy-0.13.1
rootdir: D:\練習(xí)
collected 2 items

demo_test.py .. [100%]

============================== 2 passed in 0.01s ==============================
[Finished in 0.4s]

運行通過就是.，所以2個case都通過了，有2個.。

如果我想在運行結(jié)果里看到每個case的執(zhí)行時長，那么就可以寫一個裝飾器去處理：

import pytest
import functools
import time

def log_execute_time(func):
 @functools.wraps(func)
 def wrapper(*args, **kwargs):
  start = time.perf_counter()
  res = func(*args, **kwargs)
  end = time.perf_counter()
  print("{} 用時 {} ms".format(func.__name__, (end - start) * 1000))
  return res
 return wrapper

@log_execute_time
def test_01():
 a = 1
 b = 2
 assert a < b

@log_execute_time
def test_02():
 a = 1
 b = 1
 assert a-b == 0

if __name__ == '__main__':
 pytest.main(['-s','demo_test.py'])

運行一下：

============================= test session starts =============================
platform win32 -- Python 3.8.5, pytest-6.0.1, py-1.9.0, pluggy-0.13.1
rootdir: D:\練習(xí)
collected 2 items

demo_test.py
test_01 用時 0.0007999999999674934 ms .
test_02 用時 0.0012999999999818712 ms .

============================== 2 passed in 0.03s ==============================
[Finished in 0.4s]

可以看到，每個case執(zhí)行會打印出執(zhí)行耗時。

上述的這個例子是在實現(xiàn)自動化測試時的場景，具體還是要看你怎么去挖掘你的需求。

二、其他python開發(fā)向的用法

我也去查了下其他的用法，其實在python開發(fā)中，用途還是很多的，其中就還有大家熟悉的"身份認(rèn)證"。

比如大家愛上博客園，你不登錄賬號也可以瀏覽博客。當(dāng)你看著興起，自己也想評論或者發(fā)文的時候，發(fā)現(xiàn)
提示你需要登錄，這種場景就可以用上裝飾器。代碼舉例：

import functools

def authenticate(func):
 @functools.wraps(func)
 def wrapper(*args, **kwargs):
  request = args[0]
  if check_user_logged_in(request): # 檢查用戶是否登錄
   return func(*args, **kwargs) # 如果登錄了，就可以執(zhí)行函數(shù)post_comment() 發(fā)送評論
  else:
   raise Exception('Authentication failed') # 否則，身份驗證失敗
 return wrapper
 
@authenticate
def post_comment(request)
 pass

上述代碼只是說了這個意思，輔助描述下場景。

除此之外，還可以在輸入合理性檢查、緩存等多個場景中使用，畢竟不是做開發(fā)的，這里就不再深入了。

三、裝飾器小結(jié)

裝飾器本質(zhì)上是一個python函數(shù)或者類，可以讓其他函數(shù)或類在不需要做任何代碼修改的前提下，增加額外的功能。
裝飾器的返回值也是一個函數(shù)對象或者類對象。

有了裝飾器，我們可以抽離出大量與函數(shù)功能本身無關(guān)的雷同代碼到裝飾器中，并且可以重用。
說到這，我感覺跟AOP面向切面編程有點像。

大家還知道哪些應(yīng)用場景，也歡迎留言補(bǔ)充。

以上就是python 裝飾器的實際作用有哪些的詳細(xì)內(nèi)容，更多關(guān)于python 裝飾器作用的資料請關(guān)注腳本之家其它相關(guān)文章！

最新評論