12

Trước hết, tôi tương đối mới đối với Google App Engine, vì vậy tôi có thể đang làm điều gì đó ngớ ngẩn.Các phím tùy chỉnh cho các mô hình Google App Engine (Python)

Nói rằng tôi đã có một mô hình Foo:

class Foo(db.Model): 
    name = db.StringProperty() 

Tôi muốn sử dụng name như một chìa khóa duy nhất cho mọi đối tượng Foo. Làm thế nào được thực hiện?

Khi tôi muốn có được một đối tượng cụ thể Foo, Tôi hiện truy vấn các kho dữ liệu cho tất cả Foo đối tượng với tên duy nhất mục tiêu, nhưng truy vấn là chậm (cộng với đó là một nỗi đau để đảm bảo rằng name là duy nhất khi mỗi mới Foo được tạo ra).

Có một cách tốt hơn để làm điều này!

Cảm ơn.

Trả lời

13

Tôi đã sử dụng mã dưới đây vào một dự án trước đó. Nó sẽ làm việc miễn là lĩnh vực mà bạn đang dựa trên tên khóa của bạn là bắt buộc.

class NamedModel(db.Model): 
    """A Model subclass for entities which automatically generate their own key 
    names on creation. See documentation for _generate_key function for 
    requirements.""" 

    def __init__(self, *args, **kwargs): 
     kwargs['key_name'] = _generate_key(self, kwargs) 
     super(NamedModel, self).__init__(*args, **kwargs) 


def _generate_key(entity, kwargs): 
    """Generates a key name for the given entity, which was constructed with 
    the given keyword args. The entity must have a KEY_NAME property, which 
    can either be a string or a callable. 

    If KEY_NAME is a string, the keyword args are interpolated into it. If 
    it's a callable, it is called, with the keyword args passed to it as a 
    single dict.""" 

    # Make sure the class has its KEY_NAME property set 
    if not hasattr(entity, 'KEY_NAME'): 
     raise RuntimeError, '%s entity missing KEY_NAME property' % (
      entity.entity_type()) 

    # Make a copy of the kwargs dict, so any modifications down the line don't 
    # hurt anything 
    kwargs = dict(kwargs) 

    # The KEY_NAME must either be a callable or a string. If it's a callable, 
    # we call it with the given keyword args. 
    if callable(entity.KEY_NAME): 
     return entity.KEY_NAME(kwargs) 

    # If it's a string, we just interpolate the keyword args into the string, 
    # ensuring that this results in a different string. 
    elif isinstance(entity.KEY_NAME, basestring): 
     # Try to create the key name, catching any key errors arising from the 
     # string interpolation 
     try: 
      key_name = entity.KEY_NAME % kwargs 
     except KeyError: 
      raise RuntimeError, 'Missing keys required by %s entity\'s KEY_NAME '\ 
       'property (got %r)' % (entity.entity_type(), kwargs) 

     # Make sure the generated key name is actually different from the 
     # template 
     if key_name == entity.KEY_NAME: 
      raise RuntimeError, 'Key name generated for %s entity is same as '\ 
       'KEY_NAME template' % entity.entity_type() 

     return key_name 

    # Otherwise, the KEY_NAME is invalid 
    else: 
     raise TypeError, 'KEY_NAME of %s must be a string or callable' % (
      entity.entity_type()) 

Sau đó, bạn có thể sửa đổi mô hình ví dụ của bạn như sau:

class Foo(NamedModel): 
    KEY_NAME = '%(name)s' 
    name = db.StringProperty() 

Tất nhiên, điều này có thể được đơn giản hóa đáng kể trong trường hợp của bạn, thay đổi dòng đầu tiên của __init__ phương pháp 's NamedModel một cái gì đó như :

kwargs['key_name'] = kwargs['name'] 
+3

Ah, điều này có vẻ thú vị nhưng hơi quá mức. Không có vấn đề gì, nó vẫn dẫn tôi đi đúng hướng để khám phá những gì tôi đã bỏ lỡ: kiến ​​thức về key_name! Đó là chìa khóa để tất cả mọi thứ :-) – Cameron

+0

Hahaha, tốt, tôi vui vì tôi đã chỉ cho bạn đi đúng hướng, ít nhất. –

+2

Cách tiếp cận thú vị. Ví dụ tốt về cách ghi đè __init__ mạnh mẽ, quá. –