Add all dashboard projects

MacroinfManage_MES/tools/screenshot_utils.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python3
+
+import asyncio
+from playwright.async_api import async_playwright
+import os
+import tempfile
+from pathlib import Path
+
+async def take_screenshot(url: str, output_path: str = None, width: int = 1280, height: int = 720) -> str:
+    """
+    Take a screenshot of a webpage using Playwright.
+    
+    Args:
+        url (str): The URL to take a screenshot of
+        output_path (str, optional): Path to save the screenshot. If None, saves to a temporary file.
+        width (int, optional): Viewport width. Defaults to 1280.
+        height (int, optional): Viewport height. Defaults to 720.
+    
+    Returns:
+        str: Path to the saved screenshot
+    """
+    if output_path is None:
+        # Create a temporary file with .png extension
+        temp_file = tempfile.NamedTemporaryFile(suffix='.png', delete=False)
+        output_path = temp_file.name
+        temp_file.close()
+
+    async with async_playwright() as p:
+        browser = await p.chromium.launch(headless=True)
+        page = await browser.new_page(viewport={'width': width, 'height': height})
+        
+        try:
+            await page.goto(url, wait_until='networkidle')
+            await page.screenshot(path=output_path, full_page=True)
+        finally:
+            await browser.close()
+    
+    return output_path
+
+def take_screenshot_sync(url: str, output_path: str = None, width: int = 1280, height: int = 720) -> str:
+    """
+    Synchronous wrapper for take_screenshot.
+    """
+    return asyncio.run(take_screenshot(url, output_path, width, height))
+
+if __name__ == "__main__":
+    import argparse
+    parser = argparse.ArgumentParser(description='Take a screenshot of a webpage')
+    parser.add_argument('url', help='URL to take screenshot of')
+    parser.add_argument('--output', '-o', help='Output path for screenshot')
+    parser.add_argument('--width', '-w', type=int, default=1280, help='Viewport width')
+    parser.add_argument('--height', '-H', type=int, default=720, help='Viewport height')
+    
+    args = parser.parse_args()
+    output_path = take_screenshot_sync(args.url, args.output, args.width, args.height)
+    print(f"Screenshot saved to: {output_path}")