Dashboard for Managing Docs

commit/api/convert_to_webp.py

@@ -0,0 +1,101 @@
+import frappe
+from frappe.core.doctype.file.utils import delete_file
+from urllib.parse import unquote
+import requests
+from PIL import Image
+from frappe.core.doctype.file.file import get_local_image
+from frappe.model.document import Document
+import os
+
+@frappe.whitelist()
+def convert_to_webp(image_url: str | None = None, file_doc: Document | None = None) -> str:
+	"""BETA: Convert image to webp format"""
+
+	CONVERTIBLE_IMAGE_EXTENSIONS = ["png", "jpeg", "jpg"]
+
+	def can_convert_image(extn):
+		return extn.lower() in CONVERTIBLE_IMAGE_EXTENSIONS
+
+	def get_extension(filename):
+		return filename.split(".")[-1].lower()
+
+	def convert_and_save_image(image, path):
+		image.save(path, "WEBP")
+		return path
+
+	def update_file_doc_with_webp(file_doc, image, extn):
+		webp_path = file_doc.get_full_path().replace(extn, "webp")
+		convert_and_save_image(image, webp_path)
+		delete_file(file_doc.get_full_path())
+		file_doc.file_url = f"{file_doc.file_url.replace(extn, 'webp')}"
+		file_doc.save()
+		return file_doc.file_url
+
+	def create_new_webp_file_doc(file_url, image, extn):
+		files = frappe.get_all("File", filters={"file_url": file_url}, fields=["name"], limit=1)
+		if files:
+			_file = frappe.get_doc("File", files[0].name)
+			webp_path = _file.get_full_path().replace(extn, "webp")
+			convert_and_save_image(image, webp_path)
+			new_file = frappe.copy_doc(_file)
+			new_file.file_name = f"{_file.file_name.replace(extn, 'webp')}"
+			new_file.file_url = f"{_file.file_url.replace(extn, 'webp')}"
+			new_file.save()
+			return new_file.file_url
+		return file_url
+
+	def handle_image_from_url(image_url):
+		image_url = unquote(image_url)
+		response = requests.get(image_url)
+		image = Image.open(io.BytesIO(response.content))
+		filename = image_url.split("/")[-1]
+		extn = get_extension(filename)
+		if can_convert_image(extn):
+			_file = frappe.get_doc(
+				{
+					"doctype": "File",
+					"file_name": f"{filename.replace(extn, 'webp')}",
+					"file_url": f"/files/{filename.replace(extn, 'webp')}",
+				}
+			)
+			webp_path = _file.get_full_path()
+			convert_and_save_image(image, webp_path)
+			_file.save()
+			return _file.file_url
+		return image_url
+
+	if not image_url and not file_doc:
+		return ""
+
+	if file_doc:
+		if file_doc.file_url.startswith("/files"):
+			image, filename, extn = get_local_image(file_doc.file_url)
+			if can_convert_image(extn):
+				return update_file_doc_with_webp(file_doc, image, extn)
+		if file_doc.file_url.startswith("/private"):
+			image, filename, extn = get_local_image(file_doc.file_url)
+			if can_convert_image(extn):
+				return update_file_doc_with_webp(file_doc, image, extn)
+
+		return file_doc.file_url
+
+	if image_url.startswith("/files"):
+		image, filename, extn = get_local_image(image_url)
+		if can_convert_image(extn):
+			return create_new_webp_file_doc(image_url, image, extn)
+		return image_url
+	if image_url.startswith("/private"):
+		image, filename, extn = get_local_image(image_url)
+		if can_convert_image(extn):
+			return create_new_webp_file_doc(image_url, image, extn)
+		return image_url
+	if image_url.startswith("http"):
+		return handle_image_from_url(image_url)
+
+	return image_url
+
+def save_webp_image(doctype:str,docname:str,image_field:str):
+	file_url = frappe.db.get_value(doctype,docname,image_field)
+	if file_url:
+		webp_url = convert_to_webp(file_url)
+		frappe.db.set_value(doctype,docname,image_field,webp_url)

commit/api/preview.py

@@ -0,0 +1,59 @@
+import asyncio
+import io
+from pyppeteer import launch
+import frappe
+from frappe.utils.file_manager import save_file
+from commit.api.convert_to_webp import convert_to_webp
+
+async def capture_screenshot(url, width=1366, height=800, delay=3):
+    browser = await launch(
+        headless=True,
+        handleSIGINT=False,
+        handleSIGTERM=False,
+        handleSIGHUP=False
+    )
+    page = await browser.newPage()
+
+    await page.setViewport({"width": width, "height": height})
+    await page.goto(url, {"waitUntil": "load"})
+    await asyncio.sleep(delay)  # Ensure page loads fully
+
+    # ✅ Explicitly return bytes and ensure no file is saved
+    screenshot_bytes = await page.screenshot({"fullPage": False, "encoding": "binary"}) 
+
+    await browser.close()
+
+    return screenshot_bytes  # Return raw image data
+
+def save_preview_screenshot(url, doctype, docname, field):
+    try:
+        loop = asyncio.get_running_loop()
+    except RuntimeError:
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+        screenshot_bytes = loop.run_until_complete(capture_screenshot(url))
+    else:
+        screenshot_bytes = loop.run_until_complete(capture_screenshot(url))
+
+    # ✅ Correct way: Wrap bytes in BytesIO
+    screenshot_io = io.BytesIO(screenshot_bytes)
+
+    file_doc = frappe.get_doc({
+        "doctype": "File",
+        "file_name": docname + "_" + "preview.png",
+        "attached_to_doctype": doctype,
+        "attached_to_name": docname,
+        "attached_to_field": field,
+        "is_private": 1,
+        "content": screenshot_io.getvalue()
+    })
+    file_doc.save()
+
+    # Convert to WebP
+    file_url = convert_to_webp(file_doc.file_url, file_doc)
+    # Update the document with the preview file URL
+    doc = frappe.get_doc(doctype, docname)
+    doc.set(field, file_url)
+    doc.save()
+
+    frappe.db.commit()

commit/commit/doctype/commit_docs/commit_docs.json

@@ -16,6 +16,7 @@
   "logo_section",
   "light_mode_logo",
   "night_mode_logo",
+  "preview_image",
   "navbar_tab",
   "navbar_items",
   "footer_tab",
@@ -27,6 +28,7 @@
   "github",
   "raven",
   "telegram",
+  "discord",
   "section_break_gawv",
   "footer"
  ],
@@ -166,11 +168,22 @@
    "fieldname": "description",
    "fieldtype": "Small Text",
    "label": "Description"
+  },
+  {
+   "fieldname": "preview_image",
+   "fieldtype": "Attach Image",
+   "label": "Preview Image"
+  },
+  {
+   "description": "Add Discord URl",
+   "fieldname": "discord",
+   "fieldtype": "Data",
+   "label": "Discord"
   }
  ],
  "index_web_pages_for_search": 1,
  "links": [],
- "modified": "2024-12-07 22:34:40.239467",
+ "modified": "2025-03-30 12:43:40.306104",
  "modified_by": "Administrator",
  "module": "commit",
  "name": "Commit Docs",
@@ -190,6 +203,7 @@
    "write": 1
   }
  ],
+ "row_format": "Dynamic",
  "sort_field": "creation",
  "sort_order": "DESC",
  "states": []