From: Lynn "Zhorken" Vaughan <zhorken@gmail.com>
Date: Sat, 11 Dec 2010 17:10:08 +0000 (-0500)
Subject: Give abilities separate relations for all/normal/dream Pokémon.
X-Git-Tag: veekun-promotions/2010121201~4
X-Git-Url: http://git.veekun.com/zzz-pokedex.git/commitdiff_plain/3da53ae05114c4dcfa0451782c7fc588183a51df

Give abilities separate relations for all/normal/dream Pokémon.
---

diff --git a/pokedex/db/tables.py b/pokedex/db/tables.py
index 63b2efc..c4759e0 100644
--- a/pokedex/db/tables.py
+++ b/pokedex/db/tables.py
@@ -1306,8 +1306,25 @@ Ability.changelog = relation(AbilityChangelog,
 Ability.flavor_text = relation(AbilityFlavorText, order_by=AbilityFlavorText.version_group_id, backref='ability')
 Ability.foreign_names = relation(AbilityName, backref='ability')
 Ability.generation = relation(Generation, backref='abilities')
+Ability.all_pokemon = relation(Pokemon,
+    secondary=PokemonAbility.__table__,
+    back_populates='all_abilities',
+)
 Ability.pokemon = relation(Pokemon,
     secondary=PokemonAbility.__table__,
+    primaryjoin=and_(
+        PokemonAbility.ability_id == Ability.id,
+        PokemonAbility.is_dream == False
+    ),
+    back_populates='abilities',
+)
+Ability.dream_pokemon = relation(Pokemon,
+    secondary=PokemonAbility.__table__,
+    primaryjoin=and_(
+        PokemonAbility.ability_id == Ability.id,
+        PokemonAbility.is_dream == True
+    ),
+    back_populates='dream_ability',
 )
 
 AbilityChangelog.changed_in = relation(VersionGroup, backref='ability_changelog')