最近想用Python寫一個簡單生成器,類似指定類型和范圍,返回指定列表;
比如想要 0 ~ 3 的整數,則 我只需要指定:
最小:0,
最大:3,
步長:1
則返回一個 [0,1,2,3] 的列表
思路是用Python 裝飾器,在def方法上打裝飾器,裝飾器內部替換方法調用時傳入的參數值
代碼如下:
1 # coding=utf-8 2 3 def integer(min=0, max=100, step=1): 4 def deco(func): 5 def wrapper(val): 6 val = [] 7 for i in range(min, max, step): 8 val.append(i) # 將入參修改掉 9 return func(val) 10 return wrapper 11 return deco 12 13 14 @integer(1, 40) 15 def id(val): 16 return val 17 18 a = id(None) 19 print(a)
這樣 打印 的 a的值就是我要的列表:
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22,
23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39]
比Java annotation寫起來方便多了,Python大法好!