mirror of
https://github.com/EDCD/EDMarketConnector.git
synced 2025-04-04 19:40:02 +03:00
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""Export list of ships as CSV."""
|
|
import csv
|
|
|
|
import companion
|
|
from edmc_data import ship_name_map
|
|
|
|
|
|
def export(data: companion.CAPIData, filename: str) -> None:
|
|
"""
|
|
Write shipyard data in Companion API JSON format.
|
|
|
|
:param data: The CAPI data.
|
|
:param filename: Optional filename to write to.
|
|
:return:
|
|
"""
|
|
assert data['lastSystem'].get('name')
|
|
assert data['lastStarport'].get('name')
|
|
assert data['lastStarport'].get('ships')
|
|
|
|
with open(filename, 'w', newline='') as f:
|
|
c = csv.writer(f)
|
|
c.writerow(('System', 'Station', 'Ship', 'FDevID', 'Date'))
|
|
|
|
for (name, fdevid) in [
|
|
(
|
|
ship_name_map.get(ship['name'].lower(), ship['name']),
|
|
ship['id']
|
|
) for ship in list(
|
|
(data['lastStarport']['ships'].get('shipyard_list') or {}).values()
|
|
) + data['lastStarport']['ships'].get('unavailable_list')
|
|
]:
|
|
c.writerow((
|
|
data['lastSystem']['name'], data['lastStarport']['name'],
|
|
name, fdevid, data['timestamp']
|
|
))
|