diff --git a/.gitignore b/.gitignore index cac6ac4..4ef8d84 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,9 @@ +cards_in_draft.txt +oracle_cards.json +cube_temp.json +oracle_cards_clean.json +scryfall/ + # Byte-compiled / optimized / DLL files __pycache__/ *.py[cod] @@ -128,7 +134,4 @@ dmypy.json # Pyre type checker .pyre/ .idea/workspace.xml -cards_in_draft.txt -oracle_cards.json -cube_temp.json -oracle_cards_clean.json + diff --git a/analyze_all_set_keywords.py b/analyze_all_set_keywords.py index 472d008..e60336b 100644 --- a/analyze_all_set_keywords.py +++ b/analyze_all_set_keywords.py @@ -4,7 +4,7 @@ import keyword_stats if __name__ == '__main__': - f = open('set_codes.txt') + f = open('set_codes.csv') codes = [line.strip() for line in f.readlines()] output_file = open('set_keyword_frequency.csv', 'w+') output_file.write('Set,Keywords\n') diff --git a/scryfall_requests.py b/scryfall_requests.py index 8b7654d..2d537d3 100644 --- a/scryfall_requests.py +++ b/scryfall_requests.py @@ -1,3 +1,4 @@ +import os import requests import json import time @@ -30,11 +31,12 @@ def request_set(code): f.close() -sets = [] -set_file = open('set_codes.txt') -for line in set_file: - sets += [line.strip()] -set_file.close() +if __name__ == '__main__': + sets = [] + with open('set_codes.csv') as set_file: + for line in set_file: + sets += [line.strip()] -for code in sets: - request_set(code) + os.makedirs(os.path.join('scryfall', 'sets'), exist_ok=True) + for code in sets: + request_set(code)