pdf pwd from doc name

@@ -1596,7 +1596,7 @@ def get_se_cms_compare_info(last_obj, application_entity, detect_list, auto=Fals
         jypz_field_input.append(('vinNo', vin_no))
         jypz_field_input.append(('vinNo', vin_no))
         jypz_field_input.append(('vehicleTransactionAmount', amount))
         jypz_field_input.append(('vehicleTransactionAmount', amount))
         jypz_field_input.append((consts.SE_GB_USED_FIELD[2], first_submission_date))
         jypz_field_input.append((consts.SE_GB_USED_FIELD[2], first_submission_date))
-        if fp_group.find('Non-OCU Product Group') != -1:
+        if fp_group.find('Non OCU Product Group') != -1:
             jypz_field_input.append(('type', consts.JYPZ_TYPE_1))
             jypz_field_input.append(('type', consts.JYPZ_TYPE_1))
         elif fp_group.find('OCU Product Group') != -1:
         elif fp_group.find('OCU Product Group') != -1:
             jypz_field_input.append(('type', consts.JYPZ_TYPE_2))
             jypz_field_input.append(('type', consts.JYPZ_TYPE_2))

 import os
 import os
+import re
 import json
 import json
 import cv2
 import cv2
 import shutil
 import shutil
@@ -48,8 +49,9 @@ class PDFBuild:
 
 
 class PDFHandler:
 class PDFHandler:
 
 
-    def __init__(self, path, img_dir_path, document_name=None):
+    def __init__(self, path, img_dir_path, document_name=None, pwd_list=[]):
         self.path = path
         self.path = path
+        self.pwd_list = self.get_pwd_list(document_name, pwd_list)
         self.img_dir_path = img_dir_path
         self.img_dir_path = img_dir_path
         self.img_path_list = []
         self.img_path_list = []
         self.img_count = 0
         self.img_count = 0
@@ -75,6 +77,12 @@ class PDFHandler:
         self.page_count = None
         self.page_count = None
         self.metadata = None
         self.metadata = None
 
 
+    @staticmethod
+    def get_pwd_list(doc_name, pwd_list):
+        pwd_list_from_doc_name = re.findall(r'\d{6}', doc_name)
+        pwd_list_from_doc_name.extend(pwd_list)
+        return pwd_list_from_doc_name
+
     def get_suffix(self, file_name):
     def get_suffix(self, file_name):
         if file_name is None:
         if file_name is None:
             return None
             return None
@@ -370,6 +378,12 @@ class PDFHandler:
             self.img_path_list.append(img_save_path)
             self.img_path_list.append(img_save_path)
         else:
         else:
             with fitz.Document(self.path) as pdf:
             with fitz.Document(self.path) as pdf:
+                # 解密
+                for pwd in self.pwd_list:
+                    if not pdf.isEncrypted:
+                        break
+                    pdf.authenticate(pwd)
+
                 self.metadata = pdf.metadata
                 self.metadata = pdf.metadata
                 self.page_count = pdf.pageCount
                 self.page_count = pdf.pageCount
                 if isinstance(max_img_count, int) and pdf.pageCount >= max_img_count:
                 if isinstance(max_img_count, int) and pdf.pageCount >= max_img_count: