Added get_remote_url function in FileEntity class which will return the uploaded file url

Author: sujata-m

src/example.py

@@ -60,9 +60,13 @@ def process(message):
 basename = 'sample-file'
 suffix = datetime.datetime.now().strftime("%y%m%d_%H%M%S")
 filename = '_'.join([basename, suffix])
-test_file = container.create_file(f'{filename}.txt', 'text/plain')
+try:
+    test_file = container.create_file(f'{filename}.txt')
+except Exception as e:
+    print(e)
 print('Start uploading...')
 test_file.upload(file_like_io.read())
+print(test_file.get_remote_url())
 print('Uploaded Successfully')
 
 logger = Core.get_logger()

src/python_ms_core/core/storage/abstract/file_entity.py

@@ -3,13 +3,11 @@
 
 class FileEntity(ABC):
     name = ''
-    mimetype = 'text/plain'
     file_path = ''
 
     @abstractmethod
-    def __init__(self, name: str, mimetype: str = 'text/plain'):
+    def __init__(self, name: str):
         self.name = name
-        self.mimetype = mimetype
         self.file_path = name
         pass
 
@@ -24,3 +22,7 @@ def get_body_text(self):
     @abstractmethod
     def upload(self, upload_stream):
         pass
+
+    @abstractmethod
+    def get_remote_url(self):
+        pass

src/python_ms_core/core/storage/providers/azure/azure_file_entity.py

@@ -5,9 +5,10 @@
 class AzureFileEntity(file_entity.FileEntity):
     blob_client = BlobClient
 
-    def __init__(self, name, mimetype, blob_client):
-        super().__init__(name, mimetype)
+    def __init__(self, name, blob_client):
+        super().__init__(name)
         self.blob_client = blob_client
+        self._get_remote_url = None
 
     def get_stream(self):
         return self.blob_client.download_blob().readall()
@@ -16,4 +17,8 @@ def get_body_text(self):
         return self.blob_client.download_blob().content_as_text()
 
     def upload(self, upload_stream):
-        self.blob_client.upload_blob(self.file_path, upload_stream)
+        upload_file = self.blob_client.upload_blob(self.file_path, upload_stream)
+        self._get_remote_url = upload_file.url
+
+    def get_remote_url(self):
+        return self._get_remote_url

src/python_ms_core/core/storage/providers/azure/azure_storage_container.py

@@ -15,10 +15,8 @@ def list_files(self):
         files_list = [azure_file_entity.AzureFileEntity]
         for single_item in blob_iterator:
             blob_client = self.container_client.get_blob_client(single_item.name)
-            files_list.append(
-                azure_file_entity.AzureFileEntity(single_item.name, single_item.content_settings.content_type,
-                                                  blob_client=blob_client))
+            files_list.append(azure_file_entity.AzureFileEntity(single_item.name, blob_client=blob_client))
         return files_list
 
-    def create_file(self, name, mimetype):
-        return azure_file_entity.AzureFileEntity(name, mimetype, self.container_client)
+    def create_file(self, name):
+        return azure_file_entity.AzureFileEntity(name, self.container_client)