如何实现大于或小于 FastAPI 的查询参数

Question

我想构建一个类似于< code>/cities的endpoint。population__gt=100000获取人口超过100，000的所有城市。

在FastAPI中最好的方法是什么？我不喜欢列举给定字段的所有可能的操作符，比如:

@app.get("/cities")
def get_city(
    population: int = None,
    population_gt: int = None,
    population_gte: int = None,
    population_lt: int = None,
    population_lte: int = None):
    return ...

Answer 1

实现这一点的一种方法是传递运算符（即gt、lt等），并为每个操作都有一个单独的函数，如果收到运算符，您可以调用该函数。下面的示例。您可以根据需要调整/添加函数。

def gt(population):
    return f"operator: gt, population: {population}"
 
def gte(population):
    return f"operator: gte, population: {population}"
 
def lt(population):
    return f"operator: lt, population: {population}"

def lte(population):
    return f"operator: lte, population: {population}"
   
operators = {"gt": gt, "gte": gte, "lt": lt, "lte": lte}
   
@app.get("/cities")
def get_city(population: int = None, operator: str = None):
    if operator in operators:
        result = operators[operator](population)
        return result
    else:
        return "Operator Not Found!"

正如您刚才所说，因为这是多个参数所需要的，所以我建议使用类似的方法，您可以在每个键的值中传递运算符，然后在服务器端提取它，例如/city？人口=~gt～100000

要接收查询参数的多个值，例如上面给出的示例URL中的总体，执行诸如之类的操作，其中总体大于

下面的示例让您了解了使用人口参数的这种方法背后的逻辑，并展示了如何在服务器端提取运算符和实际值。然后由您定义所需的其余参数，以及如何处理这些数据；例如，使用与上面类似的方法（每个操作都有一个函数），或者在循环数据时构建将要传递到数据库的查询。

import re
pattern = "~(.*?)~"
 
@app.get("/cities")
def get_city(population: List[str] = Query(None)):
    pop_qs = {}
    
    for p in population:
        try:
            pop_opr = re.search(pattern, p).group(1)
            pop_no_str = re.sub(r"^~.*?~", "", p)
        except:
            return "Invalid data received"
        try:
            pop_no = int(pop_no_str)
            pop_qs.update({pop_opr : pop_no})
        except ValueError:
            return f"Cannot convert '{pop_no_str}' to Integer."
            
    return pop_qs