Update replaceLink.py

Author: Dmitriy-Xawstov

Uses-Cases/Links/replace/replaceLink.py

@@ -36,37 +36,26 @@ def _init_api(self, credentials_file: Path):
         except (FileNotFoundError, json.JSONDecodeError, ValueError) as e:
             logging.error(f"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("PDF API is not initialized. Operation aborted.")
-            return False
-        return True
-
     def upload_document(self):
         """Upload a PDF document to the Aspose Cloud server."""
-        if not self._ensure_api_initialized():
-            return
-
-        file_path = Config.LOCAL_FOLDER / Config.PDF_DOCUMENT_NAME
-        try:
-            self.pdf_api.upload_file(Config.PDF_DOCUMENT_NAME, str(file_path))
-            logging.info(f"File {Config.PDF_DOCUMENT_NAME} uploaded successfully.")
-        except Exception as e:
-            logging.error(f"Failed to upload file: {e}")
+        if self.pdf_api:
+            file_path = Config.LOCAL_FOLDER / Config.PDF_DOCUMENT_NAME
+            try:
+                self.pdf_api.upload_file(Config.PDF_DOCUMENT_NAME, str(file_path))
+                logging.info(f"File {Config.PDF_DOCUMENT_NAME} uploaded successfully.")
+            except Exception as e:
+                logging.error(f"Failed to upload file: {e}")
 
     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 show_links_array(self, links, prefix):
         """Show all hyperlink annotations for a specific PDF document."""
@@ -75,52 +64,45 @@ def show_links_array(self, links, prefix):
 
     def get_all_links(self):
         """Get all hyperlink annotations for a specific PDF document."""
-        if not self._ensure_api_initialized():
-            return
-
-        try:
-            response = self.pdf_api.get_page_link_annotations( Config.PDF_DOCUMENT_NAME, Config.PAGE_NUMBER)
-            if response.code == 200:
-                self.show_links_array(response.links.list, "All: ")
-            else:
-                logging.error(f"Failed to add link to the page. Response code: {response.code}")
-        except Exception as e:
-            logging.error(f"Error while adding link: No links found - {e}")
+        if self.pdf_api:
+            try:
+                response = self.pdf_api.get_page_link_annotations( Config.PDF_DOCUMENT_NAME, Config.PAGE_NUMBER)
+                if response.code == 200:
+                    self.show_links_array(response.links.list, "All: ")
+                else:
+                    logging.error(f"Failed to add link to the page. Response code: {response.code}")
+            except Exception as e:
+                logging.error(f"Error while adding link: No links found - {e}")
 
     def get_link_by_id(self) -> LinkAnnotationResponse:
-        if not self.pdf_api:
-            return None
+        if self.pdf_api:
+            result_link: LinkAnnotationResponse = self.pdf_api.get_link_annotation(Config.PDF_DOCUMENT_NAME, Config.LINK_FIND_ID)
 
-        result_link: LinkAnnotationResponse = self.pdf_api.get_link_annotation(Config.PDF_DOCUMENT_NAME, Config.LINK_FIND_ID)
+            if result_link.code == 200:
+                return result_link
 
-        if result_link.code == 200:
-            return result_link
-
-        print("Link not found.")
-        return None
+            print("Link not found.")
+            return None
 
     def replace_link(self):
-        if not self.pdf_api:
-            return
+        if self.pdf_api:
+            link_annotation: LinkAnnotationResponse = self.get_link_by_id()
 
-        link_annotation: LinkAnnotationResponse = self.get_link_by_id()
+            if not link_annotation:
+                return
 
-        if not link_annotation:
-            return
+            link_annotation.link.action = Config.NEW_LINK_ACTION
 
-        link_annotation.link.action = Config.NEW_LINK_ACTION
-
-        response: LinkAnnotationsResponse = self.pdf_api.put_link_annotation(
-            Config.PDF_DOCUMENT_NAME,
-            Config.LINK_FIND_ID,
-            link_annotation.link,
-        )
-
-        if response.code == 200:
-            print("Link annotation replaced successfully.")
-        else:
-            print("Failed to replace link annotation.")
+            response: LinkAnnotationsResponse = self.pdf_api.put_link_annotation(
+                Config.PDF_DOCUMENT_NAME,
+                Config.LINK_FIND_ID,
+                link_annotation.link,
+            )
 
+            if response.code == 200:
+                print("Link annotation replaced successfully.")
+            else:
+                print("Failed to replace link annotation.")
 
 if __name__ == "__main__":
     pdf_links = PdfLinks()