欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python装饰器实现类Java注解功能

程序员文章站 2022-05-13 18:26:02
最近想用Python写一个简单生成器,类似指定类型和范围,返回指定列表; 比如想要 0 ~ 3 的整数,则 我只需要指定: 最小:0, 最大:3, 步长:1 则返回一个 [0,1,2,3] 的列表 思路是用Python 装饰器,在def方法上打装饰器,装饰器内部替换方法调用时传入的参数值 代码如下: ......

 

最近想用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大法好!