Support modification time for S3 & Azure (#176)

* Support modification time for S3

* Add test case for Azure and fix implementaion to support Azure

---------

Signed-off-by: Gayathri <[email protected]>

Author: gaya3-zipstack

src/unstract/sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.62.0"
+__version__ = "0.62.1"
 
 
 def get_sdk_version():

src/unstract/sdk/file_storage/impl.py

@@ -207,10 +207,23 @@ def modification_time(self, path: str) -> datetime:
             datetime: Last modified time in datetime
         """
         file_info = self.fs.info(path)
-        file_mtime = file_info["mtime"]
-        if not isinstance(file_mtime, datetime):
-            file_mtime = datetime.fromtimestamp(file_mtime)
-        return file_mtime
+
+        # Try different possible timestamp keys
+        file_mtime = None
+        for key in ["mtime", "LastModified", "last_modified"]:
+            file_mtime = file_info.get(key)
+            if file_mtime is not None:
+                break
+
+        if file_mtime is None:
+            raise FileOperationError(
+                f"Could not find modification time in file info: {file_info}"
+            )
+
+        if isinstance(file_mtime, datetime):
+            return file_mtime
+
+        return datetime.fromtimestamp(file_mtime)
 
     def mime_type(
         self,

tests/test_file_storage.py

@@ -1,3 +1,4 @@
+import datetime
 import io
 import json
 import os.path
@@ -358,6 +359,32 @@ def test_make_dir(file_storage, folder_path, expected_result):
     assert file_storage.exists(path=folder_path) == expected_result
 
 
+@pytest.mark.parametrize(
+    "file_storage, folder_path",
+    [
+        (file_storage(provider=FileStorageProvider.GCS), TEST_CONSTANTS.READ_PDF_FILE),
+        (
+            file_storage(provider=FileStorageProvider.LOCAL),
+            TEST_CONSTANTS.READ_TEXT_FILE,
+        ),
+        (file_storage(provider=FileStorageProvider.S3), TEST_CONSTANTS.READ_TEXT_FILE),
+        (
+            file_storage(provider=FileStorageProvider.MINIO),
+            TEST_CONSTANTS.READ_TEXT_FILE,
+        ),
+        (
+            file_storage(provider=FileStorageProvider.AZURE),
+            TEST_CONSTANTS.READ_TEXT_FILE,
+        ),
+    ],
+)
+def test_modification_time(file_storage, folder_path):
+    modification_time = file_storage.modification_time(path=folder_path)
+    assert modification_time is not None and isinstance(
+        modification_time, datetime.datetime
+    )
+
+
 @pytest.mark.parametrize(
     "file_storage, folder_path, expected_result",
     [