1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
|
package FS::Cron::bill;
use strict;
use vars qw( @ISA @EXPORT_OK );
use Exporter;
use Date::Parse;
use DBI 1.33; #The "clone" method was added in DBI 1.33.
use FS::UID qw(dbh);
use FS::Record qw( qsearch qsearchs );
use FS::queue;
use FS::cust_main;
use FS::part_event;
use FS::part_event_condition;
@ISA = qw( Exporter );
@EXPORT_OK = qw ( bill );
sub bill {
my %opt = @_;
my $check_freq = $opt{'check_freq'} || '1d';
my $debug = 0;
$debug = 1 if $opt{'v'};
$debug = $opt{'l'} if $opt{'l'};
$FS::cust_main::DEBUG = $debug;
#$FS::cust_event::DEBUG = $opt{'l'} if $opt{'l'};
my @search = ();
push @search, "( cust_main.archived != 'Y' OR archived IS NULL )"; #disable?
push @search, "cust_main.payby = '". $opt{'p'}. "'"
if $opt{'p'};
push @search, "cust_main.agentnum = ". $opt{'a'}
if $opt{'a'};
if ( @ARGV ) {
push @search, "( ".
join(' OR ', map "cust_main.custnum = $_", @ARGV ).
" )";
}
###
# generate where_pkg/where_event search clause
###
#we're at now now (and later).
my($time)= $opt{'d'} ? str2time($opt{'d'}) : $^T;
$time += $opt{'y'} * 86400 if $opt{'y'};
my $invoice_time = $opt{'n'} ? $^T : $time;
# select * from cust_main where
my $where_pkg = <<"END";
0 < ( select count(*) from cust_pkg
where cust_main.custnum = cust_pkg.custnum
and ( cancel is null or cancel = 0 )
and ( setup is null or setup = 0
or bill is null or bill <= $time
or ( expire is not null and expire <= $^T )
or ( adjourn is not null and adjourn <= $^T )
)
)
END
my $where_event = join(' OR ', map {
my $eventtable = $_;
my $join = FS::part_event_condition->join_conditions_sql( $eventtable );
my $where = FS::part_event_condition->where_conditions_sql( $eventtable,
'time'=>$time,
);
my $are_part_event =
"0 < ( SELECT COUNT(*) FROM part_event $join
WHERE check_freq = '$check_freq'
AND eventtable = '$eventtable'
AND ( disabled = '' OR disabled IS NULL )
AND $where
)
";
if ( $eventtable eq 'cust_main' ) {
$are_part_event;
} else {
"0 < ( SELECT COUNT(*) FROM $eventtable
WHERE cust_main.custnum = $eventtable.custnum
AND $are_part_event
)
";
}
} FS::part_event->eventtables);
push @search, "( $where_pkg OR $where_event )";
###
# get a list of custnums
###
warn "searching for customers:\n". join("\n", @search). "\n"
if $opt{'v'} || $opt{'l'};
my $cursor_dbh = dbh->clone;
$cursor_dbh->do(
"DECLARE cron_bill_cursor CURSOR FOR ".
" SELECT custnum FROM cust_main WHERE ". join(' AND ', @search)
) or die $cursor_dbh->errstr;
while ( 1 ) {
my $sth = $cursor_dbh->prepare('FETCH 100 FROM cron_bill_cursor'); #mysql?
$sth->execute or die $sth->errstr;
my @custnums = map { $_->[0] } @{ $sth->fetchall_arrayref };
last unless scalar(@custnums);
###
# for each custnum, queue or make one customer object and bill
# (one at a time, to reduce memory footprint with large #s of customers)
###
foreach my $custnum ( @custnums ) {
my %args = (
'time' => $time,
'invoice_time' => $invoice_time,
'actual_time' => $^T, #when freeside-bill was started
#(not, when using -m, freeside-queued)
'check_freq' => $check_freq,
'resetup' => ( $opt{'s'} ? $opt{'s'} : 0 ),
);
if ( $opt{'m'} ) {
if ( $opt{'r'} ) {
warn "DRY RUN: would add custnum $custnum for queued_bill\n";
} else {
#avoid queuing another job if there's one still waiting to run
next if qsearch( 'queue', { 'job' => 'FS::cust_main::queued_bill',
'custnum' => $custnum,
'status' => 'new',
}
);
#add job to queue that calls bill_and_collect with options
my $queue = new FS::queue {
'job' => 'FS::cust_main::queued_bill',
'secure' => 'Y',
'priority' => 99, #don't get in the way of provisioning jobs
};
my $error = $queue->insert( 'custnum'=>$custnum, %args );
}
} else {
my $cust_main = qsearchs( 'cust_main', { 'custnum' => $custnum } );
$cust_main->bill_and_collect( %args, 'debug' => $debug );
}
}
}
$cursor_dbh->commit or die $cursor_dbh->errstr;
}
1;
|