init

tracker.py

@@ -0,0 +1,119 @@
+import asyncio
+
+from typing import Dict
+from contextlib import AbstractAsyncContextManager
+from pathlib import Path
+
+from playwright.async_api import async_playwright
+from playwright.async_api import Playwright, Browser, Page
+from playwright.async_api import Response
+
+from imagehash import ImageHash, phash
+
+
+class Tracker(AbstractAsyncContextManager):
+    playwright: Playwright
+    browser: Browser
+
+    pages: Dict[str, Page] = {}
+    pages_lock = asyncio.Lock()
+    pages_active_lock = asyncio.Lock()
+
+    drawing_hashes: Dict[str, ImageHash] = {}
+
+
+    def __init__(self, *browser_args, **browser_kwargs):
+        self.browser_args = browser_args
+        self.browser_kwargs = browser_kwargs
+
+    async def __aenter__(self):
+        self.playwright = await async_playwright().start()
+        self.browser = await self.playwright.firefox.launch(
+            *self.browser_args, 
+            **self.browser_kwargs
+        )
+        return self
+
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        await self.playwright.stop()
+
+
+    async def open_drawing(self, drawing_id: str) -> Page:
+        async with self.pages_lock:
+            if drawing_id in self.pages:
+                if not self.pages[drawing_id].is_closed():
+                    return self.pages[drawing_id]
+
+                await self.pages[drawing_id].close()
+                del self.pages[drawing_id]
+
+            self.pages[drawing_id] = await self.browser.new_page()
+            print(f'open new tab for drawing#{drawing_id}')
+
+        # 앵귤러 프로젝트라 외부에서 데이터 긁어오기 쉽지 않음
+        # `POST /open-socket` 요청으로부터 현 프로젝트의 모든 이미지 아이디 긁어올 예정
+        socket_response = asyncio.Queue()
+        socket_response_count = 0
+
+        async def on_response(response: Response):
+            nonlocal socket_response_count
+            
+            if response.request.method != 'POST':
+                return
+            if not response.request.url.endswith('/open-socket'):
+                return
+
+            # 첫 데이터는 자신만을 담고 있기 때문에 두번째 데이터를 가져와야함
+            socket_response_count += 1
+            if socket_response_count < 2:
+                return
+            
+            drawing_ids = response.request.post_data_json['ids']
+            await asyncio.gather(*[
+                self.open_drawing(id) 
+                for id in drawing_ids
+            ])
+
+            await socket_response.put(drawing_ids)
+
+
+        page = self.pages[drawing_id]
+        page.on('response', on_response)
+
+        await page.goto(f'https://magma.com/d/{drawing_id}')
+
+        # 로그인 페이지가 나를 괴롭혀서 마음이 아파요
+        await page.add_style_tag(content='modals-box { display: none !important }')
+
+        # 현 프로젝트의 모든 그림 아이디 가져올 때까지 대기하기
+        await socket_response.get()
+
+        return page
+
+    async def download(
+        self, 
+        drawing_id: str,
+        path_to_save: Path = None
+    ):
+        page = await self.open_drawing(drawing_id)
+
+        previous_hash = self.drawing_hashes.get(drawing_id)
+
+        async with page.expect_download() as event:
+            # 파이어폭스는 활성화된 창에서만 Hover -> Click 이 가능함
+            async with self.pages_active_lock:
+                await page.click('button[aria-label=File]')
+                await page.hover('.dropdown-submenu:has(button[command=save-psd])')
+                await page.click('button[command=save-psd]')
+
+            print(f'downloading drawing#{drawing_id}')
+
+            file = await event.value
+            
+            path = await file.path()
+
+            if path_to_save:
+                path.rename(path_to_save)
+                return path_to_save
+            
+            return path