aboutsummaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorStewart Stewart <stewinsalot@gmail.com>2017-02-21 17:58:05 -0500
committerStewart Stewart <stewinsalot@gmail.com>2017-02-21 17:58:05 -0500
commit400867f01a32af311c40abd2fa6f1ddabe13deb5 (patch)
tree88f427b23de54d1e7fd6ba5c1759392fb308fed7 /src
parent49653139b80f9bad87e51aa0c8181fb48b64b15d (diff)
downloadslick-codegen-plugin-400867f01a32af311c40abd2fa6f1ddabe13deb5.tar.gz
slick-codegen-plugin-400867f01a32af311c40abd2fa6f1ddabe13deb5.tar.bz2
slick-codegen-plugin-400867f01a32af311c40abd2fa6f1ddabe13deb5.zip
override code to remove ForeignKeyAction import
Diffstat (limited to 'src')
-rw-r--r--src/main/scala/NamespacedCodegen.scala5
1 files changed, 2 insertions, 3 deletions
diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala
index 095eefb..4114f69 100644
--- a/src/main/scala/NamespacedCodegen.scala
+++ b/src/main/scala/NamespacedCodegen.scala
@@ -102,7 +102,7 @@ class Generator(pkg: String,
|}
|""".stripMargin
- val filteredCode = code.lines.drop(1).mkString
+ override def code = super.code.lines.drop(1).mkString("\n")
// Drops needless import: `"import slick.model.ForeignKeyAction\n"`.
// Alias to ForeignKeyAction is in profile.api
// TODO: fix upstream
@@ -124,9 +124,8 @@ class Generator(pkg: String,
|/** Slick data model trait for extension, choice of backend or usage in the cake pattern. (Make sure to initialize this late.) */
|trait $traitName${parentType.fold("")(" extends " + _)} {
| import profile.api._
- | ${indent(filteredCode)}
+ | ${indent(code)}
|}""".stripMargin.trim()
- // TODO: default to upstream packageCode after filteredCode is fixed
}
override def Table = new Table(_) { table =>