X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=lib%2FDBIx%2FClass%2FRelationship%2FAccessor.pm;h=b77ce00f71bf1826f3638bed8052bffb2ea382b3;hb=096f421241;hp=5c2c38d84e03e1bf5658459fa7305d242f17171f;hpb=71e65b395be8133d54b110a499f9e9619111c1f6;p=dbsrgits%2FDBIx-Class.git

diff --git a/lib/DBIx/Class/Relationship/Accessor.pm b/lib/DBIx/Class/Relationship/Accessor.pm
index 5c2c38d..b77ce00 100644
--- a/lib/DBIx/Class/Relationship/Accessor.pm
+++ b/lib/DBIx/Class/Relationship/Accessor.pm
@@ -1,4 +1,5 @@
-package DBIx::Class::Relationship::Accessor;
+package # hide from PAUSE
+    DBIx::Class::Relationship::Accessor;
 
 use strict;
 use warnings;
@@ -29,7 +30,7 @@ sub add_relationship_accessor {
       }
     };
   } elsif ($acc_type eq 'filter') {
-    $class->throw("No such column $rel to filter")
+    $class->throw_exception("No such column $rel to filter")
        unless $class->has_column($rel);
     my $f_class = $class->relationship_info($rel)->{class};
     $class->inflate_column($rel,
@@ -39,7 +40,7 @@ sub add_relationship_accessor {
         },
         deflate => sub {
           my ($val, $self) = @_;
-          $self->throw("$val isn't a $f_class") unless $val->isa($f_class);
+          $self->throw_exception("$val isn't a $f_class") unless $val->isa($f_class);
           return ($val->_ident_values)[0];
             # WARNING: probably breaks for multi-pri sometimes. FIXME
         }
@@ -47,9 +48,10 @@ sub add_relationship_accessor {
     );
   } elsif ($acc_type eq 'multi') {
     $meth{$rel} = sub { shift->search_related($rel, @_) };
+    $meth{"${rel}_rs"} = sub { shift->search_related_rs($rel, @_) };
     $meth{"add_to_${rel}"} = sub { shift->create_related($rel, @_); };
   } else {
-    $class->throw("No such relationship accessor type $acc_type");
+    $class->throw_exception("No such relationship accessor type $acc_type");
   }
   {
     no strict 'refs';
@@ -60,4 +62,39 @@ sub add_relationship_accessor {
   }
 }
 
+sub new {
+  my ($class, $attrs, @rest) = @_;
+  my ($related, $info);
+  foreach my $key (keys %{$attrs||{}}) {
+    next unless $info = $class->relationship_info($key);
+    $related->{$key} = delete $attrs->{$key}
+      if ref $attrs->{$key}
+         && $info->{attrs}{accessor}
+         && $info->{attrs}{accessor} eq 'single';
+  }
+  my $obj = $class->next::method($attrs, @rest);
+  if ($related) {
+    $obj->{_relationship_data} = $related;
+    foreach my $rel (keys %$related) {
+      $obj->set_from_related($rel, $related->{$rel});
+    }
+  }
+  return $obj;
+}
+
+sub update {
+  my ($obj, $attrs, @rest) = @_;
+  my $info;
+  foreach my $key (keys %{$attrs||{}}) {
+    next unless $info = $obj->relationship_info($key);
+    if (ref $attrs->{$key} && $info->{attrs}{accessor}
+        && $info->{attrs}{accessor} eq 'single') {
+      my $rel = delete $attrs->{$key};
+      $obj->set_from_related($key => $rel);
+      $obj->{_relationship_data}{$key} = $rel;
+    }
+  }
+  return $obj->next::method($attrs, @rest);
+}
+
 1;