计算SQLAlchemy中的关系

时间:2014-08-26 08:05:14

标签: python sql sqlalchemy

我的SQLAlchemy结构看起来像这样

papers2authors_table = Table('papers2authors', Base.metadata,
    Column('paper_id', Integer, ForeignKey('papers.id')),
    Column('author_id', Integer, ForeignKey('authors.id'))
)

class Paper(Base):
    __tablename__ = "papers"

    id = Column(Integer, primary_key=True)
    title = Column(String)
    handle = Column(String)

    authors = relationship("Author",
                    secondary="papers2authors",
                    backref="papers")

class Author(Base):
    __tablename__ = "authors"

    id = Column(Integer, primary_key=True)
    name = Column(String, unique=True)
    code = Column(String, unique=True)

我想查询两件事:

  1. 每篇论文中的作者数量
  2. 每位作者的论文数量(部分回答here
  3. 我尝试了func.count()count()的许多选项,但它们返回了无意义的结果。如何以SQLAlchemy的方式做这两件事?

    我尝试了什么

    • db.s.query(func.count(core.Paper.id)).group_by(core.Author.id).first() = sqlalchemy.exc.OperationalError: (OperationalError) no such column: authors.id
    • db.s.query(func.count(core.Author.papers)).group_by(core.Author.id).first() = (128100),这不符合我的预期
    • db.s.query(core.Author.papers).group_by(core.Author.id).count().first() = AttributeError: 'int' object has no attribute 'first'
    • ...

1 个答案:

答案 0 :(得分:5)

找到解决方案:

  1. 每篇论文的作者数量:db.s.query(core.Paper.title, func.count(core.Author.id)).join(core.Paper.authors).group_by(core.Paper.id).all()
  2. 每位作者的论文数量:db.s.query(core.Author.name, func.count(core.Author.id)).join(core.Author.papers).group_by(core.Author.id).all()
  3. 相关:http://docs.sqlalchemy.org/en/rel_0_9/orm/query.html#sqlalchemy.orm.query.Query.having