Trang chủ Khám phá Trợ giúp
Đăng ký Đăng nhập
root
/
nicegui
mirror of https://github.com/zauberzeug/nicegui.git
1
0
Fork 0
Các tập tin Các vấn đề 0 Wiki
Browse Source

split shared and regular head/body HTML

Falko Schindler 1 năm trước cách đây
mục cha
424c0df804
commit
a09e713ece
2 tập tin đã thay đổi với 48 bổ sung và 11 xóa
View chưa được định nghĩa Hiển thị tình trạng sai khác
  1. 17 4
      nicegui/client.py
  2. 31 7
      nicegui/functions/html.py

+ 17 - 4
nicegui/client.py
Xem Tập Tin 

@@ -38,11 +38,11 @@ class Client:
 
     auto_index_client: Client
 
     auto_index_client: Client
 
     """The client that is used to render the auto-index page."""
 
     """The client that is used to render the auto-index page."""
 
 
 
-    head_html = ''
 
-    """HTML to be inserted in the <head> of the page template."""
 
+    shared_head_html = ''
 
+    """HTML to be inserted in the <head> of every page template."""
 
 
 
-    body_html = ''
 
-    """HTML to be inserted in the <body> of the 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:
 
     def __init__(self, page: page, *, shared: bool = False) -> None:
 
         self.id = str(uuid.uuid4())
 
         self.id = str(uuid.uuid4())
 
@@ -65,6 +65,9 @@ class Client:
 
 
 
         self.waiting_javascript_commands: Dict[str, Any] = {}
 
         self.waiting_javascript_commands: Dict[str, Any] = {}
 
 
 
+        self._head_html = ''
 
+        self._body_html = ''
 
+
 
         self.page = page
 
         self.page = page
 
 
 
         self.connect_handlers: List[Union[Callable[..., Any], Awaitable]] = []
 
         self.connect_handlers: List[Union[Callable[..., Any], Awaitable]] = []
 
@@ -87,6 +90,16 @@ class Client:
 
         """Return True if the client is connected, False otherwise."""
 
         """Return True if the client is connected, False otherwise."""
 
         return self.environ is not None
 
         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):
 
     def __enter__(self):
 
         self.content.__enter__()
 
         self.content.__enter__()
 
         return self
 
         return self

+ 31 - 7
nicegui/functions/html.py
Xem Tập Tin 

@@ -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