pruncalc/punoted/storage.py

from __future__ import annotations
import sys
import typing

import httpx

from config import config

def main():
	items = frozenset(i.upper() for i in sys.argv[1:])
	storages: typing.Sequence[Storage] = httpx.get('https://api.punoted.net/v1/storages/user',
			headers={'X-Data-Token': config.punoted_api_key}).raise_for_status().json()

	for storage in storages:
		for item in storage['StorageItems']:
			if item['MaterialTicker'] in items:
				name = storage['Name'] or storage['Location']
				print(f'{name}: {item["MaterialAmount"]} {item["MaterialTicker"]}')

class Storage(typing.TypedDict):
	Name: str
	Location: str
	StorageItems: typing.Sequence[StorageItem]

class StorageItem(typing.TypedDict):
	MaterialTicker: str
	MaterialAmount: int

if __name__ == '__main__':
	main()