# Kemoverse - a gacha-style bot for the Fediverse. # Copyright © 2025 Waifu, Moon, VD15 and contributors. # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Affero General Public License as # published by the Free Software Foundation, either version 3 of the # License, or (at your option) any later version. # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Affero General Public License for more details. # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see https://www.gnu.org/licenses/. import requests import config from fediverse_factory import get_fediverse_service import db_utils def add_card(name: str, rarity: int, image_url: str) -> tuple[int, str]: """ Adds a card to the database, uploading the image from a public URL to the Fediverse instance. Args: name (str): Card name. rarity (int): Card rarity (e.g., 1-5). image_url (str): Public URL of the image from the post. Returns: tuple[int, str]: Card ID and file_id. Raises: ValueError: If inputs are invalid. RuntimeError: If image download/upload or database operation fails. """ stripped_name = name.strip() # Validate inputs if not stripped_name: raise ValueError('Card name cannot be empty.') if rarity < 1: raise ValueError('Rarity must be a positive integer.') if rarity not in config.RARITY_TO_WEIGHT.keys(): raise ValueError(f'Invalid rarity: {rarity}') if not image_url: raise ValueError('Image URL must be provided.') try: # Download image response = requests.get(image_url, stream=True, timeout=30) if response.status_code != 200: raise RuntimeError(f"Failed to download image from {image_url}") # Upload to Fediverse instance fediverse_service = get_fediverse_service(config.INSTANCE_TYPE) try: uploaded_file = fediverse_service.upload_file(response.raw) file_id = uploaded_file.id except RuntimeError as e: raise RuntimeError(f"Failed to upload image: {e}") from e # Insert into database using db_utils function card_id = db_utils.insert_card( stripped_name, rarity, file_id ) return card_id, file_id except Exception as e: raise