非 SQL 相关属性添加到 SQLAlchemy 模型错误

Question

我正在使用 python 和烧瓶开发一个网络应用程序。我使用 Flask、SQLAlchemy 和 PostgreSQL 进行开发。我有多对一相关模型。通过这种模型，一家公司可以拥有多个用户，但每个用户只能拥有一家公司。

models.py
class Company(ResourceMixin, db.Model):
    __tablename__ = 'companies'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(100), unique=True, index=True, 
                     nullable=False, server_default='')
    phone = db.Column(db.String(24)) 
    email = db.Column(db.String(255), index=True)
    address = db.Column(db.String(255))

    # Relations
    users = db.relationship('User', backref='company')

class User(UserMixin, ResourceMixin, db.Model):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)

    # User details
    name = db.Column(db.String(50), index=True)
    phone = db.Column(db.String(24)) 
    address = db.Column(db.String(255))
    email = db.Column(db.String(255), unique=True, index=True, nullable=False,
                      server_default='')
    password = db.Column(db.String(128), nullable=False, server_default='')

    # Relations
    company_id = db.Column(db.Integer, db.ForeignKey('companies.id', 
                                                     onupdate='CASCADE',
                                                     ondelete='SET NULL'), 
                           index=True)


views.py
app.route('/')
def index():
    company = Company.query.get(1)
    flash(company.name, company.user_count)
    return render_template('index.html')

错误摘要：“user_count”属性不是公司模型的一部分。

我想从公司模型中动态获取用户数。属性应在模型的每次调用中计算用户数，并在常规属性上提供服务（如 company.user_count）。我通过创建一个类方法并在视图函数中调用它来实现它，但我希望它使过程自动化，而无需在使用属性之前调用方法。

我尝试了这样的 init 函数：

def __init__(self):
    self.user_count = len(self.users)

像这样：

def __init__(self):
    self.status()

def status(self):
    self.user_count = len(self.users)
    return True

像这样：

def __init__(self):
    self.status()

@classmethod
def status(self):
    self.user_count = len(self.users)
    return True

所有三个版本都会引发相同的错误。我该如何克服这个问题。

非常感谢！

Answer 1

你可以使用一个属性：

class User(Base):
    ...
    @property
    def user_count(self):
        return len(self.users)