如何在CodeAttribute中获取属性值

时间:2015-04-29 13:28:38

标签: c# .net visual-studio t4

我写了一个方法来获取属性值属性:

public string GetAttributeValueByNameAttributeAndProperty(CodeClass cc, string nameAttribute, string nameProperty)
{
    var value = "";
    foreach(CodeAttribute ca in cc.Attributes) 
            { 
                if(ca.Name.Contains(nameAttribute) && ca.Value.Contains(nameProperty))
                {
                    value = ca.Value.Remove(0,ca.Value.IndexOf(nameProperty));
                    value = value.Replace(" ","");
                    if(value.Contains(","))
                        value = value.Remove(ca.Value.IndexOf(","));
                }
            }

     return value;
}

例如: 我有属性[Map(Name = "MenuItem, Availability" )]

我调用GetAttributeValueByNameAttributeAndProperty(codeclass," Map"," Name") 在该方法之后获取CodeAttribute.Value并返回字符串:Name =" MenuItem,Availability" 删除" Name ="和额外的字符和拆分","

但是我的高级程序员告诉我这个方法不灵活,我需要找到一种更方便的方法来获取CodeAttribute.Value中的内部数据。

你有什么想法/例子吗?

1 个答案:

答案 0 :(得分:2)

您可以使用CodeClass.Attributes属性来获取类的属性。每个属性的类型为CodeAttribute,并且具有NameChildren属性,其中包含属性的参数。每个参数的类型为CodeAttributeArgument,其中包含NameValue属性。

示例

现在,您拥有从CodeAttribute获取属性值所需的所有信息。这是一个例子。我使用Program属性

修饰了[MySample(Property1 = "Something")]
using System;
namespace ConsoleSample
{
    [MySample(Property1 = "Something")]
    class Program
    {
        static void Main(string[] args)
        {
        }
    }
    public class MySampleAttribute : Attribute
    {
        public string Property1 { get; set; }
    }
}

以下是示例T4模板:

<#@ template debug="true" hostSpecific="true" #>
<#@ output extension=".txt" #>
<#@ assembly Name="System.Core" #>
<#@ assembly name="EnvDte" #>
<#@ assembly name="EnvDte80" #>
<#@ import namespace="System" #>
<#@ import namespace="System.Linq" #>
<#@ import namespace="System.Collections.Generic" #> 
<#@ import namespace="EnvDTE" #> 
<#@ import namespace="EnvDTE80" #> 
<#    
var env = (this.Host as IServiceProvider).GetService(typeof(EnvDTE.DTE)) as EnvDTE.DTE;
var project = env.Solution.FindProjectItem(this.Host.TemplateFile).ContainingProject
    as EnvDTE.Project;
var codeClass = project.ProjectItems.Item("Program.cs").FileCodeModel.CodeElements
                       .OfType<CodeNamespace>().ToList()[0]
                       .Members.OfType<CodeClass>().ToList()[0];
var attribute = codeClass.Attributes.Cast<CodeAttribute>()
                         .Where(a=>a.Name=="MySample").FirstOrDefault();
if(attribute!=null)
{
    var property = attribute.Children.OfType<CodeAttributeArgument>()
                            .Where(a=>a.Name=="Property1").FirstOrDefault();
    if(property!=null)
    {
        var value = property.Value;
        WriteLine(value);
    }
}
#>

如果您运行模板,则会在输出中收到"Something"