Sqlalchemy сумма истинного - PullRequest
       0

Sqlalchemy сумма истинного

1 голос
/ 03 февраля 2020

У меня есть следующая таблица, и я хочу вычислить sum из true и group_by по дате

| Date       | Value1      | Value2       |
|:-----------|------------:|:------------:|
| 2019-12-05 |        5.5  |     true|
| 2019-12-05 |        4.5  |     true|
| 2019-12-05 |        6.5  |     false|
| 2019-12-05 |        8.5  |     false|
| 2019-12-05 |        2.5  |     true|
| 2019-12-06 |        3.5  |     true|
| 2019-12-06 |        5.5  |     true|
| 2019-12-06 |        56.5 |    true|
| 2019-12-06 |        8.5  |     true|
| 2019-12-06 |        99.5 |     false|
| ...        |        ...  |     ...      |

мой запрос в db.Model по методу class:

return db.session.query(cls.Date, func.count(cast(cls.Value2, sqlalchemy.Integer)).filter(cls.Value2== True).label("Count True Value2")) \
    .group_by(cls.Date)\
    .order_by("Date")

Результат должен быть

   | Date       | Value2     |
    |:-----------|------------:|
    | 2019-12-05 |        3  |     
    | 2019-12-06 |        4  |

EDIT тоже не работает

  return db.session.query(cls.Date, func.sum(case([(cls.Value2== True, 1)], else_=0).label('Value2'))) \
            .filter(cls.Date== current_date.strftime("%Y-%m-%d")) \
            .group_by(cls.Date)\
            .order_by("Date")  

1 Ответ

1 голос
/ 03 февраля 2020
  • Вам необходимо добавить Value2 к групповому условию.

Вот пример:

from datetime import date

from sqlalchemy import Integer
from sqlalchemy import func
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import create_engine
from sqlalchemy import Column, Date, Boolean
from sqlalchemy.orm import sessionmaker

Base = declarative_base()
engine = create_engine('...creds...', echo=True)


class TestModel(Base):
    __tablename__ = 'test_model'
    id = Column(Integer, primary_key=True)
    date = Column(Date)
    value2 = Column(Boolean)


Base.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()
# a few test records
for date_ in (date(2020, 1, 1), date(2020, 1, 2)):
    session.add(TestModel(date=date_, value2=True))
    session.add(TestModel(date=date_, value2=True))
    session.add(TestModel(date=date_, value2=True))
    session.add(TestModel(date=date_, value2=False))

session.commit()

query = (
    session.query(
        TestModel.date,
        TestModel.value2,
        func.count('*').label('counter')
    )
    # uncomment if you need only True values
    # .filter(TestModel.value2.is_(True))
    .group_by(TestModel.date, TestModel.value2)
    .order_by(TestModel.date)
)

for rec in query:
    print('date {date}, value2 {value2}, counter {counter}'.format(
        date=rec.date,
        value2=rec.value2,
        counter=rec.counter,
    ))

# date 2020-01-01, value2 True, counter 3
# date 2020-01-01, value2 False, counter 1
# date 2020-01-02, value2 False, counter 1
# date 2020-01-02, value2 True, counter 3

Надеюсь, это поможет.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...