可以在 Python 中 str 或 list 的处理函数参数

Handling function param that can str or list in Python

我不是一个经验丰富的 Python 程序员,但我觉得我对这个问题的解决方法不对,我认为 Python 中有更好的方法来处理这个问题。

在这种情况下,这是使用 Hug API 但这可能几乎无关紧要。

假设代码是这样的:

@hug.get_post('/hello')
def hello (name)
   print(type(name))
   return name

当使用 name 参数的一个实例发送请求时,hello 函数得到一个 str - 如:

POST /hello?name=Bob

但是如果请求发送多个 name 参数,该方法接收 list 个字符串,如

POST /hello?name=Bob&name=Sally

如果我这样写方法:

@hug.get_post('/hello')
def hello (name: list)
   print(type(name))
   return name

那么单个参数就变成了字符列表。 IE。 ['B', 'o', 'b']。但是如果 name 参数有多个实例(例如 ['Bob', 'Sally']

所以我现在解决它的方法是添加以下代码:

@hug.get_post('/hello')
def hello (name)
   names=list()
   if type(name) != 'list'
      names.append(name)
   else:
      names=name
   return names

这行得通,但感觉不对。我认为有更好的方法可以做到这一点,但我现在想不出来。

我不确定你能不能用更优雅的方式做到这一点。

isinstance() 是一种非常常见的类型检查方式。 下面的代码可以处理元组、列表或字符串

@hug.get_post('/hello')
def hello (name):
   if isinstance(name, (list, tuple)):
      names = name
   else:
      names = [name]      
   return names

如果你想更简洁,你可以像这样使用三元运算符:

@hug.get_post('/hello')
def hello (name)
   return name if isinstance(name, list) else [name]

拥抱为此提供了一种类型:hug.types.multiple

import hug
from hug.types import multiple

@hug.get()
def hello(name: multiple):
    print(type(name))

    return name

现在/hello?name=Bobreturns['Bob']/hello?name=Bob&name=Sallyreturns['Bob', 'Sally'].

如果要验证内部元素,可以像这样创建自定义类型:

import hug
from hug.types import Multiple, number

numbers = Multiple[number]()

@hug.get()
def hello(n: numbers):
    print(type(n))

    return n

在这种情况下 /hello?n=1&n=2 returns [1, 2]/hello?n=1&n=a 导致错误:

{"errors": {"n": "Invalid whole number provided"}}