From: Jess Robinson <castaway@desert-island.me.uk>
Date: Sun, 22 Jan 2006 17:07:46 +0000 (+0000)
Subject: Oopsie, missed some bits ..
X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=commitdiff_plain;h=f8a88db3f8e08b54a5930944cfa9bb80085b921d;p=dbsrgits%2FDBIx-Class-Historic.git

Oopsie, missed some bits ..
---

diff --git a/lib/SQL/Translator/Producer/DBIx/Class/File.pm b/lib/SQL/Translator/Producer/DBIx/Class/File.pm
index e0702c0..736bd73 100644
--- 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";
 }