X-Git-Url: http://git.tdb.fi/?p=poefilter.git;a=blobdiff_plain;f=scrape-wiki-data.py;fp=scrape-wiki-data.py;h=7f5d3da06d8c3494f3c62221cc0eecf2ce53c57a;hp=470d282e21876a6655c07a9429ba683f11daba0c;hb=ea5b219d29fabbe0ff9dea0d1ab4ca918cefb6ec;hpb=72408a21b9a598802d79a2d1845e638592ee9acb

diff --git a/scrape-wiki-data.py b/scrape-wiki-data.py
index 470d282..7f5d3da 100755
--- a/scrape-wiki-data.py
+++ b/scrape-wiki-data.py
@@ -108,6 +108,8 @@ class WikiTableParser(WikiParser):
 		pass
 
 class DivinationCardsParser(WikiTableParser):
+	reward_kinds = ("currency", "map", "unique", "links", "skillgem", "rare", "magic", "other")
+
 	def handle_starttag(self, tag, attrs):
 		super(DivinationCardsParser, self).handle_starttag(tag, attrs)
 
@@ -340,10 +342,10 @@ def main():
 		by_reward.setdefault(it.reward_kind, []).append(it)
 
 	out = open("cards.txt", "w")
-	for r, il in by_reward.items():
+	for r in DivinationCardsParser.reward_kinds:
 		out.write('category "card.{}"\n'.format(r))
 		out.write('{\n\tclass "Card";\n\tor\n\t{\n')
-		for it in il:
+		for it in by_reward[r]:
 			out.write('\t\tbase_type "{}";\n'.format(it.name))
 		out.write("\t};\n};\n")