Fix some scripts

This commit is contained in:
henriquenakashima 2021-07-29 00:31:34 -04:00
parent 4c4faf054d
commit 6cfedaa7d3
3 changed files with 8 additions and 2 deletions

View file

@ -45,6 +45,7 @@ _COLOR_CATEGORIES = {
'Red': 'r', 'Red': 'r',
'Green': 'g', 'Green': 'g',
'Multicolor': 'm', 'Multicolor': 'm',
'Multicolored': 'm',
'Hybrid': 'h', 'Hybrid': 'h',
'Colorless': 'c', 'Colorless': 'c',
'Land': 'land', 'Land': 'land',

View file

@ -28,7 +28,7 @@ def get_copies_to_add(card):
if tag == 'uncommon': if tag == 'uncommon':
return [card] * (2 - 1) return [card] * (2 - 1)
if tag == 'dual': if tag == 'dual':
return [card] * (7 - 1) return [card] * (8 - 1)
return [] return []
@ -37,9 +37,14 @@ def main():
cube: List[CubeCard] = cubecobra_csv.load_cube(csv_path) cube: List[CubeCard] = cubecobra_csv.load_cube(csv_path)
names_to_add = [] names_to_add = []
unique_cards = set()
for card in cube: for card in cube:
copies_to_add = get_copies_to_add(card) copies_to_add = get_copies_to_add(card)
names_to_add.extend([card.name for card in copies_to_add]) names_to_add.extend([card.name for card in copies_to_add])
unique_cards.add(card.name)
print(f'Unique cards: {len(unique_cards)}')
print(f'Total cards: {len(cube)}')
with open(OUTPUT_FILENAME, 'wb') as f: with open(OUTPUT_FILENAME, 'wb') as f:
full_pool_contents = '\n'.join(names_to_add).encode('utf-8') full_pool_contents = '\n'.join(names_to_add).encode('utf-8')

View file

@ -170,7 +170,7 @@ if __name__ == '__main__':
# 1. If you have your cube in Cube Cobra and want to filter per tag # 1. If you have your cube in Cube Cobra and want to filter per tag
# csv_path = cubecobra_csv.request_cube_csv('TheElegantCube_fetched', 'elegant') # csv_path = cubecobra_csv.request_cube_csv('TheElegantCube_fetched', 'elegant')
# cube_list = cube_json.load_cube_from_csv(csv_path) # cube_list = cubecobra_csv.load_cube_names(csv_path)
# 2. If you have your cube in Cube Cobra and want to consider only cards with a certain tag # 2. If you have your cube in Cube Cobra and want to consider only cards with a certain tag
csv_path = cubecobra_csv.request_cube_csv('TheElegantCube_fetched', 'elegant') csv_path = cubecobra_csv.request_cube_csv('TheElegantCube_fetched', 'elegant')