diff options
Diffstat (limited to 'scrap.py')
-rw-r--r-- | scrap.py | 20 |
1 files changed, 20 insertions, 0 deletions
@@ -53,6 +53,25 @@ def get_locations(snapshot): return solar_systems +def get_names(snapshot): + fields = ['alliance_id', 'character_id', 'corporation_id', 'faction_id'] + misc = set() + alliances = set() + corporations = set() + for km in snapshot['killmails']: + victim = km['victim'] + for field in fields: + if field in victim and victim[field] > 0: + misc.add(victim[field]) + if 'alliance_id' in victim: + alliances.add(victim['alliance_id']) + corporations.add(victim['corporation_id']) + names = esi.names(json=list(misc)) + names += [{'id': x, **esi.corporation(x)} for x in corporations] + names += [{'id': x, **esi.alliance(x)} for x in alliances] + return names + + def output_name(args): """ Generates name of the output file based on the CLI *args*. @@ -76,6 +95,7 @@ def main(): expand_details(snapshot) snapshot['types'] = get_types(snapshot) snapshot['locations'] = get_locations(snapshot) + snapshot['names'] = get_names(snapshot) filename = output_name(args) with open(filename, 'w') as fd: opts = {'indent': 4} if args.pretty else {} |