add wechart process

 import copy
 import copy
+import re
 
 
 TIME_NUM = 600000000
 TIME_NUM = 600000000
 
 
@@ -909,3 +910,5 @@ PATTERN_LIST = ['收入', '存入', '支出', '支取', '金额', '余额', '发
                 '摘要/附言', '交易发生额', '交易摘要', '借贷发生额(借:-贷:+)', '借贷发生额（借:-贷:+）', '联机余额', '交易金额(元)',
                 '摘要/附言', '交易发生额', '交易摘要', '借贷发生额(借:-贷:+)', '借贷发生额（借:-贷:+）', '联机余额', '交易金额(元)',
                 '交易金额（元）', '账户余额（元）', '账户余额(元)', '会计日期', '摘要代码', '摘要信息', '日期', '短摘要', '本次余额',
                 '交易金额（元）', '账户余额（元）', '账户余额(元)', '会计日期', '摘要代码', '摘要信息', '日期', '短摘要', '本次余额',
                 '交易后余额', '交易说明', '帐户余额', '交易日期 记账日期']
                 '交易后余额', '交易说明', '帐户余额', '交易日期 记账日期']
+
+CN_RE = re.compile(u'[\u4e00-\u9fa5]')

@@ -8,13 +8,13 @@ class DocHandler:
     @staticmethod
     @staticmethod
     def get_link(doc_id, business_type, file='pdf'):
     def get_link(doc_id, business_type, file='pdf'):
         if file == 'pdf':
         if file == 'pdf':
-            return '/data/{1}/{0}/{0}.pdf'.format(doc_id, business_type)
+            return '/data/{1}/{0}/{2}/{0}.pdf'.format(doc_id, business_type, consts.TMP_DIR_NAME)
         elif file == 'img':
         elif file == 'img':
-            return '/data/{1}/{0}/{0}_img.zip'.format(doc_id, business_type)
+            return '/data/{1}/{0}/{2}/{0}_img.zip'.format(doc_id, business_type, consts.TMP_DIR_NAME)
         elif file == 'src_excel':
         elif file == 'src_excel':
-            return '/data/{1}/{0}/src.xlsx'.format(doc_id, business_type)
+            return '/data/{1}/{0}/{2}/src.xlsx'.format(doc_id, business_type, consts.TMP_DIR_NAME)
         else:
         else:
-            return '/data/{1}/{0}/{0}.xlsx'.format(doc_id, business_type)
+            return '/data/{1}/{0}/{2}/{0}.xlsx'.format(doc_id, business_type, consts.TMP_DIR_NAME)
 
 
     def get_doc_list(self, doc_queryset, business_type):
     def get_doc_list(self, doc_queryset, business_type):
         for doc_dict in doc_queryset:
         for doc_dict in doc_queryset:

@@ -311,7 +311,17 @@ class BSWorkbook(Workbook):
                 res_str = '{0}.{1}'.format(res_str[:pre_idx], res_str[period_idx + 1:])
                 res_str = '{0}.{1}'.format(res_str[:pre_idx], res_str[period_idx + 1:])
         return res_str
         return res_str
 
 
-    def build_month_sheet(self, ms, card, month_mapping, is_reverse, statistics_header_info, max_column):
+    @staticmethod
+    def wechart_func(row_value):
+        row_value = list(row_value)
+        if isinstance(row_value[1], str):
+            cn_chars = re.findall(consts.CN_RE, row_value[1])
+            cn_str = ''.join(cn_chars)
+            row_value[2] = cn_str + row_value[2]
+            row_value[1] = re.sub(consts.CN_RE, '', row_value[1])
+        return row_value
+
+    def build_month_sheet(self, ms, card, month_mapping, is_reverse, statistics_header_info, max_column, classify):
         summary_cell_idx = statistics_header_info.get(consts.SUMMARY_KEY)
         summary_cell_idx = statistics_header_info.get(consts.SUMMARY_KEY)
         date_cell_idx = statistics_header_info.get(consts.DATE_KEY)
         date_cell_idx = statistics_header_info.get(consts.DATE_KEY)
         amount_cell_idx = statistics_header_info.get(consts.AMOUNT_KEY)  # None or src or append
         amount_cell_idx = statistics_header_info.get(consts.AMOUNT_KEY)  # None or src or append
@@ -343,6 +353,8 @@ class BSWorkbook(Workbook):
                 ws = self.get_sheet_by_name(part[0])
                 ws = self.get_sheet_by_name(part[0])
                 for row_value in ws.iter_rows(min_row=part[1], max_row=part[2], values_only=True):
                 for row_value in ws.iter_rows(min_row=part[1], max_row=part[2], values_only=True):
                     if any(row_value):
                     if any(row_value):
+                        if classify == consts.WECHART_CLASSIFY:
+                            row_value = self.wechart_func(row_value)
                         new_ws.append(row_value)
                         new_ws.append(row_value)
             # 3.2.提取信息、高亮
             # 3.2.提取信息、高亮
             amount_mapping = {}
             amount_mapping = {}
@@ -506,7 +518,7 @@ class BSWorkbook(Workbook):
             for month_list in month_mapping.values():
             for month_list in month_mapping.values():
                 month_list.sort(key=lambda x: x[-1], reverse=is_reverse)
                 month_list.sort(key=lambda x: x[-1], reverse=is_reverse)
 
 
-            self.build_month_sheet(ms, card, month_mapping, is_reverse, statistics_header_info, max_column)
+            self.build_month_sheet(ms, card, month_mapping, is_reverse, statistics_header_info, max_column, classify)
 
 
             # 4.删除原表
             # 4.删除原表
             for sheet in sheets_list:
             for sheet in sheets_list: