2013-06-20 121 views
1

我上T4Scaffolding工作,我使用模型來生成代碼,事情是模型,那是POCO的數據庫,它必須有一個定義主鍵,否則我的代碼失敗,T4腳手架 - PowerShell腳本

這裏是我的命令

腳手架MyScaffolder smArtistDetails -Force

這裏是的.ps1腳本

[T4Scaffolding.Scaffolder(Description = "Enter a description of ExtMapping here")][CmdletBinding()] 
param(  
    [parameter(Mandatory = $true, ValueFromPipelineByPropertyName = $true)] 
    [string]$ModelType, 
    [string]$Project, 
    [string]$CodeLanguage, 
    [string[]]$TemplateFolders, 
    [switch]$Force = $false 
) 

$mynamespace = "Sma" 
$projectFor = $mynamespace + ".Infrastructure.Data" 
$namespace = $projectFor # (Get-Project $projectFor).Properties.Item("DefaultNamespace").Value 
$outputPath = "Mapping\" + $ModelType 
Write-Host "Scaffolding $ModelType ExtMapping..." 

$useModelFromProject = $mynamespace + ".Domain" 
# get the modelType object from the name 
$foundModelType = Get-ProjectType $ModelType -Project $useModelFromProject -BlockUi 
$primaryKey = Get-PrimaryKey $ModelType -Project $useModelFromProject 


$modelOutputPath = "Mapping\" + $ModelType + "Map" 

Add-ProjectItemViaTemplate $modelOutputPath -Template ExtMapping ` 
    -Model @{ 
    Namespace = $namespace; 
    ModelType = [MarshalByRefObject]$foundModelType; 
    PrimaryKey = $primaryKey; 
    EntityName = $ModelType } ` 
    -SuccessMessage "Added ExtMapping output at $projectFor" ` 
    -TemplateFolders $TemplateFolders -Project $projectFor -CodeLanguage $CodeLanguage -Force:$Force 

這裏是我74模板代碼

<#@ Template Language="C#" HostSpecific="True" Inherits="DynamicTransform" #> 
<#@ Output Extension="cs" #> 
<#@ import namespace="System.Collections" #> 
<#@ import namespace="System.Collections.Generic" #> 
<#@ import namespace="System" #> 
<#@ import namespace="EnvDTE" #> 
<#@ assembly name="System.ComponentModel.DataAnnotations" #> 
<#@ assembly name="System.Core" #> 
<#@ assembly name="System.Data.Entity" #> 
<#@ assembly name="System.Data.Linq" #> 
<#@ import namespace="System.Data.Linq.Mapping" #> 
<#@ import namespace="System.Linq" #> 
<#@ import namespace="System.Collections.Generic" #> 
<#@ import namespace="System.Data.Objects.DataClasses" #> 
<#@ import namespace="System.Reflection" #> 
<#@ import namespace="System.Collections" #> 
<#@ import namespace="System.Text.RegularExpressions" #> 

using System; 
using System.Collections.Generic; 
using System.Text; 
using FluentNHibernate.Mapping; 
using Sma.Domain.Model; 
<# 
    var modelType = (CodeType)Model.ModelType; 
    var modelName = modelType.Name; 
    var properties = modelType.VisibleMembers().OfType<CodeProperty>();  
    var primaryKeyProperty = modelType.VisibleMembers().OfType<CodeProperty>().Single(x => x.Name == Model.PrimaryKey); 
#> 

namespace <#= Model.Namespace #>.Mapping 
{ 
    public class <#= Model.EntityName #>Map : ClassMap<<#= Model.EntityName #>> 
    {  
     public <#= Model.EntityName #>Map() { 
     Table("<#= Model.EntityName #>"); 
     LazyLoad();  
<# var type = ""; 
var typeName = ""; 
foreach (var property in properties) 
{ 
type = GetType(property.Type); 
typeName = GetTypeName(property.Type); 
if(isType(type)) 
{ 
if(Model.PrimaryKey == property.Name) 
{ 
#> 
     Id(x => x.<#= property.Name #>).GeneratedBy.Identity().Column("<#= property.Name #>"); 
<#  
} 
else 
{ 
#> 
     Map(x=>x.<#= property.Name #>).Column("<#= property.Name #>");  
<#} 
} 
else 
{ 
#> 
     References(x=>x.<#= property.Name #>);    
<# 
} 
} 
#>  } 
    } 
} 

<#+ 
    string GetType(EnvDTE.CodeTypeRef type) 
    {                                    
     string str = ""; 
     if (type.UnderlyingTypeIs<byte>()) 
      str = "byte"; 
     else if (type.UnderlyingTypeIs<sbyte>()) 
      str = "sbyte"; 
     else if (type.UnderlyingTypeIs<int>()) 
      str = "int"; 
     else if (type.UnderlyingTypeIs<uint>()) 
      str = "uint"; 
     else if (type.UnderlyingTypeIs<short>()) 
      str = "short"; 
     else if (type.UnderlyingTypeIs<ushort>()) 
      str = "ushort"; 
     else if (type.UnderlyingTypeIs<DateTime>()) 
      str = "DateTime"; 
     else if (type.UnderlyingTypeIs<long>()) 
      str = "long"; 
     else if (type.UnderlyingTypeIs<ulong>()) 
      str = "ulong"; 
     else if (type.UnderlyingTypeIs<double>()) 
      str = "double"; 
     else if (type.UnderlyingTypeIs<bool>()) 
      str = "bool"; 
     else if (type.UnderlyingTypeIs<string>()) 
      str = "string"; 
     else if (type.UnderlyingTypeIs<decimal>()) 
      str = "decimal"; 
     else 
      str = type.AsFullName.ToString(); 

     return str; 
    } 

    string GetTypeName(EnvDTE.CodeTypeRef type) 
    { 
     string str = type.AsFullName.ToString(); 
     return str; 
    } 

    bool isType(string _type) 
    { 
     if(_type == "byte") return true; 
     else if(_type == "sbyte") return true; 
     else if(_type == "int") return true; 
     else if(_type == "uint") return true; 
     else if(_type == "short") return true; 
     else if(_type == "ushort") return true; 
     else if(_type == "DateTime") return true; 
     else if(_type == "long") return true; 
     else if(_type == "ulong") return true; 
     else if(_type == "double") return true; 
     else if(_type == "bool") return true; 
     else if(_type == "string") return true; 
     else if(_type == "decimal") return true; 
     return false; 
    } 
#> 

做什麼。如果主鍵沒有找到我想要跟隨不同的輸出\ genration方法。

我應該從腳本切換到不同的模板,如果是,那麼如何

或者我應該處理這此模板,如果是那麼如何。

回答

1

我會檢查PK是否在PS中找到,如果不調用另一個模板...我個人會強制模型中的所有類繼承已經有主鍵的抽象基類。這樣,這永遠不會是一個問題。但是再一次,你的目的可能是我不知道的...