Bug 17278: Fix test compilation errors
[koha.git] / Koha / Calendar.pm
blob1321621d00e877870707d319450ab5219fee4bc5
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::Cache;
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::Cache->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 => C4::Context->tz()
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::Cache->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 => C4::Context->tz()
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 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 $localdt->truncate( to => 'day' );
241 if ( $self->exception_holidays->contains($localdt) ) {
242 # exceptions are not holidays
243 return 0;
246 my $dow = $localdt->day_of_week;
247 # Representation fix
248 # DateTime object dow (1-7) where Monday is 1
249 # Arrays are 0-based where 0 = Sunday, not 7.
250 if ( $dow == 7 ) {
251 $dow = 0;
254 if ( $self->{weekly_closed_days}->[$dow] == 1 ) {
255 return 1;
258 if ( exists $self->{day_month_closed_days}->{$month}->{$day} ) {
259 return 1;
262 my $ymd = $localdt->ymd('') ;
263 if ($self->single_holidays( $ymd ) == 1 ) {
264 return 1;
267 # damn have to go to work after all
268 return 0;
271 sub next_open_day {
272 my ( $self, $dt ) = @_;
273 my $base_date = $dt->clone();
275 $base_date->add(days => 1);
277 while ($self->is_holiday($base_date)) {
278 $base_date->add(days => 1);
281 return $base_date;
284 sub prev_open_day {
285 my ( $self, $dt ) = @_;
286 my $base_date = $dt->clone();
288 $base_date->add(days => -1);
290 while ($self->is_holiday($base_date)) {
291 $base_date->add(days => -1);
294 return $base_date;
297 sub days_between {
298 my $self = shift;
299 my $start_dt = shift;
300 my $end_dt = shift;
302 if ( $start_dt->compare($end_dt) > 0 ) {
303 # swap dates
304 my $int_dt = $end_dt;
305 $end_dt = $start_dt;
306 $start_dt = $int_dt;
310 # start and end should not be closed days
311 my $days = $start_dt->delta_days($end_dt)->delta_days;
312 for (my $dt = $start_dt->clone();
313 $dt <= $end_dt;
314 $dt->add(days => 1)
316 if ($self->is_holiday($dt)) {
317 $days--;
320 return DateTime::Duration->new( days => $days );
324 sub hours_between {
325 my ($self, $start_date, $end_date) = @_;
326 my $start_dt = $start_date->clone();
327 my $end_dt = $end_date->clone();
328 my $duration = $end_dt->delta_ms($start_dt);
329 $start_dt->truncate( to => 'day' );
330 $end_dt->truncate( to => 'day' );
331 # NB this is a kludge in that it assumes all days are 24 hours
332 # However for hourly loans the logic should be expanded to
333 # take into account open/close times then it would be a duration
334 # of library open hours
335 my $skipped_days = 0;
336 for (my $dt = $start_dt->clone();
337 $dt <= $end_dt;
338 $dt->add(days => 1)
340 if ($self->is_holiday($dt)) {
341 ++$skipped_days;
344 if ($skipped_days) {
345 $duration->subtract_duration(DateTime::Duration->new( hours => 24 * $skipped_days));
348 return $duration;
352 sub set_daysmode {
353 my ( $self, $mode ) = @_;
355 # if not testing this is a no op
356 if ( $self->{test} ) {
357 $self->{days_mode} = $mode;
360 return;
363 sub clear_weekly_closed_days {
364 my $self = shift;
365 $self->{weekly_closed_days} = [ 0, 0, 0, 0, 0, 0, 0 ]; # Sunday only
366 return;
370 __END__
372 =head1 NAME
374 Koha::Calendar - Object containing a branches calendar
376 =head1 SYNOPSIS
378 use Koha::Calendar
380 my $c = Koha::Calendar->new( branchcode => 'MAIN' );
381 my $dt = DateTime->now();
383 # are we open
384 $open = $c->is_holiday($dt);
385 # when will item be due if loan period = $dur (a DateTime::Duration object)
386 $duedate = $c->addDate($dt,$dur,'days');
389 =head1 DESCRIPTION
391 Implements those features of C4::Calendar needed for Staffs Rolling Loans
393 =head1 METHODS
395 =head2 new : Create a calendar object
397 my $calendar = Koha::Calendar->new( branchcode => 'MAIN' );
399 The option branchcode is required
402 =head2 addDate
404 my $dt = $calendar->addDate($date, $dur, $unit)
406 C<$date> is a DateTime object representing the starting date of the interval.
408 C<$offset> is a DateTime::Duration to add to it
410 C<$unit> is a string value 'days' or 'hours' toflag granularity of duration
412 Currently unit is only used to invoke Staffs return Monday at 10 am rule this
413 parameter will be removed when issuingrules properly cope with that
416 =head2 addHours
418 my $dt = $calendar->addHours($date, $dur, $return_by_hour )
420 C<$date> is a DateTime object representing the starting date of the interval.
422 C<$offset> is a DateTime::Duration to add to it
424 C<$return_by_hour> is an integer value representing the opening hour for the branch
427 =head2 addDays
429 my $dt = $calendar->addDays($date, $dur)
431 C<$date> is a DateTime object representing the starting date of the interval.
433 C<$offset> is a DateTime::Duration to add to it
435 C<$unit> is a string value 'days' or 'hours' toflag granularity of duration
437 Currently unit is only used to invoke Staffs return Monday at 10 am rule this
438 parameter will be removed when issuingrules properly cope with that
441 =head2 single_holidays
443 my $rc = $self->single_holidays( $ymd );
445 Passed a $date in Ymd (yyyymmdd) format - returns 1 if date is a single_holiday, or 0 if not.
448 =head2 is_holiday
450 $yesno = $calendar->is_holiday($dt);
452 passed a DateTime object returns 1 if it is a closed day
453 0 if not according to the calendar
455 =head2 days_between
457 $duration = $calendar->days_between($start_dt, $end_dt);
459 Passed two dates returns a DateTime::Duration object measuring the length between them
460 ignoring closed days. Always returns a positive number irrespective of the
461 relative order of the parameters
463 =head2 next_open_day
465 $datetime = $calendar->next_open_day($duedate_dt)
467 Passed a Datetime returns another Datetime representing the next open day. It is
468 intended for use to calculate the due date when useDaysMode syspref is set to either
469 'Datedue' or 'Calendar'.
471 =head2 prev_open_day
473 $datetime = $calendar->prev_open_day($duedate_dt)
475 Passed a Datetime returns another Datetime representing the previous open day. It is
476 intended for use to calculate the due date when useDaysMode syspref is set to either
477 'Datedue' or 'Calendar'.
479 =head2 set_daysmode
481 For testing only allows the calling script to change days mode
483 =head2 clear_weekly_closed_days
485 In test mode changes the testing set of closed days to a new set with
486 no closed days. TODO passing an array of closed days to this would
487 allow testing of more configurations
489 =head2 add_holiday
491 Passed a datetime object this will add it to the calendar's list of
492 closed days. This is for testing so that we can alter the Calenfar object's
493 list of specified dates
495 =head1 DIAGNOSTICS
497 Will croak if not passed a branchcode in new
499 =head1 BUGS AND LIMITATIONS
501 This only contains a limited subset of the functionality in C4::Calendar
502 Only enough to support Staffs Rolling loans
504 =head1 AUTHOR
506 Colin Campbell colin.campbell@ptfs-europe.com
508 =head1 LICENSE AND COPYRIGHT
510 Copyright (c) 2011 PTFS-Europe Ltd All rights reserved
512 This program is free software: you can redistribute it and/or modify
513 it under the terms of the GNU General Public License as published by
514 the Free Software Foundation, either version 2 of the License, or
515 (at your option) any later version.
517 This program is distributed in the hope that it will be useful,
518 but WITHOUT ANY WARRANTY; without even the implied warranty of
519 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
520 GNU General Public License for more details.
522 You should have received a copy of the GNU General Public License
523 along with this program. If not, see <http://www.gnu.org/licenses/>.