blob: c430f36c97c1aca9047cf7b21a2c91275a19d7c7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
import argparse
import json
import re
import sys
import time
from salvager import esi
from salvager import zkill
SYSTEM_AND_DATE = re.compile(r"/(\d+)/(\d+)/?$")
def expand_hashes(snapshot):
"""
Expands killmails in *snapshot* IN PLACE by adding their hash based on information from Zkillboard.
"""
for killmail in snapshot['killmails']:
time.sleep(1.05) # Zkillboard is very sensitive.
killmail['hash'] = zkill.hash(killmail['killmail_id'])
return snapshot
def expand_details(snapshot):
"""
Expands killmails in *snapshot* IN PLACE by adding details from EVE ESI. Some data is dropped in process as e.g.,
full information on attackers is not important in context of the visualizations.
"""
for killmail in snapshot['killmails']:
details = esi.killmail(killmail['killmail_id'], killmail['hash'])
killmail.update(details)
return snapshot
def get_types(snapshot):
ships = {x['victim']['ship_type_id'] for x in snapshot['killmails']}
return [esi.type(x) for x in ships]
def get_locations(snapshot):
solar_system_ids = {x['solar_system_id'] for x in snapshot['killmails']}
solar_systems = [esi.system(x) for x in solar_system_ids]
constellation_ids = {x['constellation_id'] for x in solar_systems}
constellations = {x: esi.constellation(x) for x in constellation_ids}
region_ids = {x['region_id'] for x in constellations.values()}
regions = {x: esi.region(x) for x in region_ids}
for system in solar_systems:
constellation = constellations[system['constellation_id']]
region_id = constellation['region_id']
system['constellation_name'] = constellation['name']
system['region_name'] = regions[region_id]['name']
system['region_id'] = region_id
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*.
"""
if args.output:
return args.output
match = SYSTEM_AND_DATE.search(args.url)
if match:
return "{}_{}.json".format(*match.groups())
return "a.json"
def main():
parser = argparse.ArgumentParser()
parser.add_argument("url")
parser.add_argument("-o", "--output")
parser.add_argument("--pretty", action='store_true')
args = parser.parse_args()
snapshot = zkill.parse_battle_report(args.url)
expand_hashes(snapshot)
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 {}
fd.write(json.dumps(snapshot, **opts))
fd.write("\n")
if __name__ == "__main__":
main()
|
