Update petroll.py

petroll.py

@@ -3,6 +3,7 @@ from discord import app_commands
 import aiohttp
 import random
 import time
+import asyncio
 
 from cash import user_cash
 
@@ -12,7 +13,18 @@ luck_opportunities = {}
 used_luck_opportunities = set()
 first_luck_claimed = set()
 
-async def perform_roll(interaction: discord.Interaction):
+# Cache for pet data
+pet_cache = None
+pet_cache_expiry = 0
+CACHE_DURATION = 300  # 5 minutes
+
+async def fetch_and_process_pets():
+    global pet_cache, pet_cache_expiry
+    
+    current_time = time.time()
+    if pet_cache and current_time < pet_cache_expiry:
+        return pet_cache
+
     async def fetch_data(url):
         async with aiohttp.ClientSession() as session:
             async with session.get(url) as response:
@@ -20,8 +32,10 @@ async def perform_roll(interaction: discord.Interaction):
                     return await response.json()
                 return None
 
-    rap_data = await fetch_data("https://petsgo.biggamesapi.io/api/Rap")
-    collection_data = await fetch_data("https://petsgo.biggamesapi.io/api/collection/Pets")
+    rap_data, collection_data = await asyncio.gather(
+        fetch_data("https://petsgo.biggamesapi.io/api/Rap"),
+        fetch_data("https://petsgo.biggamesapi.io/api/collection/Pets")
+    )
 
     if not rap_data or not collection_data:
         return None
@@ -31,15 +45,30 @@ async def perform_roll(interaction: discord.Interaction):
     if not pets:
         return None
 
-    user_id = interaction.user.id
-    luck_multiplier = luck_multipliers.get(user_id, 1)
-    
     # Sort pets by difficulty (rarity)
     sorted_pets = sorted(pets, key=lambda x: x['configData']['difficulty'])
     
     # Divide pets into tiers
     tier_size = len(sorted_pets) // 5
     tiers = [sorted_pets[i:i+tier_size] for i in range(0, len(sorted_pets), tier_size)]
+
+    pet_cache = {
+        'tiers': tiers,
+        'rap_data': rap_data
+    }
+    pet_cache_expiry = current_time + CACHE_DURATION
+    return pet_cache
+
+async def perform_roll(interaction: discord.Interaction):
+    pet_data = await fetch_and_process_pets()
+    if not pet_data:
+        return None
+
+    tiers = pet_data['tiers']
+    rap_data = pet_data['rap_data']
+
+    user_id = interaction.user.id
+    luck_multiplier = luck_multipliers.get(user_id, 1)
     
     # Adjust probabilities based on luck
     tier_probabilities = [0.5, 0.25, 0.15, 0.07, 0.03]