Some tweaks to skill gem scraping

[poefilter.git] / scrape-wiki-data.py

diff --git a/scrape-wiki-data.py b/scrape-wiki-data.py
index b12f8a67681aa79e1469c1184749eee4d6d05b25..470d282e21876a6655c07a9429ba683f11daba0c 100755 (executable)
--- a/scrape-wiki-data.py
+++ b/scrape-wiki-data.py
@@ -382,28 +382,28 @@ def main():
        sys.stdout.write(prefix)
 
        out = open("skillgems.txt", "w")
-       out.write('category "skillgem.vaal"\n{\n\tclass "Skill Gem";\n\tor\n\t{\n')
+       out.write('category "skillgem.special.vaal"\n{\n\tclass "Skill Gem";\n\tor\n\t{\n')
        for g in gems:
                if g.vaal:
                        out.write('\t\tbase_type "{}";\n'.format(g.name))
        out.write("\t};\n};\n")
 
-       out.write('category "skillgem.drop"\n{\n\tclass "Skill Gem";\n\tor\n\t{\n')
+       out.write('category "skillgem.special.drop"\n{\n\tclass "Skill Gem";\n\tor\n\t{\n')
        for g in gems:
                if (g.drop_only or not g.price) and not g.vaal:
                        out.write('\t\tbase_type "{}";\n'.format(g.name))
        out.write("\t};\n};\n")
 
        for p, n in SkillGemParser.prices:
-               out.write('category "skillgem.{}"\n'.format(p))
+               out.write('category "skillgem.price.{}"\n'.format(p))
                out.write('{\n\tclass "Skill Gem";\n\tor\n\t{\n')
                for g in gems:
-                       if g.price==p:
+                       if not g.drop_only and g.price==p:
                                out.write('\t\tbase_type "{}";\n'.format(g.name))
                out.write("\t};\n};\n")
 
        for a in ["strength", "dexterity", "intelligence"]:
-               out.write('category "skillgem.{}"\n'.format(a))
+               out.write('category "skillgem.attribute.{}"\n'.format(a))
                out.write('{\n\tclass "Skill Gem";\n\tor\n\t{\n')
                for g in gems:
                        if g.primary_attribute==a: