Change body serialization for complex type

Azure · May 10, 2016 · 3e34c16 · 3e34c16
1 parent cfdf8a8
commit 3e34c16
Show file tree

Hide file tree

Showing 3 changed files with 62 additions and 19 deletions.
diff --git a/AutoRest/Generators/Python/Azure.Python/Templates/AzureLongRunningMethodTemplate.cshtml b/AutoRest/Generators/Python/Azure.Python/Templates/AzureLongRunningMethodTemplate.cshtml
@@ -64,17 +64,32 @@
       {
         @:body_content = upload_gen(@parameter.Name)
         break;
-      }
+            }
+        }
     }
-  }
-  else if (Model.RequestBody.IsRequired)
-  {
+    else if (Model.RequestBody.IsRequired)
+    {
+
+        if (Model.RequestBody.Type is CompositeType)
+        {
+        @:body_content = self._serialize.body(@Model.RequestBody.Name, models.@(Model.RequestBody.Type.ToPythonRuntimeTypeString()))
+        }
+        else
+        {
         @:body_content = self._serialize.body(@Model.RequestBody.Name, '@(Model.RequestBody.Type.ToPythonRuntimeTypeString())')
-  }
-  else
-  {
+        }
+    }
+    else
+    {
         @:if @Model.RequestBody.Name is not None:
+            if (Model.RequestBody.Type is CompositeType)
+        {
+            @:body_content = self._serialize.body(@Model.RequestBody.Name, models.@(Model.RequestBody.Type.ToPythonRuntimeTypeString()))
+        }
+        else
+        {
             @:body_content = self._serialize.body(@Model.RequestBody.Name, '@(Model.RequestBody.Type.ToPythonRuntimeTypeString())')
+        }
         @:else:
             @:body_content = None
   }

diff --git a/AutoRest/Generators/Python/Azure.Python/Templates/AzurePagingMethodTemplate.cshtml b/AutoRest/Generators/Python/Azure.Python/Templates/AzurePagingMethodTemplate.cshtml
@@ -59,13 +59,27 @@
 {
             @:# Construct body
   if (Model.RequestBody.IsRequired)
-  {
+    {
+        if (Model.RequestBody.Type is CompositeType)
+        {
+            @:body_content = self._serialize.body(@Model.RequestBody.Name, models.@(Model.RequestBody.Type.ToPythonRuntimeTypeString()))
+        }
+        else
+        {
             @:body_content = self._serialize.body(@Model.RequestBody.Name, '@(Model.RequestBody.Type.ToPythonRuntimeTypeString())')
-  }
-  else
-  {
+        }
+    }
+    else
+    {
             @:if @Model.RequestBody.Name is not None:
+        if (Model.RequestBody.Type is CompositeType)
+        {
+                @:body_content = self._serialize.body(@Model.RequestBody.Name, models.@(Model.RequestBody.Type.ToPythonRuntimeTypeString()))
+        }
+        else
+        {
                 @:body_content = self._serialize.body(@Model.RequestBody.Name, '@(Model.RequestBody.Type.ToPythonRuntimeTypeString())')
+        }
             @:else:
                 @:body_content = None
   }

diff --git a/AutoRest/Generators/Python/Python/Templates/MethodTemplate.cshtml b/AutoRest/Generators/Python/Python/Templates/MethodTemplate.cshtml
@@ -64,17 +64,31 @@
       {
         @:body_content = self._client.stream_upload(@parameter.Name, callback)
         break;
-      }
-    }   
-  }
-  else if (Model.RequestBody.IsRequired)
-  {
+            }
+        }
+    }
+    else if (Model.RequestBody.IsRequired)
+    {
+        if (Model.RequestBody.Type is CompositeType)
+        {
+        @:body_content = self._serialize.body(@Model.RequestBody.Name, models.@(Model.RequestBody.Type.ToPythonRuntimeTypeString()))
+        }
+        else
+        {
         @:body_content = self._serialize.body(@Model.RequestBody.Name, '@(Model.RequestBody.Type.ToPythonRuntimeTypeString())')
-  }
-  else
-  {
+        }
+    }
+    else
+    {
         @:if @Model.RequestBody.Name is not None:
+        if (Model.RequestBody.Type is CompositeType)
+        {
+            @:body_content = self._serialize.body(@Model.RequestBody.Name, models.@(Model.RequestBody.Type.ToPythonRuntimeTypeString()))
+        }
+        else
+        {
             @:body_content = self._serialize.body(@Model.RequestBody.Name, '@(Model.RequestBody.Type.ToPythonRuntimeTypeString())')
+        }
         @:else:
             @:body_content = None
   }