From 17beb084cde24b194adc52074ce9c214829da8c7 Mon Sep 17 00:00:00 2001
From: Dan Niemitalo <dniemitalo@linnmar.k12.ia.us>
Date: Sun, 29 Nov 2020 19:49:21 -0600
Subject: [PATCH] Add function to request csv from CubeCobra

Add function to request csv from CubeCobra
---
 cube_json.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/cube_json.py b/cube_json.py
index 0ab467b..d58e4aa 100644
--- a/cube_json.py
+++ b/cube_json.py
@@ -1,9 +1,17 @@
 import csv
 import json
+import requests
 
 from word_count import EXPECTED_HEADER, COLUMNS, TEMP_JSON, is_actual_mtg_card
 
 
+def request_cube_csv(cube_name, cube_id):
+    url = f'https://cubecobra.com/cube/download/csv/{cube_id}'
+    url += '?primary=Color%20Category&secondary=Types-Multicolor&tertiary=CMC2'
+    response = requests.get(url)
+    open(f'cubecsvs/{cube_name}.csv', 'wb').write(response.content)
+
+
 def load_cube_from_csv(filename, tag_filter=None):
     cards = []
     with open(filename) as f: