<%@ CodeTemplate Language="C#" TargetLanguage="Text" Debug="True" Description="Template description here." %>
<%@ Property Name="Tables" Type="SchemaExplorer.TableSchemaCollection" Category="Source" %>
<%@ Assembly Name="SchemaExplorer" %>
<%@ Assembly Name="CodeSmith.CustomProperties" %>
<%@ Assembly Name="System.Data" %>
<%@ Assembly Name="System.Design" %>
<%@ Import Namespace="System.Data" %>
<%@ Import Namespace="System.IO" %>
<%@ Import Namespace="SchemaExplorer" %>
<%@ Import Namespace="System.Design" %>
<%
// This is done so that we can make some assumptions about what each table
// can contain.
Tables = TableUtilities.FilterTables(Tables);
%>
<% if(!Directory.Exists(OutputDirectory)) Directory.CreateDirectory(OutputDirectory); %>
<%
//Load Sub Template
CodeTemplate ModelObjectTemplate = GetCodeTemplate("NewTemplate10.cst");
%>
<%
string ModelDirectory = OutputDirectory + "\\Build\\";
if(!Directory.Exists(ModelDirectory)) Directory.CreateDirectory(ModelDirectory);
%>
Create completed!
<%
//Render Files
for (int i = 0; i < Tables.Count; i++)
{
RenderFileUsingProperties(ModelDirectory + "CS_"+Tables[i].Name + ".txt",true,ref ModelObjectTemplate,Tables[i]);
}
%>
<script runat="template">
Output#region Output
private string _outputDirectory = String.Empty;
[Editor(typeof(System.Windows.Forms.Design.FolderNameEditor), typeof(System.Drawing.Design.UITypeEditor))]
[CodeTemplateProperty(CodeTemplatePropertyOption.Optional)]
[Category("General")]
[Description("The directory to output the results to.")]
[DefaultValue("")]
public string OutputDirectory
{
get
{
if (_outputDirectory.Length == 0)
{
return @"F:\";
}
else
{
return _outputDirectory;
}
}
set
{
if (value.EndsWith("\\")) value = value.Substring(0, value.Length - 1);
_outputDirectory = value;
}
}
#endregion
Get Template#region Get Template
public CodeTemplate GetCodeTemplate(string TemplateName)
{
//CodeTemplate template=null;
CodeTemplateCompiler compiler = new CodeTemplateCompiler(this.CodeTemplateInfo.DirectoryName+TemplateName);
compiler.CodeTemplateInfo.ToString();
compiler.Compile();
if (compiler.Errors.Count == 0)
{
return compiler.CreateInstance();
}
else
{
System.Text.StringBuilder errorMessage = new System.Text.StringBuilder();
for (int i = 0; i < compiler.Errors.Count; i++)
{
errorMessage.Append(compiler.Errors[i].ToString()).Append("\n");
}
throw new ApplicationException(errorMessage.ToString());
}
return null;
}
#endregion
Inner Classes#region Inner Classes
TableUtilities Class#region TableUtilities Class
public class TableUtilities
{
public static bool SkipTable(TableSchema table)
{
return (table.PrimaryKey == null || table.Columns.Count < 2);
}
public static TableSchemaCollection FilterTables(TableSchemaCollection tables)
{
TableSchemaCollection filtered = new TableSchemaCollection();
foreach(TableSchema table in tables)
{
if(!SkipTable(table)) filtered.Add(table);
}
return filtered;
}
}
#endregion
#endregion
Render File#region Render File
/**//// Copies all properties to a template
/// Renders the file
/// NB. the overrite property
public void RenderFileUsingProperties(string fullPath, bool overrite, ref CodeTemplate template)
{
if( overrite || (!File.Exists(fullPath)) )
{
CopyPropertiesTo(template);
template.RenderToFile(fullPath, overrite);
}
}
/**//// Copies all properties to a template
/// Renders the file
/// NB. the overrite property
/// Set Property SourceTable
public void RenderFileUsingProperties(string fullPath, bool overrite, ref CodeTemplate template,TableSchema SourceTable)
{
if( overrite || (!File.Exists(fullPath)) )
{
CopyPropertiesTo(template);
template.SetProperty("SourceTable",SourceTable);
template.RenderToFile(fullPath, overrite);
}
}
#endregion
</script>posted on 2007-05-25 22:05
过江 阅读(188)
评论(0) 编辑 收藏 所属分类:
CodeSmith使用实例