1 
class Hero(models.Model):
   talents = models.ManyToManyField(Talent)
   (...)

class Talent(models.Model)
   required_talents = models.ManyToManyField('self', symmetrical=False)
   (...)

has_required_talents(self, talent)のメソッドを作成したいと思います。これはHero、これHeroがrequired_talents選択されているかどうかをチェックしTalentます。

私はこれを試しました：

def has_required_talents(self, talent)
   required_list = talent.talents_required.values_list('id', flat=True)
   hero_talents = self.hero.talents.values_list('id', flat=True)
   if required_list in hero_talents:
      return True
   return False

ただし、次のテストを使用してテストすると、正しく機能しません。

class HeroTalents(TestCase):
 def setUp(self):
    self.hero=Hero('Duck')
    self.hero.save() 

 def test_has_required_talents(self):
    self.talent1 = Talent(name = "Overpower")
    self.talent1.save()
    self.talent2 = Talent(name = "Overpower2")
    self.talent2.save()
    self.talent2.talents_required.add(self.talent1)
    self.assertFalse(self.hero.has_required_talents(self.talent2), "test1")
    self.hero.talents.add(self.talent1)
    self.assertTrue(self.hero.has_required_talents(self.talent2), "test2")
    self.talent3 = Talent(name = "Overpower3")
    self.talent3.save()
    self.hero.talents.add(self.talent2)
    self.assertTrue(self.hero.has_required_talents(self.talent3), "test3")
    self.talent1.talents_required.add(self.talent3)
    self.assertFalse(self.hero.has_required_talents(self.talent1), "test4")

この作業を行うには何をする必要がありますか？

4

1 に答える 1

0 
def has_required_talents(self, talent)
   required_list = talent.talents_required.values_list('id', flat=True)
   hero_talents = self.hero.talents.values_list('id', flat=True)
   for item in required_list:
      if not item in hero_talents:
         return False
   return True

それはリストベースのアプローチになります。それらをセットに変換することもできます。

def has_required_talents(self, talent)
   required_list = set(talent.talents_required.values_list('id', flat=True))
   hero_talents = set(self.hero.talents.values_list('id', flat=True))
   if required_list.issubset(hero_talents):
      return True
   return False

しかし（正確な内部についてはわかりません。いくつかのテストを実行できます）変換がないため、最初のアプローチの方が速いはずです。

于 2012-11-01T12:09:42.663 に答える