enable None key for inherited models

microsoft · Nov 7, 2024 · 78e472c · 78e472c
1 parent a0a3b84
commit 78e472c
Showing 1 changed file with 4 additions and 12 deletions.
diff --git a/src/Kiota.Builder/Writers/Python/CodeMethodWriter.cs b/src/Kiota.Builder/Writers/Python/CodeMethodWriter.cs
@@ -647,20 +647,12 @@ private void WriteSerializerBodyForInheritedModel(bool inherits, CodeClass paren
         if (inherits)
             writer.WriteLine("super().serialize(writer)");
         foreach (var otherProp in parentClass
-                                    .GetPropertiesOfKind(CodePropertyKind.Custom)
-                                    .Where(static x => !x.ExistsInBaseType && !x.ReadOnly)
-                                    .OrderBy(static x => x.Name))
+                                        .GetPropertiesOfKind(CodePropertyKind.Custom)
+                                        .Where(static x => !x.ExistsInBaseType && !x.ReadOnly)
+                                        .OrderBy(static x => x.Name))
         {
             var serializationMethodName = GetSerializationMethodName(otherProp.Type);
-            var propertyName = otherProp.Name;
-            writer.WriteLine($"if self.{propertyName} is not None:");
-            writer.IncreaseIndent();
-            writer.WriteLine($"writer.{serializationMethodName}(\"{otherProp.WireName}\", self.{propertyName})");
-            writer.DecreaseIndent();
-            writer.WriteLine("else:");
-            writer.IncreaseIndent();
-            writer.WriteLine($"writer.{serializationMethodName}(\"{otherProp.WireName}\", None)");
-            writer.DecreaseIndent();
+            writer.WriteLine($"writer.{serializationMethodName}(\"{NoneKeyword}\", self.{otherProp.Name})");
         }
     }
     private void WriteSerializerBodyForUnionModel(CodeClass parentClass, LanguageWriter writer)