feat(csec, csfc, csgc, cs4ac, cs4bc): fix CSEC, CSFC, CSGC, CS4aC, CS4bC missing abilities

11 months ago · 9985b5c6e6
parent 5f9ec9956a
commit 9985b5c6e6
3 changed files with 505 additions and 201 deletions
--- a/output/sets.json
+++ b/output/sets.json
--- a/output/sets_min.json
+++ b/output/sets_min.json
--- a/src/main.py
+++ b/src/main.py
@ -218,8 +218,15 @@ def get_pokemon_attr(card) -> PokemonAttr:
    stage = get_stage(card)
    hp = card['details']['hp']

-    ability = (card['details'].get('featureName', None), card['details'].get('featureText', None))
-    if ability[0] is not None:
+    if 'featureName' in card['details'] and 'featureText' in card['details']:
+        ability = (card['details'].get('featureName', None), card['details'].get('featureText', None))
+    # After CSEC, Raw data change it ability structure
+    elif 'cardFeatureItemList' in card['details']:
+        ability = (card['details']['cardFeatureItemList'][0]['featureName'],
+                   card['details']['cardFeatureItemList'][0]['featureDesc'])
+    else:
+        ability = None
+    if ability is not None:
        ability = Ability(ability[0], ability[1])
    else:
        ability = None