Bug 15030: Add tests
[koha.git] / Koha / Calendar.pm
blobf6cb58c0f5e1aa5a052fbcae62a809bc6313a8ec
1 package Koha::Calendar;
2 use strict;
3 use warnings;
4 use 5.010;
6 use DateTime;
7 use DateTime::Set;
8 use DateTime::Duration;
9 use C4::Context;
10 use Koha::Caches;
11 use Carp;
13 sub new {
14 my ( $classname, %options ) = @_;
15 my $self = {};
16 bless $self, $classname;
17 for my $o_name ( keys %options ) {
18 my $o = lc $o_name;
19 $self->{$o} = $options{$o_name};
21 if ( !defined $self->{branchcode} ) {
22 croak 'No branchcode argument passed to Koha::Calendar->new';
24 $self->_init();
25 return $self;
28 sub _init {
29 my $self = shift;
30 my $branch = $self->{branchcode};
31 my $dbh = C4::Context->dbh();
32 my $weekly_closed_days_sth = $dbh->prepare(
33 'SELECT weekday FROM repeatable_holidays WHERE branchcode = ? AND weekday IS NOT NULL'
35 $weekly_closed_days_sth->execute( $branch );
36 $self->{weekly_closed_days} = [ 0, 0, 0, 0, 0, 0, 0 ];
37 while ( my $tuple = $weekly_closed_days_sth->fetchrow_hashref ) {
38 $self->{weekly_closed_days}->[ $tuple->{weekday} ] = 1;
40 my $day_month_closed_days_sth = $dbh->prepare(
41 'SELECT day, month FROM repeatable_holidays WHERE branchcode = ? AND weekday IS NULL'
43 $day_month_closed_days_sth->execute( $branch );
44 $self->{day_month_closed_days} = {};
45 while ( my $tuple = $day_month_closed_days_sth->fetchrow_hashref ) {
46 $self->{day_month_closed_days}->{ $tuple->{month} }->{ $tuple->{day} } =
50 $self->{days_mode} = C4::Context->preference('useDaysMode');
51 $self->{test} = 0;
52 return;
55 sub exception_holidays {
56 my ( $self ) = @_;
58 my $cache = Koha::Caches->get_instance();
59 my $cached = $cache->get_from_cache('exception_holidays');
60 return $cached if $cached;
62 my $dbh = C4::Context->dbh;
63 my $branch = $self->{branchcode};
64 my $exception_holidays_sth = $dbh->prepare(
65 'SELECT day, month, year FROM special_holidays WHERE branchcode = ? AND isexception = 1'
67 $exception_holidays_sth->execute( $branch );
68 my $dates = [];
69 while ( my ( $day, $month, $year ) = $exception_holidays_sth->fetchrow ) {
70 push @{$dates},
71 DateTime->new(
72 day => $day,
73 month => $month,
74 year => $year,
75 time_zone => "floating",
76 )->truncate( to => 'day' );
78 $self->{exception_holidays} =
79 DateTime::Set->from_datetimes( dates => $dates );
80 $cache->set_in_cache( 'exception_holidays', $self->{exception_holidays} );
81 return $self->{exception_holidays};
84 sub single_holidays {
85 my ( $self, $date ) = @_;
86 my $branchcode = $self->{branchcode};
87 my $cache = Koha::Caches->get_instance();
88 my $single_holidays = $cache->get_from_cache('single_holidays');
90 # $single_holidays looks like:
91 # {
92 # CPL => [
93 # [0] 20131122,
94 # ...
95 # ],
96 # ...
97 # }
99 unless ($single_holidays) {
100 my $dbh = C4::Context->dbh;
101 $single_holidays = {};
103 # push holidays for each branch
104 my $branches_sth =
105 $dbh->prepare('SELECT distinct(branchcode) FROM special_holidays');
106 $branches_sth->execute();
107 while ( my $br = $branches_sth->fetchrow ) {
108 my $single_holidays_sth = $dbh->prepare(
109 'SELECT day, month, year FROM special_holidays WHERE branchcode = ? AND isexception = 0'
111 $single_holidays_sth->execute($br);
113 my @ymd_arr;
114 while ( my ( $day, $month, $year ) =
115 $single_holidays_sth->fetchrow )
117 my $dt = DateTime->new(
118 day => $day,
119 month => $month,
120 year => $year,
121 time_zone => 'floating',
122 )->truncate( to => 'day' );
123 push @ymd_arr, $dt->ymd('');
125 $single_holidays->{$br} = \@ymd_arr;
126 } # br
127 $cache->set_in_cache( 'single_holidays', $single_holidays,
128 { expiry => 76800 } ) #24 hrs ;
130 my $holidays = ( $single_holidays->{$branchcode} );
131 for my $hols (@$holidays ) {
132 return 1 if ( $date == $hols ) #match ymds;
134 return 0;
137 sub addDate {
138 my ( $self, $startdate, $add_duration, $unit ) = @_;
140 # Default to days duration (legacy support I guess)
141 if ( ref $add_duration ne 'DateTime::Duration' ) {
142 $add_duration = DateTime::Duration->new( days => $add_duration );
145 $unit ||= 'days'; # default days ?
146 my $dt;
148 if ( $unit eq 'hours' ) {
149 # Fixed for legacy support. Should be set as a branch parameter
150 my $return_by_hour = 10;
152 $dt = $self->addHours($startdate, $add_duration, $return_by_hour);
153 } else {
154 # days
155 $dt = $self->addDays($startdate, $add_duration);
158 return $dt;
161 sub addHours {
162 my ( $self, $startdate, $hours_duration, $return_by_hour ) = @_;
163 my $base_date = $startdate->clone();
165 $base_date->add_duration($hours_duration);
167 # If we are using the calendar behave for now as if Datedue
168 # was the chosen option (current intended behaviour)
170 if ( $self->{days_mode} ne 'Days' &&
171 $self->is_holiday($base_date) ) {
173 if ( $hours_duration->is_negative() ) {
174 $base_date = $self->prev_open_day($base_date);
175 } else {
176 $base_date = $self->next_open_day($base_date);
179 $base_date->set_hour($return_by_hour);
183 return $base_date;
186 sub addDays {
187 my ( $self, $startdate, $days_duration ) = @_;
188 my $base_date = $startdate->clone();
190 $self->{days_mode} ||= q{};
192 if ( $self->{days_mode} eq 'Calendar' ) {
193 # use the calendar to skip all days the library is closed
194 # when adding
195 my $days = abs $days_duration->in_units('days');
197 if ( $days_duration->is_negative() ) {
198 while ($days) {
199 $base_date = $self->prev_open_day($base_date);
200 --$days;
202 } else {
203 while ($days) {
204 $base_date = $self->next_open_day($base_date);
205 --$days;
209 } else { # Days or Datedue
210 # use straight days, then use calendar to push
211 # the date to the next open day if Datedue
212 $base_date->add_duration($days_duration);
214 if ( $self->{days_mode} eq 'Datedue' ) {
215 # Datedue, then use the calendar to push
216 # the date to the next open day if holiday
217 if ( $self->is_holiday($base_date) ) {
219 if ( $days_duration->is_negative() ) {
220 $base_date = $self->prev_open_day($base_date);
221 } else {
222 $base_date = $self->next_open_day($base_date);
228 return $base_date;
231 sub is_holiday {
232 my ( $self, $dt ) = @_;
234 my $localdt = $dt->clone();
235 my $day = $localdt->day;
236 my $month = $localdt->month;
238 #Change timezone to "floating" before doing any calculations or comparisons
239 $localdt->set_time_zone("floating");
240 $localdt->truncate( to => 'day' );
243 if ( $self->exception_holidays->contains($localdt) ) {
244 # exceptions are not holidays
245 return 0;
248 my $dow = $localdt->day_of_week;
249 # Representation fix
250 # DateTime object dow (1-7) where Monday is 1
251 # Arrays are 0-based where 0 = Sunday, not 7.
252 if ( $dow == 7 ) {
253 $dow = 0;
256 if ( $self->{weekly_closed_days}->[$dow] == 1 ) {
257 return 1;
260 if ( exists $self->{day_month_closed_days}->{$month}->{$day} ) {
261 return 1;
264 my $ymd = $localdt->ymd('') ;
265 if ($self->single_holidays( $ymd ) == 1 ) {
266 return 1;
269 # damn have to go to work after all
270 return 0;
273 sub next_open_day {
274 my ( $self, $dt ) = @_;
275 my $base_date = $dt->clone();
277 $base_date->add(days => 1);
279 while ($self->is_holiday($base_date)) {
280 $base_date->add(days => 1);
283 return $base_date;
286 sub prev_open_day {
287 my ( $self, $dt ) = @_;
288 my $base_date = $dt->clone();
290 $base_date->add(days => -1);
292 while ($self->is_holiday($base_date)) {
293 $base_date->add(days => -1);
296 return $base_date;
299 sub days_between {
300 my $self = shift;
301 my $start_dt = shift;
302 my $end_dt = shift;
304 if ( $start_dt->compare($end_dt) > 0 ) {
305 # swap dates
306 my $int_dt = $end_dt;
307 $end_dt = $start_dt;
308 $start_dt = $int_dt;
312 # start and end should not be closed days
313 my $days = $start_dt->delta_days($end_dt)->delta_days;
314 for (my $dt = $start_dt->clone();
315 $dt <= $end_dt;
316 $dt->add(days => 1)
318 if ($self->is_holiday($dt)) {
319 $days--;
322 return DateTime::Duration->new( days => $days );
326 sub hours_between {
327 my ($self, $start_date, $end_date) = @_;
328 my $start_dt = $start_date->clone();
329 my $end_dt = $end_date->clone();
330 my $duration = $end_dt->delta_ms($start_dt);
331 $start_dt->truncate( to => 'day' );
332 $end_dt->truncate( to => 'day' );
333 # NB this is a kludge in that it assumes all days are 24 hours
334 # However for hourly loans the logic should be expanded to
335 # take into account open/close times then it would be a duration
336 # of library open hours
337 my $skipped_days = 0;
338 for (my $dt = $start_dt->clone();
339 $dt <= $end_dt;
340 $dt->add(days => 1)
342 if ($self->is_holiday($dt)) {
343 ++$skipped_days;
346 if ($skipped_days) {
347 $duration->subtract_duration(DateTime::Duration->new( hours => 24 * $skipped_days));
350 return $duration;
354 sub set_daysmode {
355 my ( $self, $mode ) = @_;
357 # if not testing this is a no op
358 if ( $self->{test} ) {
359 $self->{days_mode} = $mode;
362 return;
365 sub clear_weekly_closed_days {
366 my $self = shift;
367 $self->{weekly_closed_days} = [ 0, 0, 0, 0, 0, 0, 0 ]; # Sunday only
368 return;
372 __END__
374 =head1 NAME
376 Koha::Calendar - Object containing a branches calendar
378 =head1 SYNOPSIS
380 use Koha::Calendar
382 my $c = Koha::Calendar->new( branchcode => 'MAIN' );
383 my $dt = DateTime->now();
385 # are we open
386 $open = $c->is_holiday($dt);
387 # when will item be due if loan period = $dur (a DateTime::Duration object)
388 $duedate = $c->addDate($dt,$dur,'days');
391 =head1 DESCRIPTION
393 Implements those features of C4::Calendar needed for Staffs Rolling Loans
395 =head1 METHODS
397 =head2 new : Create a calendar object
399 my $calendar = Koha::Calendar->new( branchcode => 'MAIN' );
401 The option branchcode is required
404 =head2 addDate
406 my $dt = $calendar->addDate($date, $dur, $unit)
408 C<$date> is a DateTime object representing the starting date of the interval.
410 C<$offset> is a DateTime::Duration to add to it
412 C<$unit> is a string value 'days' or 'hours' toflag granularity of duration
414 Currently unit is only used to invoke Staffs return Monday at 10 am rule this
415 parameter will be removed when issuingrules properly cope with that
418 =head2 addHours
420 my $dt = $calendar->addHours($date, $dur, $return_by_hour )
422 C<$date> is a DateTime object representing the starting date of the interval.
424 C<$offset> is a DateTime::Duration to add to it
426 C<$return_by_hour> is an integer value representing the opening hour for the branch
429 =head2 addDays
431 my $dt = $calendar->addDays($date, $dur)
433 C<$date> is a DateTime object representing the starting date of the interval.
435 C<$offset> is a DateTime::Duration to add to it
437 C<$unit> is a string value 'days' or 'hours' toflag granularity of duration
439 Currently unit is only used to invoke Staffs return Monday at 10 am rule this
440 parameter will be removed when issuingrules properly cope with that
443 =head2 single_holidays
445 my $rc = $self->single_holidays( $ymd );
447 Passed a $date in Ymd (yyyymmdd) format - returns 1 if date is a single_holiday, or 0 if not.
450 =head2 is_holiday
452 $yesno = $calendar->is_holiday($dt);
454 passed a DateTime object returns 1 if it is a closed day
455 0 if not according to the calendar
457 =head2 days_between
459 $duration = $calendar->days_between($start_dt, $end_dt);
461 Passed two dates returns a DateTime::Duration object measuring the length between them
462 ignoring closed days. Always returns a positive number irrespective of the
463 relative order of the parameters
465 =head2 next_open_day
467 $datetime = $calendar->next_open_day($duedate_dt)
469 Passed a Datetime returns another Datetime representing the next open day. It is
470 intended for use to calculate the due date when useDaysMode syspref is set to either
471 'Datedue' or 'Calendar'.
473 =head2 prev_open_day
475 $datetime = $calendar->prev_open_day($duedate_dt)
477 Passed a Datetime returns another Datetime representing the previous open day. It is
478 intended for use to calculate the due date when useDaysMode syspref is set to either
479 'Datedue' or 'Calendar'.
481 =head2 set_daysmode
483 For testing only allows the calling script to change days mode
485 =head2 clear_weekly_closed_days
487 In test mode changes the testing set of closed days to a new set with
488 no closed days. TODO passing an array of closed days to this would
489 allow testing of more configurations
491 =head2 add_holiday
493 Passed a datetime object this will add it to the calendar's list of
494 closed days. This is for testing so that we can alter the Calenfar object's
495 list of specified dates
497 =head1 DIAGNOSTICS
499 Will croak if not passed a branchcode in new
501 =head1 BUGS AND LIMITATIONS
503 This only contains a limited subset of the functionality in C4::Calendar
504 Only enough to support Staffs Rolling loans
506 =head1 AUTHOR
508 Colin Campbell colin.campbell@ptfs-europe.com
510 =head1 LICENSE AND COPYRIGHT
512 Copyright (c) 2011 PTFS-Europe Ltd All rights reserved
514 This program is free software: you can redistribute it and/or modify
515 it under the terms of the GNU General Public License as published by
516 the Free Software Foundation, either version 2 of the License, or
517 (at your option) any later version.
519 This program is distributed in the hope that it will be useful,
520 but WITHOUT ANY WARRANTY; without even the implied warranty of
521 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
522 GNU General Public License for more details.
524 You should have received a copy of the GNU General Public License
525 along with this program. If not, see <http://www.gnu.org/licenses/>.