fix bug

@@ -165,7 +165,12 @@ class Command(BaseCommand, LoggerMixin):
     def folder_process(self, input_dir, classify):
     def folder_process(self, input_dir, classify):
         while not os.path.isdir(input_dir):
         while not os.path.isdir(input_dir):
             self.folder_log.info('{0} [input dir is not dir] [input_dir={1}]'.format(self.log_base, input_dir))
             self.folder_log.info('{0} [input dir is not dir] [input_dir={1}]'.format(self.log_base, input_dir))
-            time.sleep(self.sleep_time)
+            print(self.switch)
+            if self.switch:
+                time.sleep(self.sleep_time)
+                continue
+            else:
+                return
         output_dir = os.path.join(os.path.dirname(input_dir), 'Output')
         output_dir = os.path.join(os.path.dirname(input_dir), 'Output')
         img_output_dir = os.path.join(output_dir, 'image')
         img_output_dir = os.path.join(output_dir, 'image')
         wb_output_dir = os.path.join(output_dir, 'excel')
         wb_output_dir = os.path.join(output_dir, 'excel')
@@ -177,11 +182,16 @@ class Command(BaseCommand, LoggerMixin):
         os.makedirs(pdf_output_dir, exist_ok=True)
         os.makedirs(pdf_output_dir, exist_ok=True)
         os.makedirs(failed_output_dir, exist_ok=True)
         os.makedirs(failed_output_dir, exist_ok=True)
         while self.switch:
         while self.switch:
+            if not os.path.isdir(input_dir):
+                self.folder_log.info('{0} [input dir is not dir] [input_dir={1}]'.format(self.log_base, input_dir))
+                time.sleep(self.sleep_time)
+                continue
             # 1. 从input dir获取pdf or image
             # 1. 从input dir获取pdf or image
             list_dir = os.listdir(input_dir)
             list_dir = os.listdir(input_dir)
             if not list_dir:
             if not list_dir:
                 self.folder_log.info('{0} [input dir empty] [input_dir={1}]'.format(self.log_base, input_dir))
                 self.folder_log.info('{0} [input dir empty] [input_dir={1}]'.format(self.log_base, input_dir))
                 time.sleep(self.sleep_time)
                 time.sleep(self.sleep_time)
+                continue
             for name in list_dir:
             for name in list_dir:
                 try:
                 try:
                     path = os.path.join(input_dir, name)
                     path = os.path.join(input_dir, name)

@@ -294,7 +294,7 @@ class DocView(GenericView, DocHandler):
 
 
         pdf_file = args.get('pdf_file')
         pdf_file = args.get('pdf_file')
         if isinstance(pdf_file.name, str):
         if isinstance(pdf_file.name, str):
-            if not pdf_file.name.endswith('pdf') or not pdf_file.name.endswith('PDF'):
+            if not pdf_file.name.endswith('pdf') and not pdf_file.name.endswith('PDF'):
                 self.invalid_params(msg='invalid params: not a PDF file')
                 self.invalid_params(msg='invalid params: not a PDF file')
 
 
         business_type = random.choice(consts.BUSINESS_TYPE_LIST)
         business_type = random.choice(consts.BUSINESS_TYPE_LIST)