PartialFor - resolve name of partial from type name or from UIHint (like DisplayFor)

ArticlePageViewModel.cs

public class ArticlePageViewModel {
	
  // use partial view "TopMenuViewModel.cshtml"
  public TopMenuViewModel TopMenu {get;set;}
  
  // use partial view "WidgetsViewModel.cshtml"
  public WidgetsViewModel Widgets {get;set;}
  
  // use partial view "ExtraWidgets.cshtml"
  [UIHint("ExtraWidgets")]
  public WidgetsViewModel ExtraWidgets {get;set;}

}

example.cshtml

@Html.PartialFor(m=>m.TopMenu)

<div class="right">
  @Html.PartialFor(m=>m.Widgets)
</div>

<div class="extra">
  @Html.PartialFor(m=>m.ExtraWidgets)
</div>

<div class="bottom">
  @* using BottomWidgets.cshtml , this is not necessary as it equals the regular usage of html.partial *|
  @Html.PartialFor(m=>m.ExtraWidgets, "BottomWidgets")
</div>

PartialFor.cs

using System;
using System.Linq.Expressions;
using System.Web.Mvc;
using System.Web.Mvc.Html;
public static class PartialForExtensions
{

    public static MvcHtmlString PartialFor<T>(this HtmlHelper<T> htmlHelper, Expression<Func<T, object>> expression, string templateName = "")
    {
        var model = expression.Compile()(htmlHelper.ViewData.Model);

        // if model is null return blank string
        if (model == null) return new MvcHtmlString("");

        if (templateName == "")
        {
            // try resolve template name from property attribute UIHint (e.g. [UIHint("Foo")])
            var nm = ModelMetadata.FromLambdaExpression(expression, htmlHelper.ViewData);
            templateName = nm.TemplateHint;

            // try resolve template name from property type name
            if (templateName == null)
                templateName = model.GetType().Name;

        }

        return htmlHelper.Partial(templateName, model);

    }


}