X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=lib%2FMaypole%2FModel%2FCDBI.pm;h=43b8c7be0d808b22879d7d1edc24e357a984fbfe;hb=48b7f94a68422ee2d1ee9ca6bf9cd5c2ee34e189;hp=2d2aa2b123a35c9d19ffe944eaa2acc949eff64b;hpb=d09e000fc977211b8bd1a77285a952421b60aa7a;p=maypole.git

diff --git a/lib/Maypole/Model/CDBI.pm b/lib/Maypole/Model/CDBI.pm
index 2d2aa2b..43b8c7b 100644
--- a/lib/Maypole/Model/CDBI.pm
+++ b/lib/Maypole/Model/CDBI.pm
@@ -1,8 +1,9 @@
-package Apache::MVC::Model::CDBI;
-use base qw(Apache::MVC::Model::Base Class::DBI);
+package Maypole::Model::CDBI;
+use base qw(Maypole::Model::Base Class::DBI);
 use Lingua::EN::Inflect::Number qw(to_PL);
 use Class::DBI::AsForm;
 use Class::DBI::FromCGI;
+use Class::DBI::Loader;
 use Class::DBI::AbstractSearch;
 use Class::DBI::Plugin::RetrieveAll;
 use Class::DBI::Pager;
@@ -11,7 +12,7 @@ use strict;
 
 =head1 NAME
 
-Apache::MVC::Model::CDBI - Model class based on Class::DBI
+Maypole::Model::CDBI - Model class based on Class::DBI
 
 =head1 DESCRIPTION
 
@@ -98,5 +99,22 @@ sub list :Exported {
         $r->objects([ $self->retrieve_all ]);
     }
 }
+
+sub setup_database {
+    my ($self, $config, $namespace, $dsn) = @_;
+    $config->{dsn} = $dsn;
+    $config->{loader} = Class::DBI::Loader->new(
+        namespace => $namespace,
+        dsn => $dsn
+    );
+    $config->{classes} = [ $config->{loader}->classes ];
+    $config->{tables}  = [ $config->{loader}->tables ];
+}
+
+sub class_of {
+    my ($self, $r, $table) = @_;
+    return $r->config->{loader}->_table2class($table);
+}
+
 1;