grants: base/volume stats

company.py

@@ -21,7 +21,7 @@ def main() -> None:
 	print()
 
 	experts = mat_expertise()
-	company_report = integration.pmmg_monthly_report()[company['CompanyId']]
+	company_report = integration.pmmg_monthly_report()['individual'][company['CompanyId']]
 	report_by_expertise: dict[str | None, dict[str, int]] = collections.defaultdict(dict)
 	for mat, production in company_report.items():
 		expertise = experts.get(mat)

integration.py

@@ -14,7 +14,7 @@ def main() -> None:
 	mat = sys.argv[1]
 
 	recipes: list[roi.Recipe] = cache.get('https://api.prunplanner.org/data/recipes/')
-	companies = pmmg_monthly_report()
+	companies = pmmg_monthly_report()['individual']
 
 	print(mat, '→')
 	wrought = (recipe for recipe in recipes
@@ -63,7 +63,12 @@ def main() -> None:
 			alt = f' ({recipe_count} recipes)'
 		print(f'{output_mat:3}: {total:8,.0f}{alt}')
 
-def pmmg_monthly_report() -> dict[str, dict[str, CompanyOutput]]:
+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"];
@@ -71,9 +76,14 @@ def pmmg_monthly_report() -> dict[str, dict[str, CompanyOutput]]:
 	assert match
 	months_str = match.group(1)
 	months = [m.strip().strip('"') for m in months_str.split(',')]
-	last_month = months[-1]
-	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)['individual']
+	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

mat_competitors.py

@@ -43,7 +43,7 @@ def main() -> None:
 		for company_id, d in cache.get('https://prun.raylu.net/stats/data/knownCompanies.json', expiry=cache.ONE_DAY).items()}
 
 	competitors: list[Competitor] = []
-	for company_id, co_production in integration.pmmg_monthly_report().items():
+	for company_id, co_production in integration.pmmg_monthly_report()['individual'].items():
 		if (mat_production := co_production.get(ticker)) is None:
 			continue
 		if planet_ids := coid_bases.get(company_id):

punoted/grants.py

@@ -1,10 +1,31 @@
 import sys
+import typing
 
 import httpx
 
+import cache
 from config import config
+import integration
 
 def main() -> None:
+	stats_month = integration.pmmg_month()
+	coid_totals = integration.pmmg_monthly_report()['totals']
+	bases = cache.get(f'https://prun.raylu.net/stats/data/base-data-{stats_month}.json', expiry=cache.ONE_DAY)
+
+	for co in iter_grants():
+		if co is None:
+			print('\n')
+			continue
+		company = cache.get('https://rest.fnar.net/company/code/' + co)
+		co_id = company['CompanyId']
+		username = company['UserName']
+		try:
+			url = 'https://prun.raylu.net/stats/?type=compTotals&chartType=treemap&metric=volume&group=company&companyName=' + username
+			print(f'\t{bases[co_id]["bases"]} bases\t{coid_totals[co_id]["volume"]:,.0f} volume\t{url}\n')
+		except KeyError:
+			print(f'\tno stats for {username}\n')
+
+def iter_grants() -> typing.Iterator[str | None]:
 	(username,) = sys.argv[1:]
 	params = {'username': username, 'status': 'TERMINATED', 'limit': 200, 'page': 1}
 	while True:
@@ -27,6 +48,7 @@ def main() -> None:
 			if payment is not None and payment['Status'] == 'FULFILLED':
 				print(f'{payment["Amount"]:9,} {payment["Currency"]}', contract['Status'], contract['Timestamp'],
 						contract['PartnerCode'], contract['PartnerName'])
+				yield contract['PartnerCode']
 		if len(data['Contracts']) < 200:
 			break
 		params['page'] += 1