Create main.py

main.py

@@ -0,0 +1,138 @@
+import uvicorn
+from fastapi import FastAPI, Query
+from genshin import Client
+import asyncio
+import os
+
+app = FastAPI()
+
+cookies = {"ltuid_v2": os.getenv("ltuid_v2"), "ltoken_v2": os.getenv("ltoken_v2"),"ltmid_v2":os.getenv("ltmid_v2"),"cookie_token_v2":os.getenv("cookie_token_v2")}
+client = Client(cookies)
+
+# Helper function to format data
+def format_user_data(user):
+    try:
+        # Info section
+        info = f"• Info •\n"
+        info += f"{getattr(user.stats, 'days_active', 'N/A')} Active Days\n"
+        info += f"{getattr(user.stats, 'achievements', 'N/A')} Achievements\n"
+        info += f"{len(user.characters) if hasattr(user, 'characters') and isinstance(user.characters, list) else 'N/A'} Characters\n"
+        info += f"{getattr(user.stats, 'unlocked_waypoints', 'N/A')} Waypoints\n"
+        info += f"{getattr(user.stats, 'unlocked_domains', 'N/A')} Domains\n"
+        info += f"{getattr(user.stats, 'dendroculi', 'N/A')} Dendroculus\n"
+        info += f"{getattr(user.stats, 'electroculi', 'N/A')} Electroculus\n"
+        info += f"{getattr(user.stats, 'anemoculi', 'N/A')} Anemoculus\n"
+        info += f"{getattr(user.stats, 'geoculi', 'N/A')} Geoculus\n"
+        info += f"Spiral Abyss {getattr(user.stats, 'spiral_abyss', 'N/A')}\n"
+
+        # Chests section
+        chests = f"• Chests •\n"
+        chests += f"{getattr(user.stats, 'remarkable_chests', 'N/A')} Remarkable\n"
+        chests += f"{getattr(user.stats, 'luxurious_chests', 'N/A')} Luxurious\n"
+        chests += f"{getattr(user.stats, 'precious_chests', 'N/A')} Precious\n"
+        chests += f"{getattr(user.stats, 'exquisite_chests', 'N/A')} Exquisite\n"
+        chests += f"{getattr(user.stats, 'common_chests', 'N/A')} Common\n"
+
+        # Exploration section
+        exploration = f"• Exploration •\n"
+        if hasattr(user, 'explorations') and isinstance(user.explorations, list):
+            for exploration_data in user.explorations:
+                percent = f"{getattr(exploration_data, 'raw_explored', 0) / 10.0:.1f}%"
+                exploration += f"{getattr(exploration_data, 'name', 'Unknown')} {percent}\n"
+                if hasattr(exploration_data, 'offerings') and isinstance(exploration_data.offerings, list):
+                    for offering in exploration_data.offerings:
+                        exploration += f"  {getattr(offering, 'name', 'Unknown')} level {getattr(offering, 'level', 'N/A')}\n"
+        else:
+            exploration += "No exploration data available.\n"
+
+        # Teapot section
+        teapot = f"• Teapot •\n"
+        if hasattr(user.teapot, 'realms') and isinstance(user.teapot.realms, list) and len(user.teapot.realms) > 0:
+            teapot += f"{getattr(user.teapot.realms[0], 'name', 'Unknown')}\n"
+        teapot += f"Level {getattr(user.teapot, 'level', 'N/A')} ({getattr(user.teapot, 'comfort_name', 'Unknown')})\n"
+        teapot += f"Comfort {getattr(user.teapot, 'comfort', 'N/A')}\n"
+        teapot += f"Visits {len(user.teapot.visitors) if hasattr(user.teapot, 'visitors') and isinstance(user.teapot.visitors, list) else 'N/A'}\n"
+        teapot += f"Items {getattr(user.teapot, 'items', 'N/A')}\n"
+
+        return f"{info}\n{chests}\n{exploration}\n{teapot}"
+    except Exception as e:
+        # Log unexpected issues in the formatting process
+        return f"Error formatting user data: {e}"
+
+@app.get("/user/{uid}")
+async def get_user_info(uid: int):
+    try:
+        # Fetch user data from the client
+        user = await client.get_genshin_user(uid)
+        # Format user data
+        formatted_data = format_user_data(user)
+        return {"uid": uid, "data": formatted_data}
+    except Exception as e:
+        # Log and return any errors encountered during API call or formatting
+        return {"error": str(e)}
+
+async def fetch_spiral_abyss_data(uid: int, previous: bool = False):
+    """Fetches Spiral Abyss data using the genshin API."""
+    return await client.get_spiral_abyss(uid, previous=previous)
+
+def format_abyss_data(data):
+    """Formats the Spiral Abyss data into the desired format."""
+
+    def format_rank(rank):
+        """Formats a rank entry with its value and character name."""
+        return f"{rank[0].value} ({rank[0].name})" if rank else "N/A"
+
+    return {
+        "Season": data.season,
+        "Start Time": data.start_time.isoformat(),
+        "End Time": data.end_time.isoformat(),
+        "Total Battles": data.total_battles,
+        "Total Wins": data.total_wins,
+        "Max Floor": data.max_floor,
+        "Total Stars": data.total_stars,
+        "Ranks": {
+            "Most Played": [f"{char.name} ({char.value})" for char in data.ranks.most_played],
+            "Most Kills": format_rank(data.ranks.most_kills),
+            "Strongest Strike": format_rank(data.ranks.strongest_strike),
+            "Most Damage Taken": format_rank(data.ranks.most_damage_taken),
+            "Most Bursts Used": format_rank(data.ranks.most_bursts_used),
+            "Most Skills Used": format_rank(data.ranks.most_skills_used),
+        },
+        "Floors": [
+            {
+                "Floor": floor.floor,
+                "Unlocked": floor.unlocked,
+                "Stars": f"{floor.stars}/{floor.max_stars}",
+                "Chambers": [
+                    {
+                        "Chamber": chamber.chamber,
+                        "Stars": f"{chamber.stars}/{chamber.max_stars}",
+                    }
+                    for chamber in floor.chambers
+                ],
+            }
+            for floor in data.floors
+        ],
+    }
+
+@app.get("/spiral-abyss/")
+async def get_spiral_abyss(
+    id: int = Query(...),
+    previous: bool = Query(False)
+):
+    """
+    Fetch and return Spiral Abyss data for the given user ID.
+
+    Query parameters:
+    - `id`: The user ID to fetch data for.
+    - `previous`: Whether to fetch the previous season's data (default: False).
+    """
+    try:
+        data = await fetch_spiral_abyss_data(id, previous)
+        formatted_data = format_abyss_data(data)
+        return formatted_data
+    except Exception as e:
+        return {"error": str(e)}
+# Run FastAPI app
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)