From 73b15dce5399029895572976a074a8d390164a05 Mon Sep 17 00:00:00 2001 From: Tony Tam Date: Tue, 28 Aug 2012 16:30:36 -0700 Subject: [PATCH] excluding default models --- .../swagger/codegen/BasicGenerator.scala | 24 ++++++++++--------- 1 file changed, 13 insertions(+), 11 deletions(-) diff --git a/src/main/scala/com/wordnik/swagger/codegen/BasicGenerator.scala b/src/main/scala/com/wordnik/swagger/codegen/BasicGenerator.scala index 4f6115793b4..fdb0bbb45ad 100644 --- a/src/main/scala/com/wordnik/swagger/codegen/BasicGenerator.scala +++ b/src/main/scala/com/wordnik/swagger/codegen/BasicGenerator.scala @@ -109,17 +109,19 @@ abstract class BasicGenerator extends CodegenConfig with PathUtil { val modelBundleList = new ListBuffer[Map[String, AnyRef]] for ((name, schema) <- allModels) { - val map = new HashMap[String, AnyRef] - map += "apis" -> None - map += "models" -> List((name, schema)) - map += "package" -> modelPackage - map += "invokerPackage" -> invokerPackage - map += "outputDirectory" -> (destinationDir + File.separator + modelPackage.getOrElse("").replaceAll("\\.", File.separator)) - map += "newline" -> "\n" - modelBundleList += map.toMap - for ((file, suffix) <- modelTemplateFiles) { - map += "filename" -> (name + suffix) - generateAndWrite(map.toMap, file) + if(!defaultIncludes.contains(name)){ + val map = new HashMap[String, AnyRef] + map += "apis" -> None + map += "models" -> List((name, schema)) + map += "package" -> modelPackage + map += "invokerPackage" -> invokerPackage + map += "outputDirectory" -> (destinationDir + File.separator + modelPackage.getOrElse("").replaceAll("\\.", File.separator)) + map += "newline" -> "\n" + modelBundleList += map.toMap + for ((file, suffix) <- modelTemplateFiles) { + map += "filename" -> (name + suffix) + generateAndWrite(map.toMap, file) + } } } codegen.writeSupportingClasses(apiMap.toMap, allModels.toMap)