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

diff --git a/lib/Maypole/Model/CDBI.pm b/lib/Maypole/Model/CDBI.pm
index 83c118a..43b8c7b 100644
--- a/lib/Maypole/Model/CDBI.pm
+++ b/lib/Maypole/Model/CDBI.pm
@@ -3,6 +3,7 @@ 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;
@@ -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;