1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
|
import logging
from django.template import Template
from aws_xray_sdk.core import xray_recorder
log = logging.getLogger(__name__)
def patch_template():
attr = '_xray_original_render'
if getattr(Template, attr, None):
log.debug("already patched")
return
setattr(Template, attr, Template.render)
@xray_recorder.capture('template_render')
def xray_render(self, context):
template_name = self.name or getattr(context, 'template_name', None)
if template_name:
name = str(template_name)
subsegment = xray_recorder.current_subsegment()
subsegment.name = name
return Template._xray_original_render(self, context)
Template.render = xray_render
|
