Oopsie, missed some bits ..

[dbsrgits/DBIx-Class.git] / lib / SQL / Translator / Producer / DBIx / Class / File.pm

diff --git a/lib/SQL/Translator/Producer/DBIx/Class/File.pm b/lib/SQL/Translator/Producer/DBIx/Class/File.pm
index e0702c0..736bd73 100644 (file)
--- a/lib/SQL/Translator/Producer/DBIx/Class/File.pm
+++ b/lib/SQL/Translator/Producer/DBIx/Class/File.pm
@@ -134,12 +134,15 @@ __PACKAGE__->table('${tname}');
         $tableoutput{$te} .= $tableextras{$te} . "\n";
     }
 
-
+    $schemaoutput .= "\n__PACKAGE__->load_classes(";
     foreach my $to (keys %tableoutput)
     {
         $output .= $tableoutput{$to};
+        $schemaoutput .= "'${to}',"; 
     }
+    chop $schemaoutput;
+    $schemaoutput .= ");\n";
 
-    print "$output\n";
-    return $output;
+#    print "$output\n";
+    return $schemaoutput . $output . "\n1;\n";
 }