Adding datetime_undef_if_invalid to squelch errors on DateTime inflation of bogus...
[dbsrgits/DBIx-Class.git] / lib / DBIx / Class / InflateColumn / DateTime.pm
1 package DBIx::Class::InflateColumn::DateTime;
2
3 use strict;
4 use warnings;
5 use base qw/DBIx::Class/;
6
7 =head1 NAME
8
9 DBIx::Class::InflateColumn::DateTime - Auto-create DateTime objects from date and datetime columns.
10
11 =head1 SYNOPSIS
12
13 Load this component and then declare one or more 
14 columns to be of the datetime, timestamp or date datatype.
15
16   package Event;
17   __PACKAGE__->load_components(qw/InflateColumn::DateTime Core/);
18   __PACKAGE__->add_columns(
19     starts_when => { data_type => 'datetime' }
20   );
21
22 Then you can treat the specified column as a L<DateTime> object.
23
24   print "This event starts the month of ".
25     $event->starts_when->month_name();
26
27 If you want to set a specific timezone for that field, use:
28
29   __PACKAGE__->add_columns(
30     starts_when => { data_type => 'datetime', extra => { timezone => "America/Chicago" } }
31   );
32
33 =head1 DESCRIPTION
34
35 This module figures out the type of DateTime::Format::* class to 
36 inflate/deflate with based on the type of DBIx::Class::Storage::DBI::* 
37 that you are using.  If you switch from one database to a different 
38 one your code should continue to work without modification (though note
39 that this feature is new as of 0.07, so it may not be perfect yet - bug
40 reports to the list very much welcome).
41
42 For more help with using components, see L<DBIx::Class::Manual::Component/USING>.
43
44 =cut
45
46 __PACKAGE__->load_components(qw/InflateColumn/);
47
48 __PACKAGE__->mk_group_accessors('simple' => '__datetime_parser');
49
50 =head2 register_column
51
52 Chains with the L<DBIx::Class::Row/register_column> method, and sets
53 up datetime columns appropriately.  This would not normally be
54 directly called by end users.
55
56 In the case of an invalid date, L<DateTime> will throw an exception.  To
57 bypass these exceptions and just have the inflation return undef, use
58 the C<datetime_undef_if_invalid> option in the column info:
59   
60     "broken_date",
61     {
62         data_type => "datetime",
63         default_value => '0000-00-00',
64         is_nullable => 1,
65         datetime_undef_if_invalid => 1
66     }
67
68 =cut
69
70 sub register_column {
71   my ($self, $column, $info, @rest) = @_;
72   $self->next::method($column, $info, @rest);
73   return unless defined($info->{data_type});
74   my $type = lc($info->{data_type});
75   $type = 'datetime' if ($type =~ /^timestamp/);
76   my $timezone;
77   if ( exists $info->{extra} and exists $info->{extra}{timezone} and defined $info->{extra}{timezone} ) {
78     $timezone = $info->{extra}{timezone};
79   }
80
81   if ($type eq 'datetime' || $type eq 'date') {
82     my ($parse, $format) = ("parse_${type}", "format_${type}");
83     $self->inflate_column(
84       $column =>
85         {
86           inflate => sub {
87             my ($value, $obj) = @_;
88             my $dt = eval { $obj->_datetime_parser->$parse($value); };
89             die "Error while inflating ${value} for ${column} on ${self}: $@" if $@ and not $info->{datetime_undef_if_invalid};
90             $dt->set_time_zone($timezone) if $timezone;
91             return $dt;
92           },
93           deflate => sub {
94             my ($value, $obj) = @_;
95             $value->set_time_zone($timezone) if $timezone;
96             $obj->_datetime_parser->$format($value);
97           },
98         }
99     );
100   }
101 }
102
103 sub _datetime_parser {
104   my $self = shift;
105   if (my $parser = $self->__datetime_parser) {
106     return $parser;
107   }
108   my $parser = $self->result_source->storage->datetime_parser(@_);
109   return $self->__datetime_parser($parser);
110 }
111
112 1;
113 __END__
114
115 =head1 SEE ALSO
116
117 =over 4
118
119 =item More information about the add_columns method, and column metadata, 
120       can be found in the documentation for L<DBIx::Class::ResultSource>.
121
122 =back
123
124 =head1 AUTHOR
125
126 Matt S. Trout <mst@shadowcatsystems.co.uk>
127
128 =head1 CONTRIBUTORS
129
130 Aran Deltac <bluefeet@cpan.org>
131
132 =head1 LICENSE
133
134 You may distribute this code under the same terms as Perl itself.
135