Merge pull request #2244 from IFRCGo/feature/debug-playwright

Add local Playwright debug

Author: szabozoltan69

api/tasks.py

@@ -12,9 +12,10 @@
 from playwright.sync_api import sync_playwright
 from rest_framework.authtoken.models import Token
 
-from api.models import Export
-
 from .logger import logger
+from .models import Export
+
+# from .utils import DebugPlaywright
 
 
 def build_storage_state(tmp_dir, user, token):
@@ -102,6 +103,7 @@ def generate_url(url, export_id, user, title):
                 )
                 context = browser.new_context(storage_state=storage_state)
                 page = context.new_page()
+                # DebugPlaywright.debug(page)
                 timeout = 300000
                 page.goto(url, timeout=timeout)
                 time.sleep(5)

api/utils.py

@@ -5,6 +5,36 @@
 from django.utils.translation import gettext
 
 
+class DebugPlaywright:
+    """Basic helpers to debug PlayWright issues locally"""
+
+    @staticmethod
+    def log_console(msg):
+        """Console logs"""
+        print("console:", msg.text)
+
+    @staticmethod
+    def log_request(request):
+        """Network request logs"""
+        # Add filter to remove noise: if request.url.startswith("http://api/v2"):
+        print("Network >>:", request.method, request.url)
+        print(" --- ", request.headers)
+
+    @staticmethod
+    def log_response(response):
+        """Network response logs"""
+        # Add filter to remove noise: if response.url.startswith("http://api/v2"):
+        print("Network <<:", response.status, response.url)
+        print(" --- ", response.headers)
+
+    @classmethod
+    def debug(cls, page):
+        """Add hook to receive logs from playwright"""
+        page.on("console", cls.log_console)
+        page.on("request", cls.log_request)
+        page.on("response", cls.log_response)
+
+
 def pretty_request(request):
     headers = ""
     for header, value in request.META.items():