From be11e02f1d08beb40fff7d54d1c7a343ba10b275 Mon Sep 17 00:00:00 2001
From: Constantin Wenger <constantin.wenger@googlemail.com>
Date: Tue, 01 Feb 2022 22:46:28 +0000
Subject: [PATCH] added autocreating csv folder
---
fetch_data.py | 12 ++++++++++--
1 files changed, 10 insertions(+), 2 deletions(-)
diff --git a/fetch_data.py b/fetch_data.py
index 500dc55..a43cb68 100644
--- a/fetch_data.py
+++ b/fetch_data.py
@@ -100,7 +100,10 @@
:return:
"""
if out_dir is None:
- out_dir = '%s/card_img/%s/%s' % (Config.data_dir, size, row['set'])
+ s = row['set']
+ if s == 'con':
+ s = 'con__'
+ out_dir = '%s/card_img/%s/%s' % (Config.data_dir, size, s)
if not os.path.exists(out_dir):
os.makedirs(out_dir)
@@ -132,8 +135,13 @@
def main():
# Query card data by each set, then merge them together
+ if not os.path.exists(os.path.join(Config.data_dir, 'csv')):
+ os.mkdir(os.path.join(Config.data_dir, 'csv'))
for set_name in Config.all_set_list:
- csv_name = '%s/csv/%s.csv' % (Config.data_dir, set_name)
+ set_name_m = set_name
+ if set_name_m == 'con':
+ set_name_m = 'con__'
+ csv_name = '%s/csv/%s.csv' % (Config.data_dir, set_name_m)
print(csv_name)
if not os.path.isfile(csv_name):
df = fetch_all_cards_text(url='https://api.scryfall.com/cards/search?q=set:%s+lang:en' % set_name,
--
Gitblit v1.10.0