Create app/enka_logic.py

app/enka_logic.py

@@ -0,0 +1,106 @@
+import asyncio
+import requests
+import time
+from playwright.async_api import async_playwright
+
+FREEIMAGE_API_KEY = "6d207e02198a847aa98d0a2a901485a5"
+FREEIMAGE_URL = "https://freeimage.host/api/1/upload"
+
+class EnkaApp:
+    def __init__(self):
+        self.playwright = None
+        self.browser = None
+
+    async def start(self):
+        self.playwright = await async_playwright().start()
+        self.browser = await self.playwright.chromium.launch(
+            headless=True,
+            args=['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage', '--disable-gpu']
+        )
+        print("🚀 [System] Browser logic started.")
+
+    async def stop(self):
+        if self.browser: await self.browser.close()
+        if self.playwright: await self.playwright.stop()
+
+    async def get_context_page(self, fast_mode=False):
+        context = await self.browser.new_context(
+            viewport={'width': 1280, 'height': 800},
+            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
+        )
+        page = await context.new_page()
+        if fast_mode:
+            await page.route("**/*", lambda route: route.abort() if route.request.resource_type in ["media", "font"] else route.continue_())
+        return context, page
+
+    async def fetch_info(self, uid: str):
+        start_time = time.time()
+        context, page = await self.get_context_page(fast_mode=True)
+        try:
+            await page.goto(f"https://enka.network/u/{uid}/", wait_until="domcontentloaded", timeout=30000)
+            try:
+                await page.click('button[data-icon="cards"]', timeout=1500)
+                await page.wait_for_selector('.Loda .bg', timeout=1500)
+            except: pass
+            
+            data = await page.evaluate("""() => {
+                const arText = document.querySelector('.ar')?.innerText || "";
+                const stats = Array.from(document.querySelectorAll('tr.stat')).map(tr => {
+                    const style = tr.querySelector('td.icon')?.getAttribute('style') || "";
+                    const iconMatch = style.match(/url\\(["']?(.*?)["']?\\)/);
+                    return {
+                        value: tr.querySelector('td:first-child')?.innerText.trim(),
+                        label: tr.querySelector('td:last-child')?.innerText.trim(),
+                        icon: iconMatch ? "https://enka.network" + iconMatch[1].replace(/\\\\/g, "") : null
+                    };
+                });
+                const ncBg = document.querySelector('.Loda .bg')?.style.backgroundImage;
+                const ncMatch = ncBg ? ncBg.match(/url\\(["']?(.*?)["']?\\)/) : null;
+                const characters = Array.from(document.querySelectorAll('.CharacterList .avatar')).map(el => {
+                    const style = el.querySelector('figure.chara')?.getAttribute('style') || "";
+                    const path = style.match(/url\\((.*)\\)/)?.[1].replace(/["']/g, "");
+                    return { name: path?.split('_').pop().split('.')[0], level: el.querySelector('.level')?.innerText, icon: "https://enka.network" + path };
+                }).filter(i => i.name);
+                return {
+                    player: {
+                        name: document.querySelector('.details h1')?.innerText,
+                        signature: document.querySelector('.signature')?.innerText,
+                        ar: arText.match(/AR\\s*(\\d+)/i)?.[1], wl: arText.match(/WL\\s*(\\d+)/i)?.[1],
+                        avatar: "https://enka.network" + document.querySelector('.avatar-icon img')?.getAttribute('src'),
+                        namecard: ncMatch ? "https://enka.network" + ncMatch[1] : null,
+                        stats: stats
+                    },
+                    characters: characters
+                };
+            }""")
+            await context.close()
+            return {"execution_time": f"{round(time.time() - start_time, 2)}s", "uid": uid, **data}
+        except Exception as e:
+            await context.close()
+            return {"error": str(e)}
+
+    async def fetch_gen(self, uid: str, char: str = None):
+        start_time = time.time()
+        context, page = await self.get_context_page(fast_mode=False)
+        try:
+            await page.goto(f"https://enka.network/u/{uid}/", wait_until="domcontentloaded", timeout=30000)
+            if char:
+                target = char.strip().capitalize()
+                try:
+                    await page.click(f"//div[contains(@class, 'avatar')]//figure[contains(@style, '{target}')]", timeout=3000)
+                    await asyncio.sleep(0.4)
+                except: pass
+            await page.click('button[data-icon="image"]')
+            await page.wait_for_selector('.Loda img[src^="blob:"]', timeout=15000)
+            img_base64 = await page.evaluate("""async () => {
+                const img = document.querySelector('.Loda img');
+                const res = await fetch(img.src);
+                const blob = await res.blob();
+                return new Promise(r => { const rd = new FileReader(); rd.onloadend = () => r(rd.result.split(',')[1]); rd.readAsDataURL(blob); });
+            }""")
+            await context.close()
+            res = requests.post(FREEIMAGE_URL, data={"key": FREEIMAGE_API_KEY, "action": "upload", "source": img_base64, "format": "json"}, timeout=15)
+            return {"execution_time": f"{round(time.time() - start_time, 2)}s", "uid": uid, "card_url": res.json().get('image', {}).get('url')}
+        except Exception as e:
+            await context.close()
+            return {"error": str(e)}