Question

我的查询：

       if current_user['idx'] == 1:
            groups = sess.query(
            Role
            ).filter(
                ~Role.name.in_(codes.group1) # filtering
            ).order_by(
                Role.name
            )
        else:
            groups = sess.query(
            Role
            ).filter(
                ~Role.name.in_(codes.group2) # filtering
            ).order_by(
                Role.name
            )

我的代码文件：

group1= ['User']
group2 = ['Admin', 'User']

在上述查询中，我使用 codes.group1 语句下的if和 codes.group2下的else进行过滤。由于代码的查询部分看起来是重复的，是否有任何方法可以使代码在不重复的情况下更具可读性和简洁性？

Answer 1

由于查询几乎相同，因此您可以执行以下操作：

normal_user = current_user['idx'] == 1
groups = sess.query(
    Role
).filter(
    ~Role.name.in_(codes.group1 if normal_user else codes.group2)
).order_by(
    Role.name
)

或

group = codes.group1 if current_user['idx'] == 1 else codes.group2
groups = sess.query(
    Role
).filter(
    ~Role.name.in_(group)
).order_by(
    Role.name
)

Answer 2

您可以像这样逐步创建查询：

groups = sess.query(Role)

if current_user['idx'] == 1:
    groups = groups.filter(~Role.name.in_(codes.group1))
else:
    groups = groups.filter(~Role.name.in_(codes.group2))

groups = groups.order_by(Role.name)

围绕Flask SqlAlchemy查询简化一对if-else语句

2 个答案: