diff --git a/src/Orchard/ContentManagement/Drivers/ContentFieldDriver.cs b/src/Orchard/ContentManagement/Drivers/ContentFieldDriver.cs index 31fb77c72..353fa1ce9 100644 --- a/src/Orchard/ContentManagement/Drivers/ContentFieldDriver.cs +++ b/src/Orchard/ContentManagement/Drivers/ContentFieldDriver.cs @@ -18,8 +18,12 @@ namespace Orchard.ContentManagement.Drivers { DriverResult IContentFieldDriver.BuildDisplayShape(BuildDisplayContext context) { return Process(context.ContentItem, (part, field) => { DriverResult result = Display(part, field, context.DisplayType, context.New); - result.ContentPart = part; - result.ContentField = field; + + if (result != null) { + result.ContentPart = part; + result.ContentField = field; + } + return result; }); } @@ -27,8 +31,12 @@ namespace Orchard.ContentManagement.Drivers { DriverResult IContentFieldDriver.BuildEditorShape(BuildEditorContext context) { return Process(context.ContentItem, (part, field) => { DriverResult result = Editor(part, field, context.New); - result.ContentPart = part; - result.ContentField = field; + + if (result != null) { + result.ContentPart = part; + result.ContentField = field; + } + return result; }); } @@ -36,8 +44,12 @@ namespace Orchard.ContentManagement.Drivers { DriverResult IContentFieldDriver.UpdateEditorShape(UpdateEditorContext context) { return Process(context.ContentItem, (part, field) => { DriverResult result = Editor(part, field, context.Updater, context.New); - result.ContentPart = part; - result.ContentField = field; + + if (result != null) { + result.ContentPart = part; + result.ContentField = field; + } + return result; }); } diff --git a/src/Orchard/ContentManagement/Drivers/ContentPartDriver.cs b/src/Orchard/ContentManagement/Drivers/ContentPartDriver.cs index 2aa1c4e4e..de71c6741 100644 --- a/src/Orchard/ContentManagement/Drivers/ContentPartDriver.cs +++ b/src/Orchard/ContentManagement/Drivers/ContentPartDriver.cs @@ -24,7 +24,11 @@ namespace Orchard.ContentManagement.Drivers { } DriverResult result = Display(part, context.DisplayType, context.New); - result.ContentPart = part; + + if(result != null ) { + result.ContentPart = part; + } + return result; } @@ -36,7 +40,11 @@ namespace Orchard.ContentManagement.Drivers { } DriverResult result = Editor(part, context.New); - result.ContentPart = part; + + if (result != null) { + result.ContentPart = part; + } + return result; } @@ -48,7 +56,11 @@ namespace Orchard.ContentManagement.Drivers { } DriverResult result = Editor(part, context.Updater, context.New); - result.ContentPart = part; + + if (result != null) { + result.ContentPart = part; + } + return result; }