[Storage Files] Migrating to TypeSpec

sdk/storage/azure-storage-file-share/MANIFEST.in

@@ -1,7 +1,7 @@
 include *.md
-include azure/__init__.py
-include azure/storage/__init__.py
 include LICENSE
+include azure/storage/fileshare/py.typed
 recursive-include tests *.py
 recursive-include samples *.py *.md
-include azure/storage/fileshare/py.typed
+include azure/__init__.py
+include azure/storage/__init__.py

sdk/storage/azure-storage-file-share/_metadata.json

@@ -0,0 +1,6 @@
+{
+  "apiVersion": "2026-06-06",
+  "apiVersions": {
+    "Storage.File": "2026-06-06"
+  }
+}

sdk/storage/azure-storage-file-share/azure/__init__.py

@@ -1 +1 @@
-__path__ = __import__('pkgutil').extend_path(__path__, __name__) # type: ignore
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore

sdk/storage/azure-storage-file-share/azure/storage/__init__.py

@@ -1 +1 @@
-__path__ = __import__('pkgutil').extend_path(__path__, __name__) # type: ignore
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore

sdk/storage/azure-storage-file-share/azure/storage/fileshare/__init__.py

@@ -42,52 +42,49 @@
     ShareProtocolSettings,
     ShareProtocols,
 )
-from ._generated.models import (
-    ShareAccessTier,
-    ShareRootSquash
-)
+from ._generated.models import ShareAccessTier, ShareRootSquash
 
 __version__ = VERSION
 
 
 __all__ = [
-    'AccessPolicy',
-    'AccountSasPermissions',
-    'ContentSettings',
-    'CorsRule',
-    'DirectoryProperties',
-    'ExponentialRetry',
-    'FileProperties',
-    'FileSasPermissions',
-    'generate_account_sas',
-    'generate_file_sas',
-    'generate_share_sas',
-    'Handle',
-    'LinearRetry',
-    'LocationMode',
-    'Metrics',
-    'NfsEncryptionInTransit',
-    'NTFSAttributes',
-    'ResourceTypes',
-    'RetentionPolicy',
-    'Services',
-    'ShareAccessTier',
-    'ShareClient',
-    'ShareDirectoryClient',
-    'ShareFileClient',
-    'ShareLeaseClient',
-    'ShareNfsSettings',
-    'ShareProperties',
-    'ShareProtocolSettings',
-    'ShareProtocols',
-    'ShareRootSquash',
-    'ShareSasPermissions',
-    'ShareServiceClient',
-    'ShareSmbSettings',
-    'SmbEncryptionInTransit',
-    'SmbMultichannel',
-    'StorageErrorCode',
-    'UserDelegationKey'
+    "AccessPolicy",
+    "AccountSasPermissions",
+    "ContentSettings",
+    "CorsRule",
+    "DirectoryProperties",
+    "ExponentialRetry",
+    "FileProperties",
+    "FileSasPermissions",
+    "generate_account_sas",
+    "generate_file_sas",
+    "generate_share_sas",
+    "Handle",
+    "LinearRetry",
+    "LocationMode",
+    "Metrics",
+    "NfsEncryptionInTransit",
+    "NTFSAttributes",
+    "ResourceTypes",
+    "RetentionPolicy",
+    "Services",
+    "ShareAccessTier",
+    "ShareClient",
+    "ShareDirectoryClient",
+    "ShareFileClient",
+    "ShareLeaseClient",
+    "ShareNfsSettings",
+    "ShareProperties",
+    "ShareProtocolSettings",
+    "ShareProtocols",
+    "ShareRootSquash",
+    "ShareSasPermissions",
+    "ShareServiceClient",
+    "ShareSmbSettings",
+    "SmbEncryptionInTransit",
+    "SmbMultichannel",
+    "StorageErrorCode",
+    "UserDelegationKey",
 ]
 
 
@@ -96,12 +93,10 @@
 # to prevent it from showing in intellisense/docs but we handle it here to prevent
 # breaking any existing code which may have imported it.
 def __getattr__(name):
-    if name == 'HandleItem':
+    if name == "HandleItem":
         from ._generated.models import HandleItem
-        warnings.warn(
-            "HandleItem is deprecated and should not be used. Use Handle instead.",
-            DeprecationWarning
-        )
+
+        warnings.warn("HandleItem is deprecated and should not be used. Use Handle instead.", DeprecationWarning)
         return HandleItem
 
     raise AttributeError(f"module 'azure.storage.fileshare' has no attribute {name}")

sdk/storage/azure-storage-file-share/azure/storage/fileshare/_deserialize.py

@@ -3,10 +3,7 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------_
-from typing import (
-    Any, cast, Dict, List, Optional, Tuple,
-    TYPE_CHECKING
-)
+from typing import Any, cast, Dict, List, Optional, Tuple, TYPE_CHECKING
 
 from ._generated.models import ShareFileRangeList
 from ._models import DirectoryProperties, FileProperties, ShareProperties
@@ -19,67 +16,89 @@
 
 def deserialize_share_properties(response: "PipelineResponse", obj: Any, headers: Dict[str, Any]) -> ShareProperties:
     metadata = deserialize_metadata(response, obj, headers)
-    share_properties = ShareProperties(
-        metadata=metadata,
-        **headers
-    )
+    share_properties = ShareProperties(metadata=metadata, **headers)
     return share_properties
 
 
 def deserialize_directory_properties(
-    response: "PipelineResponse",
-    obj: Any,
-    headers: Dict[str, Any]
+    response: "PipelineResponse", obj: Any, headers: Dict[str, Any]
 ) -> DirectoryProperties:
     metadata = deserialize_metadata(response, obj, headers)
-    directory_properties = DirectoryProperties(
-        metadata=metadata,
-        **headers
-    )
+    directory_properties = DirectoryProperties(metadata=metadata, **headers)
     return directory_properties
 
 
+class _StreamWrapper:
+    """Wraps a bytes iterator (sync or async) so that properties can be attached."""
+
+    def __init__(self, stream):
+        self._stream = stream
+        self.properties = None
+        self.response = None
+
+    def __iter__(self):
+        return iter(self._stream)
+
+    def __next__(self):
+        return next(self._stream)
+
+    def __aiter__(self):
+        return self._stream.__aiter__()
+
+    async def __anext__(self):
+        return await self._stream.__anext__()
+
+
 def deserialize_file_properties(response: "PipelineResponse", obj: Any, headers: Dict[str, Any]) -> FileProperties:
     metadata = deserialize_metadata(response, obj, headers)
-    file_properties = FileProperties(
-        metadata=metadata,
-        **headers
-    )
-    if 'Content-Range' in headers:
-        if 'x-ms-content-md5' in headers:
-            file_properties.content_settings.content_md5 = headers['x-ms-content-md5']
+    file_properties = FileProperties(metadata=metadata, **headers)
+    if "Content-Range" in headers:
+        if "x-ms-content-md5" in headers:
+            file_properties.content_settings.content_md5 = headers["x-ms-content-md5"]
         else:
             file_properties.content_settings.content_md5 = None
     return file_properties
 
 
 def deserialize_file_stream(
-    response: "PipelineResponse",
-    obj: Any,
-    headers: Dict[str, Any]
+    response: "PipelineResponse", obj: Any, headers: Dict[str, Any]
 ) -> Tuple["LocationMode", Any]:
     file_properties = deserialize_file_properties(response, obj, headers)
+    # The new TypeSpec-generated download returns an iterator (from iter_bytes/iter_raw)
+    # instead of a response object with settable attributes. Wrap it.
+    if not hasattr(obj, "properties") or isinstance(obj, type(iter(b""))):
+        wrapped = _StreamWrapper(obj)
+        wrapped.properties = file_properties
+        try:
+            wrapped.response = response.http_response
+        except AttributeError:
+            pass
+        return response.http_response.location_mode, wrapped
     obj.properties = file_properties
     return response.http_response.location_mode, obj
 
 
 # Extracts out file permission
-def deserialize_permission(response: "PipelineResponse", obj: Any, headers: Dict[str, Any]) -> Optional[str]:  # pylint: disable=unused-argument
+def deserialize_permission(  # pylint: disable=unused-argument
+    response: "PipelineResponse", obj: Any, headers: Dict[str, Any]
+) -> Optional[str]:
     return cast(Optional[str], obj.permission)
 
 
 # Extracts out file permission key
-def deserialize_permission_key(response: "PipelineResponse", obj: Any, headers: Dict[str, Any]) -> Optional[str]:  # pylint: disable=unused-argument
+def deserialize_permission_key(  # pylint: disable=unused-argument
+    response: "PipelineResponse", obj: Any, headers: Dict[str, Any]
+) -> Optional[str]:
     if response is None or headers is None:
         return None
-    return cast(Optional[str], headers.get('x-ms-file-permission-key', None))
+    return cast(Optional[str], headers.get("x-ms-file-permission-key", None))
 
 
 def get_file_ranges_result(ranges: ShareFileRangeList) -> Tuple[List[Dict[str, int]], List[Dict[str, int]]]:
     file_ranges = []
     clear_ranges = []
     if ranges.ranges:
-        file_ranges = [{'start': file_range.start, 'end': file_range.end} for file_range in ranges.ranges]
+        file_ranges = [{"start": file_range.start, "end": file_range.end} for file_range in ranges.ranges]
     if ranges.clear_ranges:
-        clear_ranges = [{'start': clear_range.start, 'end': clear_range.end} for clear_range in ranges.clear_ranges]
+        clear_ranges = [{"start": clear_range.start, "end": clear_range.end} for clear_range in ranges.clear_ranges]
     return file_ranges, clear_ranges