FS/FS/cust_tax_location.pm

   1 package FS::cust_tax_location;
   2
   3 use strict;
   4 use vars qw( @ISA );
   5 use FS::Record qw( qsearch qsearchs dbh );
   6 use FS::Misc qw ( csv_from_fixed );
   7
   8 @ISA = qw(FS::Record);
   9
  10 =head1 NAME
  11
  12 FS::cust_tax_location - Object methods for cust_tax_location records
  13
  14 =head1 SYNOPSIS
  15
  16   use FS::cust_tax_location;
  17
  18   $record = new FS::cust_tax_location \%hash;
  19   $record = new FS::cust_tax_location { 'column' => 'value' };
  20
  21   $error = $record->insert;
  22
  23   $error = $new_record->replace($old_record);
  24
  25   $error = $record->delete;
  26
  27   $error = $record->check;
  28
  29 =head1 DESCRIPTION
  30
  31 An FS::cust_tax_location object represents a mapping between a customer and
  32 a tax location.  FS::cust_tax_location inherits from FS::Record.  The
  33 following fields are currently supported:
  34
  35 =over 4
  36
  37 =item custlocationnum
  38
  39 primary key
  40
  41 =item data_vendor
  42
  43 a tax data vendor
  44
  45 =item zip
  46
  47 =item state
  48
  49 =item plus4hi
  50
  51 the upper bound of the last 4 zip code digits
  52
  53 =item plus4lo
  54
  55 the lower bound of the last 4 zip code digits
  56
  57 =item default_location
  58
  59 'Y' when this record represents the default for zip
  60
  61 =item geocode - the foreign key into FS::part_pkg_tax_rate and FS::tax_rate
  62
  63
  64 =back
  65
  66 =head1 METHODS
  67
  68 =over 4
  69
  70 =item new HASHREF
  71
  72 Creates a new cust_tax_location.  To add the cust_tax_location to the database,
  73 see L<"insert">.
  74
  75 Note that this stores the hash reference, not a distinct copy of the hash it
  76 points to.  You can ask the object for a copy with the I<hash> method.
  77
  78 =cut
  79
  80 sub table { 'cust_tax_location'; }
  81
  82 =item insert
  83
  84 Adds this record to the database.  If there is an error, returns the error,
  85 otherwise returns false.
  86
  87 =cut
  88
  89 =item delete
  90
  91 Delete this record from the database.
  92
  93 =cut
  94
  95 =item replace OLD_RECORD
  96
  97 Replaces the OLD_RECORD with this one in the database.  If there is an error,
  98 returns the error, otherwise returns false.
  99
 100 =cut
 101
 102 =item check
 103
 104 Checks all fields to make sure this is a valid cust_tax_location.  If there is
 105 an error, returns the error, otherwise returns false.  Called by the insert
 106 and replace methods.
 107
 108 =cut
 109
 110 sub check {
 111   my $self = shift;
 112
 113   my $error =
 114     $self->ut_numbern('custlocationnum')
 115     || $self->ut_text('data_vendor')
 116     || $self->ut_textn('city')
 117     || $self->ut_textn('postalcity')
 118     || $self->ut_textn('county')
 119     || $self->ut_text('state')
 120     || $self->ut_numbern('plus4hi')
 121     || $self->ut_numbern('plus4lo')
 122     || $self->ut_enum('default_location', [ '', 'Y' ] )
 123     || $self->ut_enum('cityflag', [ '', 'I', 'O', 'B' ] )
 124     || $self->ut_alpha('geocode')
 125   ;
 126   return $error if $error;
 127
 128   #ugh!  cch canada weirdness and more
 129   if ($self->state eq 'CN' && $self->data_vendor eq 'cch-zip' ) {
 130     $error = "Illegal cch canadian zip"
 131      unless $self->zip =~ /^[A-Z]$/;
 132   } elsif ($self->state =~ /^E([B-DFGILNPR-UW])$/ && $self->data_vendor eq 'cch-zip' ) {
 133     $error = "Illegal cch european zip"
 134      unless $self->zip =~ /^E$1$/;
 135   } else {
 136     $error = $self->ut_number('zip', $self->state eq 'CN' ? 'CA' : 'US');
 137   }
 138   return $error if $error;
 139
 140   #ugh!  cch canada weirdness and more
 141   return "must specify either city/county or plus4lo/plus4hi"
 142     unless ( $self->plus4lo && $self->plus4hi ||
 143              ( $self->city ||
 144                $self->state eq 'CN' ||
 145                $self->state =~ /^E([B-DFGILNPR-UW])$/
 146              ) && $self->county
 147            );
 148
 149   $self->SUPER::check;
 150 }
 151
 152
 153 sub batch_import {
 154   my ($param, $job) = @_;
 155
 156   my $fh = $param->{filehandle};
 157   my $format = $param->{'format'};
 158
 159   my $imported = 0;
 160   my @fields;
 161   my $hook;
 162
 163   my @column_lengths = ();
 164   my @column_callbacks = ();
 165   if ( $format =~ /^cch-fixed/ ) {
 166     $format =~ s/-fixed//;
 167     my $f = $format;
 168     my $update = 0;
 169     $f =~ s/-update// && ($update = 1);
 170     if ($f eq 'cch') {
 171       push @column_lengths, qw( 5 2 4 4 10 1 );
 172     } elsif ( $f eq 'cch-zip' ) {
 173       push @column_lengths, qw( 5 28 25 2 28 5 1 1 10 1 2 );
 174     } else {
 175       return "Unknown format: $format";
 176     }
 177     push @column_lengths, 1 if $update;
 178   }
 179
 180   my $line;
 181   my ( $count, $last, $min_sec ) = (0, time, 5); #progressbar
 182   if ( $job || scalar(@column_lengths) ) {
 183     my $error = csv_from_fixed(\$fh, \$count, \@column_lengths);
 184     return $error if $error;
 185   }
 186
 187   if ( $format eq 'cch' || $format eq 'cch-update' ) {
 188     @fields = qw( zip state plus4lo plus4hi geocode default_location );
 189     push @fields, 'actionflag' if $format eq 'cch-update';
 190
 191     $imported++ if $format eq 'cch-update'; #empty file ok
 192
 193     $hook = sub {
 194       my $hash = shift;
 195
 196       $hash->{'data_vendor'} = 'cch';
 197       $hash->{'default_location'} =~ s/ //g;
 198
 199       if (exists($hash->{actionflag}) && $hash->{actionflag} eq 'D') {
 200         delete($hash->{actionflag});
 201
 202         my $cust_tax_location = qsearchs('cust_tax_location', $hash);
 203         return "Can't find cust_tax_location to delete: ".
 204                join(" ", map { "$_ => ". $hash->{$_} } @fields)
 205           unless $cust_tax_location;
 206
 207         my $error = $cust_tax_location->delete;
 208         return $error if $error;
 209
 210         delete($hash->{$_}) foreach (keys %$hash);
 211       }
 212
 213       delete($hash->{'actionflag'});
 214
 215       '';
 216
 217     };
 218
 219   } elsif ( $format eq 'cch-zip' || $format eq 'cch-update-zip' ) {
 220     @fields = qw( zip city county state postalcity countyfips countydef default_location geocode cityflag unique );
 221     push @fields, 'actionflag' if $format eq 'cch-update-zip';
 222
 223     $imported++ if $format eq 'cch-update'; #empty file ok
 224
 225     $hook = sub {
 226       my $hash = shift;
 227
 228       $hash->{'data_vendor'} = 'cch-zip';
 229       delete($hash->{$_}) foreach qw( countyfips countydef unique );
 230
 231       $hash->{'cityflag'} =~ s/ //g;
 232       $hash->{'default_location'} =~ s/ //g;
 233
 234       if (exists($hash->{actionflag}) && $hash->{actionflag} eq 'D') {
 235         delete($hash->{actionflag});
 236
 237         my $cust_tax_location = qsearchs('cust_tax_location', $hash);
 238         return "Can't find cust_tax_location to delete: ".
 239                join(" ", map { "$_ => ". $hash->{$_} } @fields)
 240           unless $cust_tax_location;
 241
 242         my $error = $cust_tax_location->delete;
 243         return $error if $error;
 244
 245         delete($hash->{$_}) foreach (keys %$hash);
 246       }
 247
 248       delete($hash->{'actionflag'});
 249
 250       '';
 251
 252     };
 253
 254   } elsif ( $format eq 'extended' ) {
 255     die "unimplemented\n";
 256     @fields = qw( );
 257   } else {
 258     die "unknown format $format";
 259   }
 260
 261   eval "use Text::CSV_XS;";
 262   die $@ if $@;
 263
 264   my $csv = new Text::CSV_XS;
 265
 266   local $SIG{HUP} = 'IGNORE';
 267   local $SIG{INT} = 'IGNORE';
 268   local $SIG{QUIT} = 'IGNORE';
 269   local $SIG{TERM} = 'IGNORE';
 270   local $SIG{TSTP} = 'IGNORE';
 271   local $SIG{PIPE} = 'IGNORE';
 272
 273   my $oldAutoCommit = $FS::UID::AutoCommit;
 274   local $FS::UID::AutoCommit = 0;
 275   my $dbh = dbh;
 276
 277   while ( defined($line=<$fh>) ) {
 278     $csv->parse($line) or do {
 279       $dbh->rollback if $oldAutoCommit;
 280       return "can't parse: ". $csv->error_input();
 281     };
 282
 283     if ( $job ) {  # progress bar
 284       if ( time - $min_sec > $last ) {
 285         my $error = $job->update_statustext(
 286           int( 100 * $imported / $count ). ",Importing locations"
 287         );
 288         die $error if $error;
 289         $last = time;
 290       }
 291     }
 292
 293     my @columns = $csv->fields();
 294
 295     my %cust_tax_location = ( 'data_vendor' => $format );;
 296     foreach my $field ( @fields ) {
 297       $cust_tax_location{$field} = shift @columns;
 298     }
 299     if ( scalar( @columns ) ) {
 300       $dbh->rollback if $oldAutoCommit;
 301       return "Unexpected trailing columns in line (wrong format?): $line";
 302     }
 303
 304     my $error = &{$hook}(\%cust_tax_location);
 305     if ( $error ) {
 306       $dbh->rollback if $oldAutoCommit;
 307       return $error;
 308     }
 309
 310     next unless scalar(keys %cust_tax_location);
 311
 312     my $cust_tax_location = new FS::cust_tax_location( \%cust_tax_location );
 313     $error = $cust_tax_location->insert;
 314
 315     if ( $error ) {
 316       $dbh->rollback if $oldAutoCommit;
 317       return "can't insert cust_tax_location for $line: $error";
 318     }
 319
 320     $imported++;
 321   }
 322
 323   $dbh->commit or die $dbh->errstr if $oldAutoCommit;
 324
 325   return "Empty file!" unless ( $imported || $format =~ /^cch-update/ );
 326
 327   ''; #no error
 328
 329 }
 330
 331 =back
 332
 333 =head1 BUGS
 334
 335 The author should be informed of any you find.
 336
 337 =head1 SEE ALSO
 338
 339 L<FS::Record>, schema.html from the base documentation.
 340
 341 =cut
 342
 343 1;
 344