Notice Publisher

Author: Kolea Plesco

requirements.txt

@@ -21,3 +21,5 @@ agraph-python==101.0.10
 decorator~=5.1.1
 urllib3[secure]
 semantic-version==2.10.0
+pysftp
+

ted_sws/notice_publisher/__init__.py

@@ -0,0 +1,7 @@
+from abc import ABC
+from ted_sws.core.model.notice import Notice
+
+
+class NoticePublisherABC(ABC):
+    def upload(self, source_path, remote_path=None) -> bool:
+        pass

ted_sws/notice_publisher/adapters/__init__.py

@@ -0,0 +1,50 @@
+import pysftp
+
+from ted_sws.core.model.notice import Notice
+from ted_sws.notice_publisher.adapters.notice_publisher import NoticePublisherABC
+
+
+class SFTPNoticePublisher(NoticePublisherABC):
+    def __init__(self, hostname, username, password, port=22, remote_path=None):
+        """Constructor Method"""
+        # Set connection object to None (initial value)
+        self.connection = None
+        self.hostname = hostname
+        self.username = username
+        self.password = password
+        self.remote_path = remote_path
+        self.port = port
+
+    def connect(self):
+        """Connects to the sftp server and returns the sftp connection object"""
+
+        try:
+            # Get the sftp connection object
+            self.connection = pysftp.Connection(
+                host=self.hostname,
+                username=self.username,
+                password=self.password,
+                port=self.port,
+            )
+        except Exception as err:
+            raise Exception(err)
+
+    def disconnect(self):
+        """Closes the sftp connection"""
+        self.connection.close()
+
+    def upload(self, source_path, remote_path=None) -> bool:
+        """
+       Uploads the notice's METS manifestation to the sftp server remote path.
+       """
+        if remote_path is None:
+            remote_path = self.remote_path
+
+        if remote_path is None:
+            raise ValueError("No remote path specified.")
+
+        try:
+            self.connection.put(source_path, remote_path)
+            return True
+        except Exception as err:
+            raise Exception(err)

ted_sws/notice_publisher/adapters/notice_publisher.py

@@ -0,0 +1,34 @@
+from ted_sws.notice_publisher.adapters.notice_publisher import NoticePublisherABC
+from ted_sws.notice_publisher.adapters.sftp_notice_publisher import SFTPNoticePublisher
+from ted_sws.core.model.notice import Notice, NoticeStatus
+from ted_sws.data_manager.adapters.notice_repository import NoticeRepositoryABC
+import base64
+import tempfile
+
+
+def publish_notice_package(notice: Notice, notice_publisher: NoticePublisherABC, notice_repository: NoticeRepositoryABC,
+                           remote_path=None):
+    mets_manifestation = notice.mets_manifestation
+    if not mets_manifestation:
+        raise ValueError("Notice does not have a METS manifestation to be published.")
+
+    package_content = base64.b64decode(bytes(mets_manifestation.object_data), validate=True)
+
+    source_file = tempfile.NamedTemporaryFile()
+    source_file.write(package_content)
+
+    if notice_publisher.upload(source_file, remote_path):
+        notice.update_status_to(NoticeStatus.PUBLISHED)
+        notice_repository.update(notice)
+    else:
+        raise Exception(f"Notice {notice.ted_id} could not be published.")
+
+
+def publish_notice(notice: Notice, notice_repository: NoticeRepositoryABC, hostname, username, password,
+                   remote_path=None, port=22):
+    notice_publisher = SFTPNoticePublisher(hostname=hostname, username=username, password=password, port=port,
+                                           remote_path=remote_path)
+    notice_publisher.connect()
+    publish_notice_package(notice, notice_publisher, notice_repository)
+    notice_publisher.disconnect()
+    pass