forgejoadmin/cubescripts
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make cubecobra_csv an API

Browse source
This commit is contained in:
henriquenakashima 2021-03-19 21:37:27 -04:00
parent b52958aa81
commit 067e849973
8 changed files with 109 additions and 118 deletions
Download patch file Download diff file Expand all files Collapse all files

70
cube_pool.py
View file

@ -8,80 +8,58 @@ from collections import defaultdict
from typing import Dict, List, Set
import cubecobra_csv
@dataclasses.dataclass
class Card:
name: str
color_category: str
colors: List[str]
tags: List[str]
def __post_init__(self):
assert self.color_category in 'wubrgchml0', f'{self}'
def __repr__(self):
return self.name
from cubecobra_csv import CubeCard
class CubePool:
def __init__(self):
self._categories: List[Card] = defaultdict(list)
self._categories: List[CubeCard] = defaultdict(list)
def add_card(self, card: Card):
def add_card(self, card: CubeCard):
self._categories[card.color_category].append(card)
def wide_sample(self, n: int) -> List[Card]:
def wide_sample(self, n: int) -> List[CubeCard]:
wide_iterator = list(itertools.chain.from_iterable(self._categories.values()))
sample = random.sample(wide_iterator, n)
return sample
def category_sample(self, color_category: str, n: int) -> List[Card]:
def category_sample(self, color_category: str, n: int) -> List[CubeCard]:
return random.sample(self._categories[color_category], n)
def draw_from_category(self, color_category: str, n: int) -> List[Card]:
def draw_from_category(self, color_category: str, n: int) -> List[CubeCard]:
drawn = self.category_sample(color_category, n)
self.remove_cards(drawn, color_category)
return drawn
def remove_cards(self, removed: List[Card], color_category: str):
def remove_cards(self, removed: List[CubeCard], color_category: str):
existing = self._categories[color_category]
self._categories[color_category] = [c for c in existing if not c in removed]
def fetch(self, card_name: str, color_category: str) -> Card:
found: List[Card] = list(filter(lambda c: c.name == card_name, self._categories[color_category]))
def fetch(self, card_name: str, color_category: str) -> CubeCard:
found: List[CubeCard] = list(filter(lambda c: c.name == card_name, self._categories[color_category]))
assert len(found) >= 1
card_taken: Card = found[0]
card_taken: CubeCard = found[0]
self.remove_cards([card_taken], color_category)
return card_taken
def get_category(self, color_category: str) -> List[Card]:
def get_category(self, color_category: str) -> List[CubeCard]:
return self._categories[color_category]
def load_pools(csv_path: str, pool_tags: Set[str]) -> Dict[str, CubePool]:
pools = defaultdict(CubePool)
with open(csv_path) as f:
reader = csv.reader(f)
header_line = next(reader)
cubecobra_csv.assert_header(header_line)
for line in reader:
card_name = cubecobra_csv.get_name(line)
tags = cubecobra_csv.get_tags(line)
# print(f'{card_name} {tags}')
pool_tags_for_card = pool_tags.intersection(tags)
if len(pool_tags_for_card) == 0:
print(f'{card_name} does not have any of the tags: {pool_tags}')
elif len(pool_tags_for_card) > 1:
print(f'{card_name} should only be tagged as one in {pool_tags} '
'but is tagged as {pool_tags_for_card}')
else:
(pool_tag,) = pool_tags_for_card
color_category = cubecobra_csv.get_color_category(line)
# print(f'{card_name} {color_category}')
colors = cubecobra_csv.get_colors(line)
# print(f'{card_name} {colors}')
card = Card(card_name, color_category, colors, tags)
pools[pool_tag].add_card(card)
cube: List[CubeCard] = cubecobra_csv.load_cube(csv_path)
for cube_card in cube:
pool_tags_for_card = pool_tags.intersection(cube_card.tags)
if len(pool_tags_for_card) == 0:
print(f'{cube_card.name} does not have any of the tags: {pool_tags}')
elif len(pool_tags_for_card) > 1:
print(f'{cube_card.name} should only be tagged as one in {pool_tags} '
'but is tagged as {pool_tags_for_card}')
else:
(pool_tag,) = pool_tags_for_card
pools[pool_tag].add_card(cube_card)
return pools