FS/FS/cust_pkg/Import.pm

   1 package FS::cust_pkg::Import;
   2
   3 use strict;
   4 use vars qw( $DEBUG ); #$conf );
   5 use Data::Dumper;
   6 use FS::Misc::DateTime qw( parse_datetime );
   7 use FS::Record qw( qsearchs );
   8 use FS::cust_pkg;
   9 use FS::cust_main;
  10 use FS::svc_acct;
  11 use FS::svc_external;
  12 use FS::svc_phone;
  13 use FS::svc_domain;
  14
  15 $DEBUG = 0;
  16
  17 #install_callback FS::UID sub {
  18 #  $conf = new FS::Conf;
  19 #};
  20
  21 =head1 NAME
  22
  23 FS::cust_pkg::Import - Batch customer importing
  24
  25 =head1 SYNOPSIS
  26
  27   use FS::cust_pkg::Import;
  28
  29   #import
  30   FS::cust_pkg::Import::batch_import( {
  31     file      => $file,      #filename
  32     type      => $type,      #csv or xls
  33     format    => $format,    #extended, extended-plus_company, svc_external,
  34                              # or svc_external_svc_phone
  35     agentnum  => $agentnum,
  36     job       => $job,       #optional job queue job, for progressbar updates
  37     pkgbatch  => $pkgbatch, #optional batch unique identifier
  38   } );
  39   die $error if $error;
  40
  41   #ajax helper
  42   use FS::UI::Web::JSRPC;
  43   my $server =
  44     new FS::UI::Web::JSRPC 'FS::cust_pkg::Import::process_batch_import', $cgi;
  45   print $server->process;
  46
  47 =head1 DESCRIPTION
  48
  49 Batch package importing.
  50
  51 =head1 SUBROUTINES
  52
  53 =item process_batch_import
  54
  55 Load a batch import as a queued JSRPC job
  56
  57 =cut
  58
  59 sub process_batch_import {
  60   my $job = shift;
  61   my $param = shift;
  62   warn Dumper($param) if $DEBUG;
  63
  64   my $files = $param->{'uploaded_files'}
  65     or die "No files provided.\n";
  66
  67   my (%files) = map { /^(\w+):([\.\w]+)$/ ? ($1,$2):() } split /,/, $files;
  68
  69   my $dir = '%%%FREESIDE_CACHE%%%/cache.'. $FS::UID::datasrc. '/';
  70   my $file = $dir. $files{'file'};
  71
  72   my $type;
  73   if ( $file =~ /\.(\w+)$/i ) {
  74     $type = lc($1);
  75   } else {
  76     #or error out???
  77     warn "can't parse file type from filename $file; defaulting to CSV";
  78     $type = 'csv';
  79   }
  80
  81   my $error =
  82     FS::cust_pkg::Import::batch_import( {
  83       job      => $job,
  84       file     => $file,
  85       type     => $type,
  86       'params' => { pkgbatch => $param->{pkgbatch} },
  87       agentnum => $param->{'agentnum'},
  88       'format' => $param->{'format'},
  89     } );
  90
  91   unlink $file;
  92
  93   die "$error\n" if $error;
  94
  95 }
  96
  97 =item batch_import
  98
  99 =cut
 100
 101 my %formatfields = (
 102   'default'      => [],
 103   'all_dates'    => [],
 104   'svc_acct'     => [qw( username _password domsvc )],
 105   'svc_phone'    => [qw( countrycode phonenum sip_password pin )],
 106   'svc_external' => [qw( id title )],
 107   'location'     => [qw( address1 address2 city state zip country )],
 108 );
 109
 110 sub _formatfields {
 111   \%formatfields;
 112 }
 113
 114 my %import_options = (
 115   'table'         => 'cust_pkg',
 116
 117   'preinsert_callback'  => sub {
 118     my($record, $param) = @_;
 119     my @location_params = grep /^location\./, keys %$param;
 120     if (@location_params) {
 121       my $cust_location = FS::cust_location->new({
 122           'custnum' => $record->custnum,
 123       });
 124       foreach my $p (@location_params) {
 125         $p =~ /^location.(\w+)$/;
 126         $cust_location->set($1, $param->{$p});
 127       }
 128
 129       my $error = $cust_location->find_or_insert; # this avoids duplicates
 130       return "error creating location: $error" if $error;
 131       $record->set('locationnum', $cust_location->locationnum);
 132     }
 133     '';
 134   },
 135
 136   'postinsert_callback' => sub {
 137     my( $record, $param ) = @_;
 138
 139     my $formatfields = _formatfields;
 140     foreach my $svc_x ( grep /^svc/, keys %$formatfields ) {
 141
 142       my $ff = $formatfields->{$svc_x};
 143
 144       if ( grep $param->{"$svc_x.$_"}, @$ff ) {
 145         my $svc = "FS::$svc_x"->new( {
 146           'pkgnum'  => $record->pkgnum,
 147           'svcpart' => $record->part_pkg->svcpart($svc_x),
 148           map { $_ => $param->{"$svc_x.$_"} } @$ff
 149         } );
 150
 151         #this whole thing should be turned into a callback or config to turn on
 152         if ( $svc_x eq 'svc_acct' && $svc->username =~ /\@/ ) {
 153           my($username, $domain) = split(/\@/, $svc->username);
 154           my $svc_domain = qsearchs('svc_domain', { 'domain' => $domain } )
 155                          || new FS::svc_domain { 'svcpart' => 1,
 156                                                  'domain'  => $domain, };
 157           unless ( $svc_domain->svcnum ) {
 158             my $error = $svc_domain->insert;
 159             return "error auto-inserting domain: $error" if $error;
 160           }
 161           $svc->username($username);
 162           $svc->domsvc($svc_domain->svcnum);
 163         }
 164
 165         my $error = $svc->insert;
 166         return "error inserting service: $error" if $error;
 167       }
 168
 169     }
 170
 171     return ''; #no error
 172
 173   },
 174 );
 175
 176 sub _import_options {
 177   \%import_options;
 178 }
 179
 180 sub batch_import {
 181   my $opt = shift;
 182
 183   my $iopt = _import_options;
 184   $opt->{$_} = $iopt->{$_} foreach keys %$iopt;
 185
 186   my $agentnum  = delete $opt->{agentnum}; # i like closures (delete though?)
 187
 188   my $format = delete $opt->{'format'};
 189   my @fields = ();
 190
 191   if ( $format =~ /^(.*)-agent_custid(-agent_pkgid)?$/ ) {
 192     $format = $1;
 193     my $agent_pkgid = $2;
 194     @fields = (
 195       sub {
 196         my( $self, $value ) = @_; # $conf, $param
 197         my $cust_main = qsearchs('cust_main', {
 198           'agentnum'     => $agentnum,
 199           'agent_custid' => $value,
 200         });
 201         $self->custnum($cust_main->custnum) if $cust_main;
 202       },
 203     );
 204     push @fields, 'agent_pkgid' if $agent_pkgid;
 205   } else {
 206     @fields = ( 'custnum' );
 207   }
 208
 209   if ( $format =~ /^(.*)-locationnum$/ ) {
 210     $format = $1;
 211     push @fields, 'locationnum';
 212   }
 213
 214   push @fields, ( 'pkgpart', 'discountnum' );
 215
 216   my @date_fields = ();
 217   if ( $format =~ /all_dates/ ) {
 218     @date_fields = qw(
 219       order_date
 220       start_date setup bill last_bill susp adjourn
 221       resume
 222       cancel expire
 223       contract_end dundate
 224     );
 225   } else {
 226     @date_fields = qw(
 227       start_date setup bill last_bill susp adjourn
 228       cancel expire
 229     );
 230   }
 231
 232   foreach my $field (@date_fields) {
 233     push @fields, sub {
 234       my( $self, $value ) = @_; # $conf, $param
 235       #->$field has undesirable effects
 236       $self->set($field, parse_datetime($value) ); #$field closure
 237     };
 238   }
 239
 240   my $formatfields = _formatfields();
 241
 242   die "unknown format $format" unless $formatfields->{$format};
 243
 244   foreach my $field ( @{ $formatfields->{$format} } ) {
 245
 246     push @fields, sub {
 247       my( $self, $value, $conf, $param ) = @_;
 248       $param->{"$format.$field"} = $value;
 249     };
 250
 251   }
 252
 253   $opt->{'fields'} = \@fields;
 254
 255   FS::Record::batch_import( $opt );
 256
 257 }
 258
 259 =head1 BUGS
 260
 261 Not enough documentation.
 262
 263 =head1 SEE ALSO
 264
 265 L<FS::cust_main>, L<FS::cust_pkg>,
 266 L<FS::svc_acct>, L<FS::svc_external>, L<FS::svc_phone>
 267
 268 =cut
 269
 270 1;