sqlalchemy使列遵循与规则的关系

时间:2017-05-25 16:18:41

标签: python python-2.7 sqlalchemy flask-sqlalchemy

如何将attr作为要检查的列的名称并重置要检查的列的值?

models.py

class Dates(db.Model):
    username=db.Column(db.String, primary_key=True)
    apples = db.Column(db.DateTime)
    oranges = db.Column(db.DateTime)
    milk = db.Column(db.DateTime)
    bananas = db.Column(db.DateTime)
    beans = db.Column(db.DateTime)

    def is_expired(self, date)
        #I want attr to equal the column it is coming from
        #example if I typed Dates.apples.is_expired('2017-01-01')
        #I want attr to equal 'apples', that way Rules.rule brings back
        # the rule # associated with apples
        #I want value to be the date from Dates.apples
        a=Rules.query.filter_by(name=attr).first()
        if a.rule=='1':
            value=datetime.datetime.strptime(value, '%Y-%m-%d')
            date=datetime.datetime.strptime(date, '%Y-%m-%d')
            for i in range(36):
                value=add_one_month(value)
            if date>value:
                return True
            else:
                return False
        elif a.rule=='2':
            value=datetime.datetime.strptime(value, '%Y-%m-%d')
            date=datetime.datetime.strptime(date, '%Y-%m-%d')
            for i in range(12):
                value=add_one_month(value)
            if date>value:
                return True
            else:
                return False

    def __repr__(self):  # pragma: no cover
        return '<Mobility %r>' % (self.username)

class Rules(db.model)
    rule=db.Column(db.Integer, primary_key=True)
    name=db.Column(db.String(45))

    def __repr__(self):  # pragma: no cover
        return '<Mobility %r>' % (self.name)

另一个函数中的调用示例:

user=Dates.query.filter_by(username='bob').first()
user.apples='2017-02-11'
if user.apples.is_expired('2017-04-12'):
    color=red

在这种情况下,attr =&#39; apples&#39;,value =&#39; 2017 = 02-11&#39;

1 个答案:

答案 0 :(得分:1)

如果要在函数中包含attr,则应将其作为参数传递。 这实际上与SQLAlchemy无关 - python对象不会像你在这里尝试使用它那样工作:object.attr.method()不是调用object方法的有效方法。

相反,你需要像user.is_expired('apples', '2017-04-12')这样的东西。在方法内部,您可以将字段值设为getattr(self, attr_name),假设您将方法签名更改为def is_expired(self, attr_name, date)