Acasă Explorează Ajutor
Înregistrare Autentificare
root
/
nicegui
oglindă de https://github.com/zauberzeug/nicegui.git
1
0
Bifurcare 0
Fisiere Probleme 0 Wiki
Răsfoiți Sursa

Merge pull request #2127 from DaelonSuzuka/expose_page_template

Allow access to page template head and body
Falko Schindler 1 an în urmă
părinte
f40963549e a09e713ece
comite
2fce6cf05f
2 a modificat fișierele cu 49 adăugiri și 9 ștergeri
Vizualizare divizată Arată Statisticile Diff
  1. 18 2
      nicegui/client.py
  2. 31 7
      nicegui/functions/html.py

+ 18 - 2
nicegui/client.py
Vizualizați fișierul 

@@ -38,6 +38,12 @@ class Client:
 
     auto_index_client: Client
 
     """The client that is used to render the auto-index page."""
 
 
+    shared_head_html = ''
 
+    """HTML to be inserted in the <head> of every page template."""
 
+
 
+    shared_body_html = ''
 
+    """HTML to be inserted in the <body> of every page template."""
 
+
 
     def __init__(self, page: page, *, shared: bool = False) -> None:
 
         self.id = str(uuid.uuid4())
 
         self.created = time.time()
 
@@ -59,8 +65,8 @@ class Client:
 
 
         self.waiting_javascript_commands: Dict[str, Any] = {}
 
 
-        self.head_html = ''
 
-        self.body_html = ''
 
+        self._head_html = ''
 
+        self._body_html = ''
 
 
         self.page = page
 
 
@@ -84,6 +90,16 @@ class Client:
 
         """Return True if the client is connected, False otherwise."""
 
         return self.environ is not None
 
 
+    @property
 
+    def head_html(self) -> str:
 
+        """Return the HTML code to be inserted in the <head> of the page template."""
 
+        return self.shared_head_html + self._head_html
 
+
 
+    @property
 
+    def body_html(self) -> str:
 
+        """Return the HTML code to be inserted in the <body> of the page template."""
 
+        return self.shared_body_html + self._body_html
 
+
 
     def __enter__(self):
 
         self.content.__enter__()
 
         return self

+ 31 - 7
nicegui/functions/html.py
Vizualizați fișierul 

@@ -1,11 +1,35 @@
 
-from .. import context
 
+from .. import Client, context
 
 
 
-def add_body_html(code: str) -> None:
 
-    """Add HTML code to the body of the page."""
 
-    context.get_client().body_html += code + '\n'
 
+def add_head_html(code: str, *, shared: bool = False) -> None:
 
+    """Add HTML code to the head of the page.
 
 
+    Note that this function can only be called before the page is sent to the client.
 
 
-def add_head_html(code: str) -> None:
 
-    """Add HTML code to the head of the page."""
 
-    context.get_client().head_html += code + '\n'
 
+    :param code: HTML code to add
 
+    :param shared: if True, the code is added to all pages
 
+    """
 
+    if shared:
 
+        Client.shared_head_html += code + '\n'
 
+    else:
 
+        client = context.get_client()
 
+        if client.has_socket_connection:
 
+            raise RuntimeError('Cannot add head HTML after the page has been sent to the client.')
 
+        client._head_html += code + '\n'  # pylint: disable=protected-access
 
+
 
+
 
+def add_body_html(code: str, *, shared: bool = False) -> None:
 
+    """Add HTML code to the body of the page.
 
+
 
+    Note that this function can only be called before the page is sent to the client.
 
+
 
+    :param code: HTML code to add
 
+    :param shared: if True, the code is added to all pages
 
+    """
 
+    if shared:
 
+        Client.shared_body_html += code + '\n'
 
+    else:
 
+        client = context.get_client()
 
+        if client.has_socket_connection:
 
+            raise RuntimeError('Cannot add body HTML after the page has been sent to the client.')
 
+        client._body_html += code + '\n'  # pylint: disable=protected-access