Create generate_draft_pool.py

generate_draft_pool.py

@@ -0,0 +1,57 @@
+# Generate a draft pool for dr4ft.info
+
+import csv
+import random
+
+
+OUTPUT_POOL_SIZE = 360
+CARDS_FROM_OCCASIONAL = 48
+CARDS_FROM_CORE = OUTPUT_POOL_SIZE - CARDS_FROM_OCCASIONAL
+
+CSV_FILENAME = 'TheElegantCube20201102.csv'
+OUTPUT_FILENAME = 'cards_in_draft.txt'
+EXPECTED_HEADER = [
+    'Name',
+    'CMC',
+    'Type',
+    'Color',
+    'Set',
+    'Collector Number',
+    'Rarity',
+    'Color Category',
+    'Status',
+    'Finish',
+    'Maybeboard',
+    'Image URL',
+    'Tags',
+    'Notes',
+    'MTGO ID'
+]
+COLUMNS = {column_name: i for i, column_name in enumerate(EXPECTED_HEADER)}
+
+
+def main():
+    core_pool = []
+    occasional_pool = []
+    with open(CSV_FILENAME) as f:
+        reader = csv.reader(f)
+        header_line = next(reader)
+        assert header_line == EXPECTED_HEADER
+        for line in reader:
+            card = line[COLUMNS['Name']]
+            tags = line[COLUMNS['Tags']].split(', ')
+            # print(f'{card} {tags}')
+            if 'occasional' in tags:
+                occasional_pool.append(card)
+            elif 'core' in tags:
+                core_pool.append(card)
+            else:
+                print('{card} is tagged as neither "core" nor "occasional"')
+
+    main_pool = random.sample(core_pool, CARDS_FROM_CORE) + random.sample(occasional_pool, CARDS_FROM_OCCASIONAL)
+    with open(OUTPUT_FILENAME, 'w') as f:
+        f.write('\n'.join(main_pool))
+
+
+if __name__ == '__main__':
+    main()