add ltgt yh

@@ -50,6 +50,7 @@ class Command(BaseCommand, LoggerMixin):
         self.log_base = '[folder ltgt process]'
         self.log_base = '[folder ltgt process]'
         # 处理文件开关
         # 处理文件开关
         self.switch = True
         self.switch = True
+        self.amount_key_set = {'债权金额', '贷款本金', '罚息', '律师费', '案件受理费', '支付金额'}
         self.ltgt_classify_mapping = {
         self.ltgt_classify_mapping = {
             128: '执行裁定书',
             128: '执行裁定书',
             129: '民事判决书',
             129: '民事判决书',
@@ -287,7 +288,7 @@ class Command(BaseCommand, LoggerMixin):
                     result_dict = ocr_res.get('data', {})
                     result_dict = ocr_res.get('data', {})
 
 
                     wb = BSWorkbook(set(), set(), set(), set(), set())
                     wb = BSWorkbook(set(), set(), set(), set(), set())
-                    rebuild_res = wb.ltgt_build(label, result_dict)
+                    rebuild_res = wb.ltgt_build(label, result_dict, self.amount_key_set)
                     wb.remove_base_sheet()
                     wb.remove_base_sheet()
                     wb.save(excel_path)
                     wb.save(excel_path)
                     return rebuild_res
                     return rebuild_res

@@ -702,7 +702,14 @@ class BSWorkbook(Workbook):
             if field_str is not None:
             if field_str is not None:
                 count_list.append((field_str, count))
                 count_list.append((field_str, count))
 
 
-    def ltgt_build(self, label, result_dict):
+    @staticmethod
+    def remove_yuan(amount_key_set, key, src_str):
+        if key in amount_key_set and isinstance(src_str, str):
+            return src_str.replace('元', '')
+        else:
+            return src_str
+
+    def ltgt_build(self, label, result_dict, amount_key_set):
         ws = self.create_sheet(label)
         ws = self.create_sheet(label)
         rebuild_res = {}
         rebuild_res = {}
         for key, value in result_dict.items():
         for key, value in result_dict.items():
@@ -712,19 +719,23 @@ class BSWorkbook(Workbook):
                 rebuild_res[key] = '、'.join(value_list)
                 rebuild_res[key] = '、'.join(value_list)
             elif isinstance(value, dict):
             elif isinstance(value, dict):
                 if 'words' in value:
                 if 'words' in value:
-                    ws.append((key, value['words']))
-                    rebuild_res[key] = value['words']
+                    new_value = self.remove_yuan(amount_key_set, key, value['words'])
+                    ws.append((key, new_value))
+                    rebuild_res[key] = new_value
                 else:
                 else:
                     for sub_key, sub_value in value.items():
                     for sub_key, sub_value in value.items():
                         if isinstance(sub_value, dict):
                         if isinstance(sub_value, dict):
-                            ws.append(('{0}: {1}'.format(key, sub_key), sub_value.get('words', '')))
-                            rebuild_res['{0}: {1}'.format(key, sub_key)] = sub_value.get('words', '')
+                            new_value = self.remove_yuan(amount_key_set, sub_key, sub_value.get('words', ''))
+                            ws.append(('{0}: {1}'.format(key, sub_key), new_value))
+                            rebuild_res['{0}: {1}'.format(key, sub_key)] = new_value
                         else:
                         else:
-                            ws.append(('{0}: {1}'.format(key, sub_key), sub_value))
-                            rebuild_res['{0}: {1}'.format(key, sub_key)] = sub_value
+                            new_value = self.remove_yuan(amount_key_set, sub_key, sub_value)
+                            ws.append(('{0}: {1}'.format(key, sub_key), new_value))
+                            rebuild_res['{0}: {1}'.format(key, sub_key)] = new_value
             else:
             else:
-                ws.append((key, value))
-                rebuild_res[key] = value
+                new_value = self.remove_yuan(amount_key_set, key, value)
+                ws.append((key, new_value))
+                rebuild_res[key] = new_value
         return rebuild_res
         return rebuild_res
 
 
     def simple_license_rebuild(self, license_summary, document_scheme):
     def simple_license_rebuild(self, license_summary, document_scheme):