@@ -28,8 +28,11 @@ protected override void Execute(SourceProductionContext context,
2828 var collectionUpdateCode = GenerateCollectionUpdateCode ( item ) ;
2929 var includesCode = GetIncludesIfAny ( item . PropertiesInfos , mediaInfoPropertyNames , item . ClassName ) ;
3030 var authorizationTags = PropagateDappiAuthorizationTags ( item . AuthorizeAttributes , "GET" ) ;
31- var mediaInfoUpdateCode =
32- GenerateMediaInfoCreationCode ( "model" , "existingModel" , mediaInfoPropertyNames [ item . ClassName ] ) ;
31+ var mediaInfoUpdateCode = string . Empty ;
32+ if ( mediaInfoPropertyNames . ContainsKey ( item . ClassName ) )
33+ {
34+ mediaInfoUpdateCode = GenerateMediaInfoCreationCode ( "model" , "existingModel" , mediaInfoPropertyNames [ item . ClassName ] ) ;
35+ }
3336 ( string includeCode , string removeCode ) = GenerateDeleteCodeForMediaInfo ( item ) ;
3437 // TODO: Change to new project names
3538 var generatedCode = $@ "using Microsoft.AspNetCore.Mvc;
@@ -330,6 +333,10 @@ private static Dictionary<string, IEnumerable<string>> GetMediaInfoPropertyNames
330333 private static string GenerateMediaInfoCreationCode ( string updatedModelName , string existingModelName ,
331334 IEnumerable < string > mediaInfoPropertyNames )
332335 {
336+ if ( ! mediaInfoPropertyNames . Any ( ) )
337+ {
338+ return string . Empty ;
339+ }
333340 var sb = new StringBuilder ( ) ;
334341 foreach ( var prop in mediaInfoPropertyNames )
335342 {
0 commit comments