SqlAlchemy (Postgres + Flask): как суммировать несколько столбцов?

У меня есть класс Score со столбцом item_id и несколькими полями, имеющими разные типы оценок (score1, score2, score3...), все из которых имеют целочисленные значения.

Мне нужно написать запрос, который берет список типов оценок и возвращает список с объектами, имеющими itemid и сумму оценок всех типов оценок, упомянутых в списке рядом. Я пытаюсь сделать это с помощью гибридного метода, но не понимаю, как написать запрос.

model.py

class Score(db.Model):

    __tablename__ = 'scores'

    item_id                     = db.Column(db.Integer(), primary_key=True)
    score1                      = db.Column(db.Integer(), nullable=False)
    score2                      = db.Column(db.Integer(), nullable=False)
    score3                      = db.Column(db.Integer(), nullable=False)
    score4                      = db.Column(db.Integer(), nullable=False)

    @hybrid_method
    def total_score(self, fields):
        ts = 0
        for field in fields : 
            ts = ts + self[field]
        return ts

controller.py

app.route('/scores', methods=['POST'])
def scores():
    fields = ['score1', 'score2']
    scores = Score.query.all().order_by('total_score')

Очевидно, что это не работает. Не могли бы вы помочь мне написать запрос, большое спасибо!

Вот как мне нужно получить окончательный результат:

[{'item_id' : 'x1', 'total_score' : y1},{'item_id' : 'x2', 'total_score' : y2},{'item_id' : 'x3', 'total_score' : y3}, ...]

vivekanon 28.02.2016 источник

Ответы (1)

arrow_upward
1
arrow_downward

Вам необходимо создать выражение для hybrid_method

class Score(db.Model):
    __tablename__ = 'scores'
    item_id  = db.Column(db.Integer(), primary_key=True)
    score1 = db.Column(db.Integer(), nullable=False)
    score2 = db.Column(db.Integer(), nullable=False)
    score3 = db.Column(db.Integer(), nullable=False)
    score4 = db.Column(db.Integer(), nullable=False)

    @hybrid_method
    def total_score(self, fields):
        return sum(getattr(self, field) for field in fields)

    @total_score.expression
    def total_score(cls, fields):
        return sum(getattr(cls, field) for field in fields)


fields = ['score1', 'score2']
scores = db.session.query(Score.item_id, Score.total_score(fields).label('total_score')).order_by('total_score')
final_output = [score._asdict() for score in scores]

r-m-n 28.02.2016

comment

Разве явное выражение не является ненужным в данном конкретном случае? Другими словами, поскольку методы экземпляра и класса идентичны, SQLA может неявно использовать метод экземпляра в качестве метода класса (выражения). - Ilja Everilä; 07.05.2018