pruncalc/mat_competitors.py

from __future__ import annotations

import collections
import dataclasses
import json
import sys
import typing

import cache
import company
import integration
import planet_bases

def main() -> None:
	cx, ticker = sys.argv[1:]

	(expertise,) = iter_expertise(ticker)

	with open('www/closest.json') as f:
		planet_ids = {planet_id for planet_id, closest_cx in json.load(f).items() if closest_cx == cx}

	planets = []
	for planet, cogc in company.iter_planet_cogc():
		if planet['PlanetId'] in planet_ids and cogc == expertise:
			planets.append(planet['PlanetName'])
	print(len(planets), expertise, 'planets near', cx)

	coid_code_name: dict[str, tuple[str, str]] = {}
	coid_bases = collections.defaultdict(list)
	for planet in planets:
		print('\t' + planet)
		bases = planet_bases.get_bases(planet)
		for base in bases:
			if (code := base['OwnerCode']) is None:
				continue
			coid_code_name[base['OwnerId']] = code, base['OwnerName']
			coid_bases[base['OwnerId']].append(planet)

	coid_users: dict[str, str] = {company_id: d['Username']
		for company_id, d in cache.get('https://pmmg-products.github.io/reports/data/knownCompanies.json', expiry=cache.ONE_DAY).items()}

	competitors: list[Competitor] = []
	for company_id, co_production in integration.pmmg_monthly_report().items():
		if (mat_production := co_production.get(ticker)) is None:
			continue
		if bases := coid_bases.get(company_id):
			code, co_name = coid_code_name[company_id]
			username = coid_users[company_id]
			competitors.append(Competitor(code, co_name, username, mat_production['amount'], bases))
	competitors.sort(reverse=True)

	total = 0.0
	for c in competitors:
		print(f'{c.code:4} {c.company_name:30} {c.username:20} {c.production:9,.1f}', ', '.join(c.bases))
		total += c.production
	print(f'total: {total:,.1f}')

def iter_expertise(ticker: str) -> typing.Iterator[str]:
	buildings: typing.Sequence[company.Building] = cache.get('https://rest.fnar.net/building/allbuildings', expiry=cache.ONE_DAY)
	for building in buildings:
		for recipe in building['Recipes']:
			for output in recipe['Outputs']:
				if output['CommodityTicker'] == ticker:
					print(ticker, 'requires', building['Expertise'])
					yield building['Expertise']

@dataclasses.dataclass(eq=False, frozen=True, slots=True)
class Competitor:
	code: str
	company_name: str
	username: str
	production: float
	bases: typing.Sequence[str]

	def __lt__(self, other: Competitor) -> bool:
		return self.production < other.production

if __name__ == '__main__':
	main()