You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
2.2 KiB
63 lines
2.2 KiB
import requests
|
|
from misskey.exceptions import MisskeyAPIException
|
|
from client import client_connection
|
|
from db_utils import get_db_connection
|
|
|
|
def add_character(name: str, rarity: int, weight: float, image_url: str) -> tuple[int, str]:
|
|
"""
|
|
Adds a character to the database, uploading the image from a public URL to the bot's Misskey Drive.
|
|
|
|
Args:
|
|
name (str): Character name.
|
|
rarity (int): Character rarity (e.g., 1-5).
|
|
weight (float): Pull weight (e.g., 0.02).
|
|
image_url (str): Public URL of the image from the post (e.g., from note['files'][i]['url']).
|
|
|
|
Returns:
|
|
tuple[int, str]: Character ID and bot's Drive file_id.
|
|
|
|
Raises:
|
|
ValueError: If inputs are invalid.
|
|
RuntimeError: If image download/upload or database operation fails.
|
|
"""
|
|
try:
|
|
# Validate inputs
|
|
if not name or not name.strip():
|
|
raise ValueError("Character name cannot be empty.")
|
|
if not isinstance(rarity, int) or rarity < 1:
|
|
raise ValueError("Rarity must be a positive integer.")
|
|
if not isinstance(weight, (int, float)) or weight <= 0:
|
|
raise ValueError("Weight must be a positive number.")
|
|
if not image_url:
|
|
raise ValueError("Image URL must be provided.")
|
|
|
|
# Download image
|
|
response = requests.get(image_url, stream=True)
|
|
if response.status_code != 200:
|
|
raise RuntimeError(f"Failed to download image from {image_url}")
|
|
|
|
# Upload to bot's Drive
|
|
mk = client_connection()
|
|
try:
|
|
media = mk.drive_files_create(response.raw)
|
|
file_id = media["id"]
|
|
except MisskeyAPIException as e:
|
|
raise RuntimeError(f"Failed to upload image to bot's Drive: {e}") from e
|
|
|
|
# Insert into database
|
|
conn = get_db_connection()
|
|
cur = conn.cursor()
|
|
cur.execute(
|
|
'INSERT INTO characters (name, rarity, weight, file_id) VALUES (?, ?, ?, ?)',
|
|
(name.strip(), rarity, float(weight), file_id)
|
|
)
|
|
conn.commit()
|
|
character_id = cur.lastrowid
|
|
|
|
return character_id, file_id
|
|
|
|
except Exception as e:
|
|
raise
|
|
finally:
|
|
if 'conn' in locals():
|
|
conn.close() |