add file as content

Author: AllyW

src/azure-cli-core/azure/cli/core/aaz/__init__.py

@@ -13,7 +13,7 @@
     AAZFreeFormDictArg, AAZFloatArg, AAZBaseArg, AAZBoolArg, AAZListArg, AAZResourceGroupNameArg, \
     AAZResourceLocationArg, AAZResourceIdArg, AAZSubscriptionIdArg, AAZUuidArg, AAZDateArg, AAZTimeArg, \
     AAZDateTimeArg, AAZDurationArg, AAZFileArg, AAZPasswordArg, AAZPaginationTokenArg, AAZPaginationLimitArg, \
-    AAZAnyTypeArg
+    AAZAnyTypeArg, AAZFileBytesArg
 from ._arg_fmt import AAZStrArgFormat, AAZIntArgFormat, AAZFloatArgFormat, AAZBoolArgFormat, AAZObjectArgFormat, \
     AAZDictArgFormat, AAZFreeFormDictArgFormat, AAZListArgFormat, AAZResourceLocationArgFormat, \
     AAZResourceIdArgFormat, AAZSubscriptionIdArgFormat, AAZUuidFormat, AAZDateFormat, AAZTimeFormat, \

src/azure-cli-core/azure/cli/core/aaz/_arg.py

@@ -22,7 +22,7 @@
 from ._arg_fmt import AAZObjectArgFormat, AAZListArgFormat, AAZDictArgFormat, AAZFreeFormDictArgFormat, \
     AAZSubscriptionIdArgFormat, AAZResourceLocationArgFormat, AAZResourceIdArgFormat, AAZUuidFormat, AAZDateFormat, \
     AAZTimeFormat, AAZDateTimeFormat, AAZDurationFormat, AAZFileArgTextFormat, AAZPaginationTokenArgFormat, \
-    AAZIntArgFormat
+    AAZIntArgFormat, AAZFileBytesArgFormat
 from .exceptions import AAZUnregisteredArg
 from ._prompt import AAZPromptInput
 
@@ -599,6 +599,18 @@ def __init__(self, fmt=None, **kwargs):
         super().__init__(fmt=fmt, **kwargs)
 
 
+class AAZFileBytesArg(AAZStrArg):
+    """Argument that accepts a file path and returns both file content in bytes"""
+    
+    def __init__(self, fmt=None, **kwargs):
+        fmt = fmt or AAZFileBytesArgFormat()
+        super().__init__(fmt=fmt, **kwargs)
+        
+    @property
+    def _type_in_help(self):
+        return "File Bytes"
+
+
 # Generic Update arguments
 class AAZGenericUpdateForceStringArg(AAZBoolArg):
 

src/azure-cli-core/azure/cli/core/aaz/_arg_fmt.py

@@ -748,6 +748,28 @@ def read_file(self, file_path):
         return data
 
 
+class AAZFileBytesArgFormat(AAZBaseArgFormat):
+    """Format for file info argument that returns both filename and content"""
+    
+    def __call__(self, ctx, value):
+        assert isinstance(value, AAZSimpleValue)
+        data = value._data
+        if data == AAZUndefined or data is None or value._is_patch:
+            return value
+
+        assert isinstance(data, str)
+        
+        if not os.path.isfile(data):
+            raise AAZInvalidArgValueError(f"File '{data}' doesn't exist")
+
+        try:
+            with open(data, 'rb') as f:
+                value._data = f.read()
+            return value
+        except Exception as e:
+            raise AAZInvalidArgValueError(f"Failed to read file '{data}': {str(e)}")
+
+
 class AAZPaginationTokenArgFormat(AAZBaseArgFormat):
     def __call__(self, ctx, value):
         def validate_json(s):