pruncalc/buy.py

from __future__ import annotations

import collections
import dataclasses
import typing

import cache
from config import config
import supply

if typing.TYPE_CHECKING:
	import market

def main() -> None:
	raw_prices: typing.Mapping[str, market.RawPrice] = {p['MaterialTicker']: p
			for p in cache.get('https://refined-prun.github.io/refined-prices/all.json') if p['ExchangeCode'] == 'IC1'}

	# what we need to buy
	planets = [supply.Planet(fio_burn) for fio_burn in cache.get('https://rest.fnar.net/fioweb/burn/user/' + config.username,
			headers={'Authorization': config.fio_api_key})]
	buy: dict[str, int] = collections.defaultdict(int)
	for planet in planets:
		for mat, amount in planet.buy_for_target(7).items():
			buy[mat] += amount

	# what we have
	warehouses: typing.Sequence[market.Warehouse] = cache.get('https://rest.fnar.net/sites/warehouses/' + config.username,
			headers={'Authorization': config.fio_api_key})
	for warehouse in warehouses:
		if warehouse['LocationNaturalId'] == 'HRT':
			storage: market.Storage = cache.get(f'https://rest.fnar.net/storage/{config.username}/{warehouse["StoreId"]}',
				headers={'Authorization': config.fio_api_key})
			assert storage['Type'] == 'WAREHOUSE_STORE'
			warehouse = {item['MaterialTicker']: item['MaterialAmount'] for item in storage['StorageItems']}
			break
	else:
		raise Exception("couldn't find HRT warehouse")

	# what we already are bidding for
	orders: typing.Sequence[market.ExchangeOrder] = cache.get('https://rest.fnar.net/cxos/' + config.username,
			headers={'Authorization': config.fio_api_key})
	bids = collections.defaultdict(int)
	for order in orders:
		if order['OrderType'] == 'SELLING' or order['Status'] == 'FILLED' or order['ExchangeCode'] != 'IC1':
			continue
		bids[order['MaterialTicker']] += order['Amount']

	# what's left to buy
	materials: list[Material] = []
	for mat, amount in buy.items():
		remaining = amount - bids[mat] - warehouse.get(mat, 0)
		if remaining <= 0:
			continue
		price = raw_prices[mat]
		if price['Bid'] is None or price['Ask'] is None:
			print(mat, 'has no bid/ask')
			continue
		spread = price['Ask'] - price['Bid']
		materials.append(Material(mat, amount=amount, bids=bids[mat], warehouse=warehouse.get(mat, 0),
				spread=spread, savings=spread * remaining))
	materials.sort(reverse=True)

	print('mat   want  bids  have   buy  savings')
	for m in materials:
		print(f'{m.ticker:4} {m.amount:>5} {m.bids:>5} {m.warehouse:>5} {m.amount - m.bids - m.warehouse:>5} {m.savings:8.0f}')

@dataclasses.dataclass(eq=False, slots=True)
class Material:
	ticker: str
	amount: int
	bids: int
	warehouse: int
	spread: float
	savings: float

	def __lt__(self, o: Material) -> bool:
		return self.savings< o.savings


if __name__ == '__main__':
	main()