如何覆盖python app引擎模型上的put()方法?

时间:2011-09-27 12:55:23

标签: python google-app-engine

在Appengine中,我试图自动计算属性值并与对象一起存储。

我有一个类Rectangle,它有宽度,高度和面积。显然该区域是宽度和高度的函数,但我希望它是一个属性,因为我想用它来进行排序。因此我尝试修改put()函数以在存储Rectangle时隐藏该区域,如下所示:

class Rectangle(db.Model):
    width   = db.IntegerProperty()
    height  = db.IntegerProperty()
    area    = db.IntegerProperty()

    def put(self, **kwargs):
        self.area = self.width * self.height
        super(Rectangle, self).put(**kwargs)

当我直接在Area对象上调用put()时,这是有效的:

re1 = Rectangle(width=10, height=10)
re1.put()
print re1.area      # >> 10

但是当我使用db.put()时(例如,一次性保存很多),这会中断。

re2 = Rectangle(width=5, height=5)
db.put(re2)
print re2.area      # >> None

“潜入”计算值的正确方法是什么?

2 个答案:

答案 0 :(得分:7)

不要覆盖put - 正如您所观察到的那样,它很脆弱,如果您调用db.put而不是模型的put函数,则不会调用它。

幸运的是,App Engine提供了ComputedProperty,这使您的用例非常简单:

class Rectangle(db.Model):
    width   = db.IntegerProperty()
    height  = db.IntegerProperty()

    @db.ComputedProperty
    def area(self):
      return self.width * self.height

答案 1 :(得分:1)

我同意ComputedProperty是描述特定方案的方法。但是,重载put函数仍然有用。例如,我们使用以下代码来跟踪发出数据存储区写入的所有调用方,以便我们可以轻松地调试写入中的峰值。

from google.appengine.ext import db

_orig_db_put_async = db.put_async
_orig_db_model_put = db.Model.put

def _new_db_put_async(models, *args, **kwargs):
    """Instrumented version of db.put_async (which db.put also calls)."""
    retval = _orig_db_put_async(models, *args, **kwargs)
    msg = ['query: %s' % _get_caller()]
    # 'models' can be either a single model instance, or a list of them.
    try:
        for model in models:
            msg.append(model.__class__.__name__ + '.<db.put>')
    except TypeError:
        msg.append(models.__class__.__name__ + '.<db.put>')
    instance_cache.increment(' -- '.join(msg))
    return retval


def _new_db_model_put(self, *args, **kwargs):
    """Like entity.put() but stores put-stats in the instance cache."""
    retval = _orig_db_model_put(self, *args, **kwargs)
    msg = ['query: %s' % _get_caller()]
    msg.append(self.__class__.__name__ + '.<put>')
    instance_cache.increment(' -- '.join(msg))
    return retval

此代码记录在memcache中发出写入的代码路径,然后偶尔将其刷新到日志中。日志行看起来像这样:

3041: activity_summary.py:312 -- UserData.<put>

其中3041是activity_summary.py的行312发出UserData.put()的次数。