X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=scrape-item-data.py;h=84def9232b6cf2ddaedc1e3353ed00e17d82f58a;hb=aaa9d76db026c832bfdd936c4069128bde6b7311;hp=e78883710705105a06b1127f9d16bafabc14c155;hpb=14294b0b2b0cf7cc8aa2c854d612378211057b1a;p=poefilter.git

diff --git a/scrape-item-data.py b/scrape-item-data.py
index e788837..84def92 100755
--- a/scrape-item-data.py
+++ b/scrape-item-data.py
@@ -136,7 +136,8 @@ class ItemDataParser(html.parser.HTMLParser):
 			self.mod_names = []
 		elif tag=="td":
 			self.column += 1
-			self.in_cell = True
+			if self.in_items_table:
+				self.in_cell = True
 		elif tag=="h1":
 			self.in_heading = True
 
@@ -166,7 +167,7 @@ class ItemDataParser(html.parser.HTMLParser):
 			if self.current_heading=="Staff":
 				self.current_heading = "Stave"
 			self.headings.append(self.current_heading)
-		elif self.in_items_table and self.in_cell:
+		elif self.in_cell:
 			if self.mod_row:
 				if self.column==1:
 					self.mod_names.append(data)
@@ -319,6 +320,13 @@ def main():
 			out.write('\t\tbase_type "{}";\n'.format(it.name))
 		out.write("\t};\n};\n")
 
+	for b in ("best", "second"):
+		out.write('category "armor.{}_at_level"\n'.format(b))
+		out.write('{\n\tor\n\t{\n')
+		for t in types.keys():
+			out.write('\t\tcategory "armor.{}.{}_at_level";\n'.format(t, b))
+		out.write("\t};\n};\n")
+
 	out.write('category "armor"\n{\n\tor\n\t{\n')
 	for h in p.headings:
 		out.write('\t\tclass "{}";\n'.format(h))
@@ -339,6 +347,13 @@ def main():
 		for i in range(1, 3):
 			write_best_category(out, "weapon.{}".format(h.lower().replace(' ', '_')), items, i)
 
+	for b in ("best", "second"):
+		out.write('category "weapon.{}_at_level"\n'.format(b))
+		out.write('{\n\tor\n\t{\n')
+		for h in p.headings:
+			out.write('\t\tcategory "weapon.{}.{}_at_level";\n'.format(h.lower().replace(' ', '_'), b))
+		out.write("\t};\n};\n")
+
 	out.write('category "weapon"\n{\n\tor\n\t{\n')
 	for h in p.headings:
 		out.write('\t\tclass "{}";\n'.format(h))