Update appendImageHeader.py

Author: Dmitriy-Xawstov

Uses-Cases/HeaderFooter/appendImageHeader.py

@@ -35,89 +35,74 @@ def _init_api(self, credentials_file: Path):
         except (FileNotFoundError, json.JSONDecodeError, ValueError) as e:
             logging.error(f"init_api(): Failed to load credentials: {e}")
 
-    def _ensure_api_initialized(self):
-        """Check if the API is initialized before making API calls."""
-        if not self.pdf_api:
-            logging.error("ensure_api_initialized(): PDF API is not initialized. Operation aborted.")
-            return False
-        return True
-
     def upload_file(self, fileName: str):
         """ Upload a local fileName to the Aspose Cloud server. """
-        if not self._ensure_api_initialized():
-            return
-
-        file_path = Config.LOCAL_FOLDER / fileName
-        try:
-            self.pdf_api.upload_file(fileName, str(file_path))
-            logging.info(f"upload_file(): File '{fileName}' uploaded successfully.")
-        except Exception as e:
-            logging.error(f"upload_document(): Failed to upload file: {e}")
+        if self.pdf_api:
+            file_path = Config.LOCAL_FOLDER / fileName
+            try:
+                self.pdf_api.upload_file(fileName, str(file_path))
+                logging.info(f"upload_file(): File '{fileName}' uploaded successfully.")
+            except Exception as e:
+                logging.error(f"upload_document(): Failed to upload file: {e}")
 
     def upload_document(self):
         """ Upload a PDF document to the Aspose Cloud server. """
         self.upload_file(Config.PDF_DOCUMENT_NAME)
 
     def download_result(self):
         """ Download the processed PDF document from the Aspose Cloud server. """
-        if not self._ensure_api_initialized():
-            return
-
-        try:
-            temp_file = self.pdf_api.download_file(Config.PDF_DOCUMENT_NAME)
-            local_path = Config.LOCAL_FOLDER / Config.LOCAL_RESULT_DOCUMENT_NAME
-            shutil.move(temp_file, str(local_path))
-            logging.info(f"download_result(): File successfully downloaded: {local_path}")
-        except Exception as e:
-            logging.error(f"download_result(): Failed to download file: {e}")
+        if self.pdf_api:
+            try:
+                temp_file = self.pdf_api.download_file(Config.PDF_DOCUMENT_NAME)
+                local_path = Config.LOCAL_FOLDER / Config.LOCAL_RESULT_DOCUMENT_NAME
+                shutil.move(temp_file, str(local_path))
+                logging.info(f"download_result(): File successfully downloaded: {local_path}")
+            except Exception as e:
+                logging.error(f"download_result(): Failed to download file: {e}")
 
     def append_image_header(self):
         """Append a new image header to the PDF document."""
-        if not self._ensure_api_initialized():
-            return
-
-        new_header = ImageHeader(
-            background = True,
-            horizontal_alignment = HorizontalAlignment.CENTER,
-            file_name = Config.IMAGE_HEADER_FILE,
-            width = 24,
-            height = 24
-        )
-
-        try:
-            response = self.pdf_api.post_document_image_header(
-                Config.PDF_DOCUMENT_NAME, new_header
+        if self.pdf_api:
+            new_header = ImageHeader(
+                background = True,
+                horizontal_alignment = HorizontalAlignment.CENTER,
+                file_name = Config.IMAGE_HEADER_FILE,
+                width = 24,
+                height = 24
             )
-            if response.code == 200:
-                logging.info(f"append_image_header(): Header '{new_header.file_name}' added to the document '{Config.PDF_DOCUMENT_NAME}'.")
-            else:
-                logging.error(f"append_image_header(): Failed to add header '{new_header.file_name}' to the document '{Config.PDF_DOCUMENT_NAME}'. Response code: {response.code}")
-        except Exception as e:
-            logging.error(f"append_image_header(): Error while adding header: {e}")
+
+            try:
+                response = self.pdf_api.post_document_image_header(
+                    Config.PDF_DOCUMENT_NAME, new_header
+                )
+                if response.code == 200:
+                    logging.info(f"append_image_header(): Header '{new_header.file_name}' added to the document '{Config.PDF_DOCUMENT_NAME}'.")
+                else:
+                    logging.error(f"append_image_header(): Failed to add header '{new_header.file_name}' to the document '{Config.PDF_DOCUMENT_NAME}'. Response code: {response.code}")
+            except Exception as e:
+                logging.error(f"append_image_header(): Error while adding header: {e}")
 
     def append_image_heade_page(self):
         """Append a new image header to the page on PDF document."""
-        if not self._ensure_api_initialized():
-            return
-
-        new_header = ImageHeader(
-            background = True,
-            horizontal_alignment = HorizontalAlignment.LEFT,
-            file_name = Config.IMAGE_HEADER_FILE,
-            width = 24,
-            height = 24
-        )
-
-        try:
-            response = self.pdf_api.post_document_image_header(
-                Config.PDF_DOCUMENT_NAME, new_header, start_page_number=Config.PAGE_NUMBER, end_page_number=Config.PAGE_NUMBER
+        if self.pdf_api:
+            new_header = ImageHeader(
+                background = True,
+                horizontal_alignment = HorizontalAlignment.LEFT,
+                file_name = Config.IMAGE_HEADER_FILE,
+                width = 24,
+                height = 24
             )
-            if response.code == 200:
-                logging.info(f"append_image_heade_page(): Header '{new_header.file_name}' added to the page #{Config.PAGE_NUMBER}.")
-            else:
-                logging.error(f"append_image_heade_page(): Failed to add header '{new_header.file_name}' to the page #{Config.PAGE_NUMBER}. Response code: {response.code}")
-        except Exception as e:
-            logging.error(f"append_image_heade_page(): Error while adding header: {e}")
+
+            try:
+                response = self.pdf_api.post_document_image_header(
+                    Config.PDF_DOCUMENT_NAME, new_header, start_page_number=Config.PAGE_NUMBER, end_page_number=Config.PAGE_NUMBER
+                )
+                if response.code == 200:
+                    logging.info(f"append_image_heade_page(): Header '{new_header.file_name}' added to the page #{Config.PAGE_NUMBER}.")
+                else:
+                    logging.error(f"append_image_heade_page(): Failed to add header '{new_header.file_name}' to the page #{Config.PAGE_NUMBER}. Response code: {response.code}")
+            except Exception as e:
+                logging.error(f"append_image_heade_page(): Error while adding header: {e}")
 
 
 if __name__ == "__main__":