2009-10-02 5 views
0

Je voudrais accéder au Foo exemple foo au sein de ma méthode de gestion baz:Django: Comment accéder à l'instance d'origine à partir d'un RelationshipManager?

foo.bar_set.baz() 

baz aurait normalement prendre un argument de type Foo:

BarManager(models.Manager): 
    def baz(self, foo=None): 
     if foo is None: 
      # assume this call originates from 
      # a RelatedManager and set `foo`. 
      # Otherwise raise an exception 
     # do something cool with foo 

De cette façon, à la fois la première requête ci-dessus et la le suivant fonctionne de manière identique:

Bar.objects.baz(foo) 

Bar aurait un ForeignKey à Foo:

class Bar(models.Model): 
    foo = models.ForeignKey(Foo) 
    objects = BarManager() 

Répondre

1

Si je comprends bien ce que vous voulez correctement, vous devez faire ceci:

BarManager(models.Manager): 
     use_for_related_fields = True 

Edit: apparemment je réussi à rater la pointer complètement. Vous pouvez utiliser quelque chose comme ça (peut-être un peu trop "magique" à mon goût, mais bon):

class BarManager(models.Manager): 
    use_for_related_fields = True 

    def bar(self, foo=None): 
     if foo == None: 
      qs = Foo.objects.all() 
      for field_name, field_val in self.core_filters.items(): 
       field_name = field_name.split('__')[1] 
       qs = qs.filter(**{ field_name: field_val }) 
      foo = qs.get() 
     # do k00l stuff with foo 
+0

Non, ce n'est pas nécessaire. La méthode baz est déjà disponible sur Foo.bar_set – muhuk