From 97508385ab53249a0a8a524d05f1dbe259d9300e Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Mon, 27 Aug 2018 20:46:13 +0000
Subject: [PATCH] bounding box for mana symbol - only tested for RTR set

---
 fetch_data.py |    9 +++++++--
 1 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/fetch_data.py b/fetch_data.py
index 6fc526a..2be9fe6 100644
--- a/fetch_data.py
+++ b/fetch_data.py
@@ -32,6 +32,11 @@
     return df
 
 
+def load_all_cards_text(csv_name):
+    df = pd.read_csv(csv_name, sep=';')
+    return df
+
+
 # Pulled from Django framework (https://github.com/django/django/blob/master/django/utils/text.py)
 def get_valid_filename(s):
     """
@@ -50,7 +55,7 @@
     for ind, row in df.iterrows():
         png_url = row['image_uris'][size]
         if out_dir == '':
-            out_dir = 'data/png/%s' % row['set']
+            out_dir = 'data/%s/%s' % (size, row['set'])
         if not os.path.exists(out_dir):
             os.makedirs(out_dir)
         img_name = '%s/%s_%s.png' % (out_dir, row['collector_number'], get_valid_filename(row['name']))
@@ -62,7 +67,7 @@
 def main():
     df = fetch_all_cards_text(url='https://api.scryfall.com/cards/search?q=layout:normal+set:rtr+lang:en',
                               csv_name='data/all_cards.csv')
-    fetch_cards_image(df)
+    #fetch_cards_image(df)
     pass
 
 

--
Gitblit v1.10.0