Add the hability to disable the context menu and render JSON (#21)

Author: mrcingo

pypulse/Socket/handler.py

@@ -33,7 +33,7 @@ def __check_request(self):
 
         self.response = view[0](request) if not view[1] else view[0](request, **view[1])
 
-        if type(self.response).__name__ not in ["Redirect", "RenderTemplate", "Reload"]:
+        if type(self.response).__name__ not in ["Redirect", "RenderTemplate", "Reload", "RenderJson"]:
             return False
 
         return True

pypulse/Template/__init__.py

@@ -1,4 +1,4 @@
-from pypulse.Template.render import RenderTemplate
+from pypulse.Template.render import RenderTemplate, RenderJson
 from pypulse.Template.redirect import Redirect
 from pypulse.Template.reload import Reload
-from pypulse.Template.template import Template
+from pypulse.Template.template import Template

pypulse/Template/render.py

@@ -1,23 +1,43 @@
+from __future__ import annotations
+
+import json
+from typing import TYPE_CHECKING, Union
+
 from jinja2 import Environment, FileSystemLoader
+
 from .template import Template
 
+if TYPE_CHECKING:
+    from ..Socket.handler import Request
 
-class RenderTemplate():
-    def __init__(self, template_route: str, variables_dict: dict = None) -> None:
-        self.template_route = template_route
-        self.variables_dict = variables_dict
 
-        self.envirenemnt = Environment(
-            loader=FileSystemLoader(Template.TEMPLATE_PATH))
+class RenderTemplate:
+    
+    def __init__(self, template_route: str, variables_dict: dict = None):
+        self.template_route: str = template_route
+        self.variables_dict: dict = variables_dict
 
-    def render_template(self, request):
+        self.jinja_env: Environment = Environment(loader=FileSystemLoader(Template.TEMPLATE_PATH))
+
+    def render_template(self, request: Request) -> Union[str, bool]:
         request.send_response(200)
-        request.send_header('Content-type', 'text/html')
+        request.send_header('Content-Type', 'text/html')
 
-        template = self.envirenemnt.get_template(self.template_route)
+        jinja_template = self.jinja_env.get_template(self.template_route)
+        
         if self.variables_dict is not None:
-            output = template.render(self.variables_dict)
-        else:
-            output = template.render()
+            return jinja_template.render(self.variables_dict), False
+
+        return jinja_template.render(), False
+
+
+class RenderJson:
+
+    def __init__(self, json_to_render: Union[dict, list]):
+        self.json_to_render: Union[dict, list] = json_to_render
+
+    def render_template(self, request: Request) -> Union[str, bool]:
+        request.send_response(200)
+        request.send_header('Content-Type', 'application/json')
 
-        return output, False
+        return json.dumps(self.json_to_render), False

pypulse/Window/cef.py

@@ -90,6 +90,17 @@ def event(self, type: Event) -> None:
         ...
 
 
+class Handler:
+        """
+        See [CEFPython handlers](https://github.com/cztomczak/cefpython/blob/master/api/API-categories.md#client-handlers-interfaces) if you want to check all the 
+        handlers currently available.
+        """
+
+        def OnGotFocus(self, browser):
+            if system() == 'Linux':
+                browser.SetFocus(True)
+
+
 class Browser(CEF):
     """
     A more friendly way to manage CEF browser.
@@ -102,15 +113,6 @@ def __init__(self, url: str, *args, **kwargs) -> None:
                          window_info=self.window, 
                          url=url, 
                          *args, **kwargs)
-        
-    class Handler:
-        """
-        See [CEFPython handlers](https://github.com/cztomczak/cefpython/blob/master/api/API-categories.md#client-handlers-interfaces) if you want to check all the 
-        handlers currently available.
-        """
-        def OnGotFocus(self, browser):
-            if system() == 'Linux':
-                browser.SetFocus(True)
 
     def event(self, type: Event) -> None:
         """
@@ -119,4 +121,4 @@ def event(self, type: Event) -> None:
         """
         if type is not Event.BROWSER_AFTER:
             return
-        self.instance.SetClientHandler(self.Handler())
+        self.instance.SetClientHandler(Handler())

pypulse/Window/frames.py

@@ -61,6 +61,7 @@ def __init__(self,
 
         self.browser.config['product_version'] = 'PyPulse/0.1.9'
         self.browser.config['debug'] = debug
+        self.browser.config['context_menu'] = dict(enabled=debug)
         if log_file:
             self.browser.config['log_file'] = log_file
 
@@ -116,7 +117,6 @@ def __init__(self,
         log(LogTypes.SUCCESS, 'Window added')
 
         self.browser.open()
-        # os.system('clear' if self.browser.os != 'Windows' else 'cls')
 
         log(LogTypes.INFO, 'Browser open')