Adds watermark and stylesheets support to python sdk

OnedocLabs · Mar 12, 2024 · 6a2d945 · 6a2d945
1 parent 9f18d68
commit 6a2d945
Show file tree

Hide file tree

Showing 4 changed files with 75 additions and 18 deletions.
diff --git a/packages/client-py/onedoc/htmlBuilder.py b/packages/client-py/onedoc/htmlBuilder.py
@@ -0,0 +1,40 @@
+class HtmlBuilder:
+    def __init__(self, title=None):
+        self._title = title
+        self._start = """<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width">"""
+        self._middle = """
+</head>
+<body>"""
+        self._end = """</body>
+</html>"""
+        self._watermark = '<div id="watermark-onedoc" > <a href="https://www.onedoclabs.com/" target="_blank"> <svg style="transform: rotate(90deg);display:inline;margin-top:30px;" width=75 xmlns="http://www.w3.org/2000/svg" x="0" y="0" enableBackground="new 0 0 46.15 9.31" version="1.1" viewBox="0 0 46.15 9.31" xmlSpace="preserve" fill="black" {...props} > <path d="M10 9.13V2.55h1.83v.91c.35-.62 1.13-1.09 2.07-1.09.71 0 1.32.24 1.81.71s.74 1.15.74 2.03v4.02h-1.88V5.6c0-.96-.5-1.5-1.28-1.5-.85 0-1.42.62-1.42 1.55v3.48H10zM23.84 6.48h-4.83c.23.83.83 1.24 1.79 1.24.74 0 1.43-.22 2.05-.64l.74 1.28c-.8.61-1.76.91-2.88.91-1.16 0-2.05-.34-2.67-1-.61-.66-.92-1.47-.92-2.45 0-1 .32-1.81.96-2.46.64-.66 1.48-.98 2.51-.98.97 0 1.76.3 2.39.89.62.59.94 1.39.94 2.41-.01.23-.04.5-.08.8zM19 5.13h3.09c-.18-.76-.73-1.22-1.51-1.22-.76 0-1.38.46-1.58 1.22zM29.43 0h1.88v9.13h-1.82v-.71c-.52.59-1.16.88-1.96.88-.92 0-1.69-.32-2.31-.98-.61-.66-.92-1.47-.92-2.47 0-.98.31-1.8.92-2.46.62-.66 1.39-1 2.31-1 .74 0 1.38.26 1.89.8V0zm-.39 4.6c-.31-.34-.71-.5-1.2-.5s-.89.17-1.21.5c-.31.34-.47.74-.47 1.22 0 .49.16.91.47 1.25.32.34.72.5 1.21.5s.89-.17 1.2-.5c.32-.34.48-.76.48-1.25 0-.47-.15-.88-.48-1.22zM33.03 8.31c-.66-.67-.98-1.5-.98-2.47s.32-1.8.98-2.46c.66-.67 1.51-1.01 2.55-1.01 1.04 0 1.91.34 2.57 1.01.66.66 1 1.49 1 2.46s-.34 1.8-1 2.47c-.66.66-1.52 1-2.57 1-1.04 0-1.89-.34-2.55-1zm3.74-3.68c-.32-.34-.72-.5-1.19-.5s-.86.17-1.19.5c-.32.32-.48.73-.48 1.2 0 .49.16.9.48 1.24.32.32.72.49 1.19.49s.86-.17 1.19-.49c.32-.34.49-.74.49-1.24 0-.47-.17-.88-.49-1.2zM40.5 8.31c-.65-.65-.97-1.47-.97-2.48s.32-1.83.98-2.47c.66-.65 1.5-.97 2.54-.97 1.36 0 2.55.67 3.09 1.87l-1.5.8c-.38-.62-.9-.94-1.56-.94-.49 0-.89.17-1.21.49-.32.32-.48.73-.48 1.21 0 .49.16.91.47 1.24.32.32.72.48 1.2.48.66 0 1.27-.38 1.55-.92l1.52.9c-.58 1.07-1.74 1.75-3.12 1.75-1.02 0-1.86-.32-2.51-.96zM9.26 4.7c0-1.29-.44-2.36-1.34-3.25C7.03.55 5.94.1 4.63.1c-1.3 0-2.39.45-3.29 1.35C.45 2.34 0 3.43 0 4.71c0 .37.05.72.12 1.05l4.3-3.39h2.22v6.46c.47-.22.9-.5 1.29-.88.89-.89 1.33-1.97 1.33-3.25z"></path> <path d="M1.49 8.09c.62.56 1.34.94 2.17 1.1v-2.8l-2.17 1.7z"></path> </svg> <a /> </div>'
+
+    def build(self, document, style_sheets=None, dev=True):
+        if style_sheets:
+            for path in style_sheets:
+                self._start += f'<link rel="stylesheet" href="{path}" />'
+
+        if self._title:
+            self._start += f'<title>{self._title}</title>'
+
+        self._middle += document
+
+        if dev:
+            # adds watermark for dev renderings
+            self._start += """
+<style>
+@page {
+    @left-middle {
+        content: flow(watermark);
+    }
+}
+#watermark-onedoc { -prince-flow: static(watermark, start) }
+</style>"""
+
+            self._middle += self._watermark
+
+        return self._start + self._middle + self._end
diff --git a/packages/client-py/onedoc/onedoc.py b/packages/client-py/onedoc/onedoc.py
@@ -1,5 +1,6 @@
 import requests
 import json
+from htmlBuilder import HtmlBuilder
 
 from typing import Dict, Union, List, Any, BinaryIO
 
@@ -9,14 +10,14 @@
     "upsert": False,
 }
 
-class _HtmlBuilder:
-    def __init__(self, title: str = None):
-        self.title = title or "Document"
+# class _HtmlBuilder:
+#     def __init__(self, title: str = None):
+#         self.title = title or "Document"
 
-    def build(self, html_content: str, stylesheets: List[str] = None, test: bool = True) -> str:
-        # Implementation of HTML building logic goes here
-        # This is a placeholder implementation
-        return "<!DOCTYPE html><html><head><title>{}</title></head><body>{}</body></html>".format(self.title, html_content)
+#     def build(self, html_content: str, stylesheets: List[str] = None, test: bool = True) -> str:
+#         # Implementation of HTML building logic goes here
+#         # This is a placeholder implementation
+#         return "<!DOCTYPE html><html><head><title>{}</title></head><body>{}</body></html>".format(self.title, html_content)
 
 class Onedoc:
     def __init__(self, api_key: str):
@@ -80,7 +81,7 @@ def render(self, document: Dict) -> Dict:
                 self._upload_to_signed_url(e['signedUrl'], e['path'], asset['content'])
 
             elif e['path'] == "/index.html":
-                html_builder = _HtmlBuilder(document.get('title'))
+                html_builder = HtmlBuilder(document.get('title'))
                 style_sheets = [asset['path'] for asset in document.get('assets', []) if asset['path'].endswith(".css")]
                 html = html_builder.build(document['html'], style_sheets, test)
                 self._upload_to_signed_url(e['signedUrl'], e['path'], html)

diff --git a/packages/client-py/onedoc/styles.css b/packages/client-py/onedoc/styles.css
@@ -0,0 +1,3 @@
+body{
+    background-color: aqua;
+}
diff --git a/packages/client-py/onedoc/test.py b/packages/client-py/onedoc/test.py
@@ -6,17 +6,30 @@
 
 onedoc = Onedoc(getenv("ONEDOC_API_KEY"))
 
+
+with open("packages/client-py/onedoc/styles.css", 'r') as file:
+    # Read the entire content of the file into a variable
+    css_content = file.read()
+print(css_content)
+
 # Define your document
 document = {
     "html": "<h1>Table of contents</h1><a href='file://test.pdf#page=1'>First page</a><br/><a href='file://test.pdf#page=2'>Second page</a>",  # Simple HTML content
     "title": "My First Document",
-    "test": True,  # Set to False to use in production
-    "save": False,  # Set to True if you want to save the document
+    "test": False,  # Set to False to use in production
+    "save": True,  # Set to True if you want to save the document
+    "assets":[
+        {
+            "content": css_content ,
+            "path": "/styles.css"
+        }
+    ]
 }
 
 # Render the document
-#result = onedoc.render(document)
+result = onedoc.render(document)
 
+print(result)
 #for files that are not saved, remember to use "wb" when writing file
 # onedoc = Onedoc(api_key)
 
@@ -40,12 +53,12 @@
 #f.write(result.get("file"))
 #f.close()
 
-firstFile = open("toc.pdf", "rb")
-secondFile = open("test.pdf", "rb")
+# firstFile = open("toc.pdf", "rb")
+# secondFile = open("test.pdf", "rb")
 
-result = onedoc.merge(firstFile, "toc.pdf", secondFile, "test.pdf")
+# result = onedoc.merge(firstFile, "toc.pdf", secondFile, "test.pdf")
 
-# Result is a pdf file
-f = open("merged.pdf", "wb")
-f.write(result.get('file'))
-f.close()
+# # Result is a pdf file
+# f = open("merged.pdf", "wb")
+# f.write(result.get('file'))
+# f.close()