我有兩張桌子。一個被映射為一個 ORM 類，如下所示：
from typing import Union
from dataclasses import dataclass
from api.models import db, SerializerMixin
@dataclass
class Company(db.Model, SerializerMixin):
__tablename__ = "Company"
company_id: int = db.Column(db.Integer, primary_key = True)
name: str = db.Column(db.String(50), nullable = False, unique = True)
primary_contact: Union[int, None] = db.Column(db.Integer, db.ForeignKey("User.user_id"))
website: Union[str, None] = db.Column(db.Text)
第二個只是寫成一個普通的 SQL 表（這樣寫是因為我不打算直接從這個表查詢，只用于 JOINS 和 COUNTS）：
user_company_table = db.Table("UserCompany",
db.Column("user_id", db.Integer, db.ForeignKey("User.user_id"), primary_key = True, unique = True),
db.Column("company_id", db.Integer, db.ForeignKey("Company.company_id"), primary_key = True),
db.Column("assigned", db.DateTime, server_default = text("CURRENT_TIMESTAMP"), onupdate = datetime.utcnow),
db.Column("approved", db.Boolean, nullable = False, default = False)
)
我需要從 Company 中獲取所有列，同時獲取擁有approved = true并分配給一個特定公司的所有用戶的 COUNT 。生成的 SQL 將如下所示：
SELECT Company.company_id, Company.name, Company.primary_contact, Company.website, (SELECT COUNT(1) FROM UserCompany WHERE approved = true and company_id = Company.company_id) AS user_count FROM Company;
這應該給我這個示例結果：
user_count附加到 Company 表的子查詢的結果集也是如此。我將如何user_count向 Company 類添加一個屬性，以便在我運行時立即加載此子查詢并將結果附加為一列Company.query.all()？
注意：使用 Flask 2.0 、Flask-SQLAlchemy、Python 3.9（解釋了 @dataclass 和模型屬性中的型別提示）
uj5u.com熱心網友回復：
我認為您可以為此使用相關子查詢關系混合：
class Company:
...
@hybrid_property
def user_count(self):
return db.session.query(user_company_table)\
.filter_by(company_id=self.id, approved=True)\
.count()
@user_count.expression
def user_count(cls):
return select(func.count(1))\
.where(user_company_table.c.company_id==cls.id)\
.where(user_company_table.c.approved==True)\
.label('user_count')
您甚至可以選擇帶有標量子查詢的非常短的 column_property：https ://docs.sqlalchemy.org/en/14/orm/mapped_sql_expr.html#using-column-property
轉載請註明出處，本文鏈接：https://www.uj5u.com/net/318729.html