82 lines
3.3 KiB
Python
82 lines
3.3 KiB
Python
import requests
|
|
import pandas as pd
|
|
import os
|
|
import time
|
|
|
|
# Function to fetch card details from Scryfall using the set code and collector number
|
|
def fetch_card_images_by_set_and_number(set_code, collector_number):
|
|
url = f"https://api.scryfall.com/cards/{set_code}/{collector_number}"
|
|
response = requests.get(url)
|
|
|
|
if response.status_code == 200:
|
|
card_data = response.json()
|
|
images = []
|
|
|
|
# Handle single-sided or multi-sided cards
|
|
if 'image_uris' in card_data:
|
|
images.append((card_data['image_uris']['art_crop'], card_data['name']))
|
|
if 'card_faces' in card_data:
|
|
for face in card_data['card_faces']:
|
|
if 'image_uris' in face:
|
|
images.append((face['image_uris']['art_crop'], face['name']))
|
|
return images
|
|
else:
|
|
print(f"Error fetching card with set code {set_code} and collector number {collector_number}: {response.status_code}")
|
|
return None
|
|
|
|
# Function to download and save an image
|
|
def download_image(image_url, save_path):
|
|
response = requests.get(image_url)
|
|
if response.status_code == 200:
|
|
with open(save_path, 'wb') as file:
|
|
file.write(response.content)
|
|
else:
|
|
print(f"Failed to download image from {image_url}")
|
|
|
|
# Function to process the CSV and download images
|
|
def download_images_from_csv(csv_path, output_folder):
|
|
# Create output directory if it doesn't exist
|
|
if not os.path.exists(output_folder):
|
|
os.makedirs(output_folder)
|
|
|
|
# Read the CSV file
|
|
df = pd.read_csv(csv_path)
|
|
|
|
# Assuming the CSV has columns 'Card Name', 'Set Code', and 'Collector Number'
|
|
for index, row in df.iterrows():
|
|
card_name = row['Card Name']
|
|
set_code = row['Set Code']
|
|
collector_number = row['Collector Number']
|
|
|
|
# Naming convention: set_code_card_name_face_name.jpg
|
|
image_base_name = f"{set_code}_{card_name.replace('/', '-').replace(':', '').replace(' ', '-').replace(',', '-')}"
|
|
|
|
print(f"Processing {card_name} from set {set_code} with collector number {collector_number}...")
|
|
|
|
images = fetch_card_images_by_set_and_number(set_code, collector_number)
|
|
if images:
|
|
for image_url, face_name in images:
|
|
# Sanitize file name by removing invalid characters
|
|
image_file_name = f"{image_base_name}_art_crop.jpg"
|
|
save_path = os.path.join(output_folder, image_file_name)
|
|
|
|
# Skip downloading if the image already exists
|
|
if os.path.exists(save_path):
|
|
print(f"Image already exists for {card_name} ({face_name}), skipping download.")
|
|
continue
|
|
|
|
download_image(image_url, save_path)
|
|
time.sleep(0.1) # Sleep to avoid hitting API rate limits
|
|
else:
|
|
print(f"No images found for {card_name} from set {set_code} with collector number {collector_number}")
|
|
|
|
# Main execution
|
|
if __name__ == "__main__":
|
|
# Define the path to the input CSV and output folder
|
|
csv_path = 'cards.csv' # Replace with your CSV file path
|
|
output_folder = 'artcrop_images'
|
|
|
|
# Start the download process
|
|
download_images_from_csv(csv_path, output_folder)
|
|
print("Download completed.")
|