add with profile block

Author: PetrDlouhy

template_profiler_panel/panels/template.py

@@ -27,6 +27,16 @@
 
 @wrapt.decorator
 def profile_method(wrapped, instance, args, kwargs):
+    """
+
+    Profile method decorator
+    Usage:
+
+    from template_profiler_panel.panels.template import profile_method
+    @profile method
+    def method():
+        ...
+    """
     start = time()
 
     result = wrapped(*args, **kwargs)
@@ -47,6 +57,34 @@ def profile_method(wrapped, instance, args, kwargs):
     return result
 
 
+class Profile:
+    """
+    Profile with block
+    Usage:
+
+    from template_profiler_panel.panels.template import Profile
+    with Profile("block name"):
+        ...
+    """
+    def __init__(self, name="Profile with", *args, **kwargs):
+        self.name = name
+
+    def __enter__(self):
+        self.start = time()
+
+    def __exit__(self, type, value, traceback):
+        self.end = time()
+
+        template_rendered.send(
+            sender=self.__class__,
+            instance=self.name,
+            start=self.start,
+            end=self.end,
+            processing_timeline=[],
+            level=1,
+        )
+
+
 def get_nodelist_timeline(nodelist, level):
     timeline = []
     for node in nodelist: