69 lines
2.7 KiB
Python
69 lines
2.7 KiB
Python
|
import csv
|
||
|
|
import os
|
||
|
|
import urllib.request
|
||
|
|
|
||
|
|
# Function to download images
|
||
|
|
def download_image(image_url, folder_path, file_name):
|
||
|
|
image_path = os.path.join(folder_path, file_name)
|
||
|
|
urllib.request.urlretrieve(image_url, image_path)
|
||
|
|
|
||
|
|
# List all .txt files in the Deck Lists folder
|
||
|
|
txt_files = [file for file in os.listdir('Deck Lists') if file.endswith('.txt')]
|
||
|
|
|
||
|
|
# Display the list of .txt files and prompt for selection
|
||
|
|
print("Available text files:")
|
||
|
|
for index, file in enumerate(txt_files, start=1):
|
||
|
|
print(f"{index}. {file}")
|
||
|
|
|
||
|
|
selection = input("Enter the number corresponding to the text file: ")
|
||
|
|
|
||
|
|
# Validate the user input and extract the selected text file name
|
||
|
|
try:
|
||
|
|
selection_index = int(selection)
|
||
|
|
if 1 <= selection_index <= len(txt_files):
|
||
|
|
text_file_name = txt_files[selection_index - 1]
|
||
|
|
|
||
|
|
# Extract the deck name from the text file name
|
||
|
|
deck_name = os.path.splitext(text_file_name)[0]
|
||
|
|
|
||
|
|
# Read CSV file and store image URLs and quantities
|
||
|
|
image_urls = {}
|
||
|
|
with open('data.csv', newline='', encoding='utf-8') as csvfile:
|
||
|
|
reader = csv.DictReader(csvfile)
|
||
|
|
for row in reader:
|
||
|
|
name_title = f"{row['name']} - {row['title']}" if row['title'] else row['name']
|
||
|
|
if name_title not in image_urls:
|
||
|
|
image_urls[name_title] = {'image': row['image'], 'qty': int(row['number'])}
|
||
|
|
|
||
|
|
# Create output folder using the deck name
|
||
|
|
output_folder = f'images/{deck_name}'
|
||
|
|
os.makedirs(output_folder, exist_ok=True)
|
||
|
|
|
||
|
|
# Read text file and organize images
|
||
|
|
with open(f'Deck Lists/{text_file_name}', 'r') as textfile:
|
||
|
|
for line in textfile:
|
||
|
|
line = line.strip()
|
||
|
|
parts = line.split(' - ')
|
||
|
|
if len(parts) >= 2:
|
||
|
|
quantity, name, title = parts[0].split()[0], ' '.join(parts[0].split()[1:]), parts[1]
|
||
|
|
else:
|
||
|
|
quantity, name = line.split()[0], ' '.join(line.split()[1:])
|
||
|
|
title = None # Set title as None for entries without titles
|
||
|
|
|
||
|
|
name_title = f"{name} - {title}" if title else name
|
||
|
|
|
||
|
|
card_info = image_urls.get(name_title)
|
||
|
|
|
||
|
|
if card_info:
|
||
|
|
qty_needed = int(quantity)
|
||
|
|
qty_available = card_info['qty']
|
||
|
|
|
||
|
|
# Download images based on quantity needed
|
||
|
|
for i in range(min(qty_needed, qty_available)):
|
||
|
|
file_name = f"{name_title} ({i + 1}).webp"
|
||
|
|
download_image(card_info['image'], output_folder, file_name)
|
||
|
|
else:
|
||
|
|
print("Invalid selection.")
|
||
|
|
except ValueError:
|
||
|
|
print("Invalid input. Please enter a number.")
|