pruncalc/integration.py

from __future__ import annotations

import collections
import re
import sys
import typing

import cache

if typing.TYPE_CHECKING:
	import roi

def main() -> None:
	mat = sys.argv[1]

	recipes: list[roi.Recipe] = cache.get('https://api.prunplanner.org/data/recipes/')
	companies = pmmg_monthly_report()['individual']

	print(mat, '→')
	wrought = (recipe for recipe in recipes
			if mat in (i['material_ticker'] for i in recipe['inputs']) and len(recipe['outputs']) == 1)
	output_mats = {}
	for recipe in sorted(wrought, key=lambda r: r['outputs'][0]['material_ticker']):
		(output,) = recipe['outputs']
		(input,) = (i for i in recipe['inputs'] if i['material_ticker'] == mat)
		ratio = input['material_amount'] / output['material_amount']
		output_mats[output['material_ticker']] = ratio
		print(f'\t{output["material_ticker"]:3}:', ratio)

	companies_produced = companies_consumed = companies_consumed_only = companies_consumed_own = companies_consumed_80_own = 0
	into = dict.fromkeys(output_mats.keys(), 0)
	for company in companies.values():
		if (co_production := company.get(mat)) is not None:
			companies_produced += 1
		consumed = 0
		for output_mat, per_run_consumption in output_mats.items():
			if co_production := company.get(output_mat):
				co_consumption = co_production['amount'] * per_run_consumption
				consumed += co_consumption
				into[output_mat] += co_consumption
		if consumed > 0:
			companies_consumed += 1
			if co_production is None:
				companies_consumed_only += 1
			else:
				companies_consumed_own += 1
				if consumed > 0.8 * co_production['amount']:
					companies_consumed_80_own += 1
	print(f'{companies_produced} companies producing')
	print(f'{companies_consumed} companies consuming')
	print(f'{companies_consumed_only} companies consuming without producing any')
	print(f'{companies_consumed_own} companies consuming their own production')
	print(f'{companies_consumed_80_own} companies consuming 80%+ of their own production')

	recipes_per_output = collections.defaultdict(int)
	for recipe in recipes:
		for output in recipe['outputs']:
			if output['material_ticker'] in into:
				recipes_per_output[output['material_ticker']] += 1
	total_consumed = 0
	for output_mat, consumed in sorted(into.items(), key=lambda kv: kv[1], reverse=True):
		alt = ''
		if (recipe_count := recipes_per_output[output_mat]) != 1:
			alt = f' ({recipe_count} recipes)'
		print(f'{output_mat:3}: {consumed:8,.0f}{alt}')
		total_consumed += consumed
	print('total consumed:', total_consumed)

def pmmg_monthly_report() -> CompanyData:
	last_month = pmmg_month()
	print('getting report for', last_month)
	return cache.get(f'https://prun.raylu.net/stats/data/company-data-{last_month}.json', expiry=cache.ONE_DAY)

def pmmg_month() -> str:
	report_constants = cache.get('https://git.raylu.net/raylu/prunstats/raw/main/src/staticData/constants.ts',
			json=False, expiry=cache.ONE_DAY)
	# export const months = ["mar25", "apr25", ..., "dec25", "jan26"];
	match = re.search(r'export const months = \[(.*?)\];', report_constants)
	assert match
	months_str = match.group(1)
	months = [m.strip().strip('"') for m in months_str.split(',')]
	return months[-1]

class CompanyData(typing.TypedDict):
	totals: dict[str, CompanyTotal]
	individual: dict[str, dict[str, CompanyOutput]]

class CompanyTotal(typing.TypedDict):
	volume: float

class CompanyOutput(typing.TypedDict):
	amount: int

if __name__ == '__main__':
	main()