将字符串转换为数字的装饰器[重复]

Question

我想使用装饰器将字符串格式的数字转换为 int/float 值，这就是我的尝试

def str_to_int(func):
     """
        This wrapper converts string value into integer
     """
     def wrapper(*args, **kwargs):
         for arg in args:
             arg = int(arg)
         return func(*args, **kwargs)
     return wrapper
 @str_to_int
 def number_as_string(a, b, c):
     return a,b,c
 print (number_as_string('1', '2', '3'))

输出

('1', '2', '3')

但是，我想要像下面这样的东西

(1, 2, 3)

上面的输出我可以用下面的代码生成

def str_to_int(func):
     """
        This wrapper converts string value into integer
     """
     def wrapper(x, y, z):
         return int(x), int(y), int(z)
     return wrapper
 @str_to_int
 def number_as_string(a, b, c):
     return a,b,c
 print (number_as_string('1', '2', '3'))

但是上面的代码首先破坏了使用装饰器的目的，因为我想转换字符串值而不管原始函数的参数。

任何人都可以建议第一个程序有什么问题以及如何解决它。

Answer 1

它没有按预期工作的原因是因为您从未更新传入的args，您只是不断覆盖循环中的值。

考虑这个修订版：

>>> def str_to_int(f):
...   def wrapper(*args, **kwargs):
...     args = map(int, args)
...     return f(*args, **kwargs)
...   return wrapper
...
>>> @str_to_int
... def foo(a,b,c):
...   return a,b,c
...
>>> foo('1','2','3')
(1, 2, 3)

关键是列表行：

args = map(int, args)

它确保原始 args 元组被完全替换，在您的循环中，您默默地丢弃了该值。

对于 Python3，您应该使用：

args = tuple(map(int, args))

因为地图会返回一个地图对象。

当然，我希望你意识到这个装饰器只有在你传递它而不是实际转换为数字时才有效，如果你传递一个字符串，那么它会引发一个ValueError：

>>> foo('hello')
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 3, in wrapper
ValueError: invalid literal for int() with base 10: 'hello'

如果你想防止这种情况，你必须忽略那些不能转换为整数的值：

def wrapper(*args, **kwargs):
   _temp = []
   for i in args:
     try:
       _temp.append(int(i))
     except ValueError:
       _temp.append(i)
   args = tuple(_temp)
   return f(*args, **kwargs)

现在它会简单地忽略它无法转换的东西：

>>> foo('hello', '1', '2')
('hello', 1, 2)

您可以稍微缩短代码，因为* 会自动扩展参数，您可以利用它来发挥自己的优势：

 # args = tuple(_temp)
 return f(*_temp, **kwargs)