init

+from mimetypes import suffix_map
 import os
 import io
 import time
@@ -1125,10 +1126,18 @@ class DocView(DocGenericView, DocHandler):
         random_int = random.randint(0, consts.TIME_NUM)
         metadata_version_id = str(int(time.time()) - random_int)
 
+        # 现在不只接受pdf 图片也接收
         pdf_file = request.FILES.get('pdf_file')
+        file_suffux = ''
         if isinstance(pdf_file.name, str):
-            if not pdf_file.name.endswith('pdf') and not pdf_file.name.endswith('PDF'):
-                self.invalid_params(msg='invalid params: not a PDF file')
+        #     if not pdf_file.name.endswith('pdf') and not pdf_file.name.endswith('PDF'):
+                # self.invalid_params(msg='invalid params: not a PDF file')
+            if '.' in pdf_file.name:
+                file_suffux = pdf_file.name.split('.')[-1]
+            else:
+                self.invalid_params(msg='invalid params: invalid file name')
+        else:
+            self.invalid_params(msg='invalid params: invalid file name')
 
         business_type = request.POST.get('business_type', '')
         document_scheme = request.POST.get('document_scheme', '')
@@ -1138,24 +1147,25 @@ class DocView(DocGenericView, DocHandler):
         args = {'business_type':business_type,'document_scheme':document_scheme,'data_source':data_source,'document_name':document_name,'pdf_file':pdf_file,'fsm':fsm}
         # business_type = random.choice(consts.BUSINESS_TYPE_LIST)
         # business_type = consts.BUSINESS_TYPE_LIST[0]
-        tmp_save_path = os.path.join(conf.DATA_DIR, business_type, '{0}.pdf'.format(metadata_version_id))
+        tmp_save_path = os.path.join(conf.DATA_DIR, business_type, '{0}.{1}}'.format(metadata_version_id, file_suffux))
         file_write(pdf_file, tmp_save_path)
 
-        try:
-            file = fitz.Document(tmp_save_path)
-        except Exception as e:
-            os.remove(tmp_save_path)
-            raise self.invalid_params(msg='invalid params: not a PDF file')
-        else:
-            if not file.isPDF:
-                file.close()
+        if file_suffux.lower() == 'pdf':
+            try:
+                file = fitz.Document(tmp_save_path)
+            except Exception as e:
                 os.remove(tmp_save_path)
                 raise self.invalid_params(msg='invalid params: not a PDF file')
-            # elif not self.xss_pass(file):
-            #     os.remove(tmp_save_path)
-            #     raise self.invalid_params(msg='invalid params: PDF file XSS')
-
-        file.close()
+            else:
+                if not file.isPDF:
+                    file.close()
+                    os.remove(tmp_save_path)
+                    raise self.invalid_params(msg='invalid params: not a PDF file')
+                # elif not self.xss_pass(file):
+                #     os.remove(tmp_save_path)
+                #     raise self.invalid_params(msg='invalid params: PDF file XSS')
+
+            file.close()
         # 1. 上传信息记录
         application_id = '{0}{1}'.format(consts.FIXED_APPLICATION_ID_PREFIX, metadata_version_id)
         upload_finish_time = timezone.now()
@@ -1194,7 +1204,7 @@ class DocView(DocGenericView, DocHandler):
 
         # 3.pdf文件移动
         save_dir_path = os.path.join(conf.DATA_DIR, business_type, consts.TMP_DIR_NAME, str(doc.id))
-        save_file_path = os.path.join(save_dir_path, '{0}.pdf'.format(doc.id))
+        save_file_path = os.path.join(save_dir_path, '{0}.{1}'.format(doc.id, file_suffux))
         os.makedirs(save_dir_path, exist_ok=True)
         # file_write(pdf_file, save_file_path)
         shutil.move(tmp_save_path, save_file_path)