在Django中使用用户输入构造动态查询

时间:2018-10-10 19:25:56

标签: python django django-orm

我有一个大模型,可以存储10个基于文本的值和10个数字值。像这样:

class Dataset(models.Model):
    text_field_1 = models.CharField(max_length=255)
    .
    .
    .
    text_field_10 = models.CharField(max_length=255)

    number_field_1 = models.IntegerField()
    .
    .
    .
    number_field_10 = models.IntegerField()

现在,我想做的是通过将对象传递给视图,为用户提供一种使用这些字段进行过滤和排序数据集的方法。希望下面的示例显示我想要做的事情:

obj = {
    "filters" : [
        "text_field_1='vaccines'",
        "number_field_5__gte=1000",
    ],
    "order_by" : "text_field_3, -number_field_7",
    "excludes" : [],
}

generate_query(obj) # Dataset.objects.filter(text_field_1='vaccines', number_field_5__gte=1000).order_by('text_field_3', '-number_field_7')

因此,通过调用generate_query(obj),我们将在注释中获得查询集。现在,由于该模型的性质,我无法通过考虑过滤器,订单和排除的每种可能组合来手动执行此操作。

什么是最好的,最安全的实现方法?我唯一想到的是创建一个大字符串,然后使用execeval执行该字符串。

2 个答案:

答案 0 :(得分:2)

使用字典进行过滤和排除,并使用order_by列表,您可以使用参数***来解压缩参数。

obj = {
    "filters" : {
        "text_field_1": "vaccines",
        "number_field_5__gte": "1000",
    },
    "order_by" : ["text_field_3", "-number_field_7"],
    "excludes" : {},
}

Dataset.objects.filter(**obj['filters']).exclude(**obj['excludes']).order_by(*obj['order_by'])

只要您不允许用户构造自己的__参数,它应该是相当安全的。例如,过滤器event__owner__password='hunter2'可用于间接查询相关模型中的字段。

答案 1 :(得分:0)

有一个很好的软件包专门用于根据用户输入进行文件归档:

https://django-filter.readthedocs.io/en/master/