checkpoint

1 files changed, 10 insertions, 13 deletions

diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala
index bdb9fc6..f08bd6c 100644
--- a/src/main/scala/NamespacedCodegen.scala
+++ b/src/main/scala/NamespacedCodegen.scala
@@ -46,6 +46,9 @@ object Generator {
                                        Some(Map(schemaName -> tables)))),
             Duration.Inf)
 
+          val camelSchemaName =
+            schemaName.split('_').map(_.capitalize).mkString("")
+
           val tableGenerator = new TableGenerator(pkg,
                                                   dbModel,
                                                   schemaOnlyModel,
@@ -54,13 +57,13 @@ object Generator {
                                                   idType,
                                                   header,
                                                   schemaImports,
-                                                  typeReplacements,
-                                                  schemaName)
+                                                  typeReplacements)
           tableGenerator.writeToFile(profile = profile,
                                      folder = outputPath,
-                                     pkg = pkg,
+                                     pkg = s"$pkg.$schemaName",
                                      container = schemaName,
-                                     fileName = s"${schemaName}.scala")
+                                     fileName =
+                                       s"${camelSchemaName}Tables.scala")
 
           val rowGenerator = new RowGenerator(pkg,
                                               dbModel,
@@ -73,9 +76,8 @@ object Generator {
                                               typeReplacements)
           rowGenerator.writeToFile(schemaName = schemaName,
                                    folder = outputPath,
-                                   pkg = pkg,
-                                   fileName =
-                                     s"${schemaName.capitalize}Rows.scala")
+                                   pkg = s"$pkg.$schemaName",
+                                   fileName = s"${camelSchemaName}Rows.scala")
       }
     } finally {
       dc.db.close()
@@ -339,8 +341,7 @@ class TableGenerator(
     idType: Option[String],
     override val headerComment: String,
     schemaImports: List[String],
-    typeReplacements: Map[String, String],
-    schemaName: String)
+    typeReplacements: Map[String, String])
     extends Generator(pkg,
                       fullDatabaseModel,
                       schemaOnlyModel,
@@ -355,9 +356,5 @@ class TableGenerator(
     override def EntityType = new EntityType {
       override def enabled = false
     }
-    override def TableClass = new TableClass {
-      override def elementType = s"$schemaName.${super.elementType}"
-      override def optionEnabled = false
-    }
   }
 }