[FIX]docx文档打印时，如内容包含 & 符号会报错 osbzr#1543

shinhaku · Jun 5, 2018 · 981e714 · 981e714
1 parent c87f4e0
commit 981e714
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/report_docx/report/report_docx.py b/report_docx/report/report_docx.py
@@ -70,6 +70,8 @@ def __getattr__(self, key):
             return ""
         temp = getattr(self.data, key)
         field = self.data._fields.get(key)
+        if isinstance(temp, unicode) and ('&' in temp or '<' in temp or '>' in temp):
+            temp = temp.replace('&', '&amp;').replace('<', '&lt;').replace('>', '&gt;')
 
         if isinstance(temp, models.Model):
             return DataModelProxy(temp)
@@ -89,7 +91,16 @@ def __iter__(self):
 
     def __str__(self):
         '''支持直接在word 上写 many2one 字段'''
-        return self.data and self.data.display_name or ''
+        name = ''
+        if self.data and self.data.display_name:
+            name = self.data.display_name
+            if '&' in self.data.display_name:
+                name = name.replace('&', '&amp;')
+            if '<' in self.data.display_name:
+                name = name.replace('<', '&lt;')
+            if '>' in self.data.display_name:
+                name = name.replace('>', '&gt;')
+        return name
 
 
 class IterDataModelProxy(object):