2 use base qw( Exporter );
6 %virtual_fields_cache %fk_method_cache
7 $money_char $lat_lower $lon_upper
9 use Carp qw(carp cluck croak confess);
10 use Scalar::Util qw( blessed );
11 use File::Slurp qw( slurp );
12 use File::CounterFile;
14 use DBI qw(:sql_types);
15 use DBIx::DBSchema 0.43; #0.43 for foreign keys
18 use NetAddr::IP; # for validation
19 use FS::UID qw(dbh datasrc driver_name);
21 use FS::Schema qw(dbdef);
23 use FS::Msgcat qw(gettext);
24 #use FS::Conf; #dependency loop bs, in install_callback below instead
26 use FS::part_virtual_field;
30 our @encrypt_payby = qw( CARD DCRD CHEK DCHK );
32 #export dbdef for now... everything else expects to find it here
34 dbh fields hfields qsearch qsearchs dbdef jsearch
35 str2time_sql str2time_sql_closing regexp_sql not_regexp_sql concat_sql
40 our $me = '[FS::Record]';
42 our $nowarn_identical = 0;
43 our $nowarn_classload = 0;
44 our $no_update_diff = 0;
47 our $no_check_foreign = 1; #well, not inefficiently in perl by default anymore
55 our $conf_encryption = '';
56 FS::UID->install_callback( sub {
60 $conf = FS::Conf->new;
61 $conf_encryption = $conf->exists('encryption');
62 $money_char = $conf->config('money_char') || '$';
63 my $nw_coords = $conf->exists('geocode-require_nw_coordinates');
64 $lat_lower = $nw_coords ? 1 : -90;
65 $lon_upper = $nw_coords ? -1 : 180;
67 $File::CounterFile::DEFAULT_DIR = $conf->base_dir . "/counters.". datasrc;
69 if ( driver_name eq 'Pg' ) {
70 eval "use DBD::Pg ':pg_types'";
73 eval "sub PG_BYTEA { die 'guru meditation #9: calling PG_BYTEA when not running Pg?'; }";
76 foreach my $table ( dbdef->tables ) {
77 $fk_method_cache{$table} = fk_methods($table);
84 FS::Record - Database record objects
89 use FS::Record qw(dbh fields qsearch qsearchs);
91 $record = new FS::Record 'table', \%hash;
92 $record = new FS::Record 'table', { 'column' => 'value', ... };
94 $record = qsearchs FS::Record 'table', \%hash;
95 $record = qsearchs FS::Record 'table', { 'column' => 'value', ... };
96 @records = qsearch FS::Record 'table', \%hash;
97 @records = qsearch FS::Record 'table', { 'column' => 'value', ... };
99 $table = $record->table;
100 $dbdef_table = $record->dbdef_table;
102 $value = $record->get('column');
103 $value = $record->getfield('column');
104 $value = $record->column;
106 $record->set( 'column' => 'value' );
107 $record->setfield( 'column' => 'value' );
108 $record->column('value');
110 %hash = $record->hash;
112 $hashref = $record->hashref;
114 $error = $record->insert;
116 $error = $record->delete;
118 $error = $new_record->replace($old_record);
120 # external use deprecated - handled by the database (at least for Pg, mysql)
121 $value = $record->unique('column');
123 $error = $record->ut_float('column');
124 $error = $record->ut_floatn('column');
125 $error = $record->ut_number('column');
126 $error = $record->ut_numbern('column');
127 $error = $record->ut_snumber('column');
128 $error = $record->ut_snumbern('column');
129 $error = $record->ut_money('column');
130 $error = $record->ut_text('column');
131 $error = $record->ut_textn('column');
132 $error = $record->ut_alpha('column');
133 $error = $record->ut_alphan('column');
134 $error = $record->ut_phonen('column');
135 $error = $record->ut_anything('column');
136 $error = $record->ut_name('column');
138 $quoted_value = _quote($value,'table','field');
141 $fields = hfields('table');
142 if ( $fields->{Field} ) { # etc.
144 @fields = fields 'table'; #as a subroutine
145 @fields = $record->fields; #as a method call
150 (Mostly) object-oriented interface to database records. Records are currently
151 implemented on top of DBI. FS::Record is intended as a base class for
152 table-specific classes to inherit from, i.e. FS::cust_main.
158 =item new [ TABLE, ] HASHREF
160 Creates a new record. It doesn't store it in the database, though. See
161 L<"insert"> for that.
163 Note that the object stores this hash reference, not a distinct copy of the
164 hash it points to. You can ask the object for a copy with the I<hash>
167 TABLE can only be omitted when a dervived class overrides the table method.
173 my $class = ref($proto) || $proto;
175 bless ($self, $class);
177 unless ( defined ( $self->table ) ) {
178 $self->{'Table'} = shift;
179 carp "warning: FS::Record::new called with table name ". $self->{'Table'}
180 unless $nowarn_classload;
183 $self->{'Hash'} = shift;
185 foreach my $field ( grep !defined($self->{'Hash'}{$_}), $self->fields ) {
186 $self->{'Hash'}{$field}='';
189 $self->_rebless if $self->can('_rebless');
191 $self->{'modified'} = 0;
193 $self->_cache($self->{'Hash'}, shift) if $self->can('_cache') && @_;
200 my $class = ref($proto) || $proto;
202 bless ($self, $class);
204 $self->{'Table'} = shift unless defined ( $self->table );
206 my $hashref = $self->{'Hash'} = shift;
208 if ( defined( $cache->cache->{$hashref->{$cache->key}} ) ) {
209 my $obj = $cache->cache->{$hashref->{$cache->key}};
210 $obj->_cache($hashref, $cache) if $obj->can('_cache');
213 $cache->cache->{$hashref->{$cache->key}} = $self->new($hashref, $cache);
220 my $class = ref($proto) || $proto;
222 bless ($self, $class);
223 if ( defined $self->table ) {
224 cluck "create constructor is deprecated, use new!";
227 croak "FS::Record::create called (not from a subclass)!";
231 =item qsearch PARAMS_HASHREF | TABLE, HASHREF, SELECT, EXTRA_SQL, CACHE_OBJ, ADDL_FROM
233 Searches the database for all records matching (at least) the key/value pairs
234 in HASHREF. Returns all the records found as `FS::TABLE' objects if that
235 module is loaded (i.e. via `use FS::cust_main;'), otherwise returns FS::Record
238 The preferred usage is to pass a hash reference of named parameters:
240 @records = qsearch( {
241 'table' => 'table_name',
242 'hashref' => { 'field' => 'value'
243 'field' => { 'op' => '<',
248 #these are optional...
250 'extra_sql' => 'AND field = ? AND intfield = ?',
251 'extra_param' => [ 'value', [ 5, 'int' ] ],
252 'order_by' => 'ORDER BY something',
253 #'cache_obj' => '', #optional
254 'addl_from' => 'LEFT JOIN othtable USING ( field )',
259 Much code still uses old-style positional parameters, this is also probably
260 fine in the common case where there are only two parameters:
262 my @records = qsearch( 'table', { 'field' => 'value' } );
264 Also possible is an experimental LISTREF of PARAMS_HASHREFs for a UNION of
265 the individual PARAMS_HASHREF queries
267 ###oops, argh, FS::Record::new only lets us create database fields.
268 #Normal behaviour if SELECT is not specified is `*', as in
269 #C<SELECT * FROM table WHERE ...>. However, there is an experimental new
270 #feature where you can specify SELECT - remember, the objects returned,
271 #although blessed into the appropriate `FS::TABLE' package, will only have the
272 #fields you specify. This might have unwanted results if you then go calling
273 #regular FS::TABLE methods
278 my %TYPE = (); #for debugging
281 my($type, $value) = @_;
283 my $bind_type = { TYPE => SQL_VARCHAR };
285 if ( $type =~ /(big)?(int|serial)/i && $value =~ /^-?\d+(\.\d+)?$/ ) {
287 $bind_type = { TYPE => SQL_INTEGER };
289 } elsif ( $type =~ /^bytea$/i || $type =~ /(blob|varbinary)/i ) {
291 if ( driver_name eq 'Pg' ) {
293 $bind_type = { pg_type => PG_BYTEA };
295 # $bind_type = ? #SQL_VARCHAR could be fine?
298 #DBD::Pg 1.49: Cannot bind ... unknown sql_type 6 with SQL_FLOAT
299 #fixed by DBD::Pg 2.11.8
300 #can change back to SQL_FLOAT in early-mid 2010, once everyone's upgraded
301 #(make a Tron test first)
302 } elsif ( _is_fs_float( $type, $value ) ) {
304 $bind_type = { TYPE => SQL_DECIMAL };
313 my($type, $value) = @_;
314 if ( ( $type =~ /(numeric)/i && $value =~ /^[+-]?\d+(\.\d+)?$/ ) ||
315 ( $type =~ /(real|float4)/i && $value =~ /[-+]?\d*\.?\d+([eE][-+]?\d+)?/)
323 my( @stable, @record, @cache );
324 my( @select, @extra_sql, @extra_param, @order_by, @addl_from );
326 my %union_options = ();
327 if ( ref($_[0]) eq 'ARRAY' ) {
330 foreach my $href ( @$optlist ) {
331 push @stable, ( $href->{'table'} or die "table name is required" );
332 push @record, ( $href->{'hashref'} || {} );
333 push @select, ( $href->{'select'} || '*' );
334 push @extra_sql, ( $href->{'extra_sql'} || '' );
335 push @extra_param, ( $href->{'extra_param'} || [] );
336 push @order_by, ( $href->{'order_by'} || '' );
337 push @cache, ( $href->{'cache_obj'} || '' );
338 push @addl_from, ( $href->{'addl_from'} || '' );
339 push @debug, ( $href->{'debug'} || '' );
341 die "at least one hashref is required" unless scalar(@stable);
342 } elsif ( ref($_[0]) eq 'HASH' ) {
344 $stable[0] = $opt->{'table'} or die "table name is required";
345 $record[0] = $opt->{'hashref'} || {};
346 $select[0] = $opt->{'select'} || '*';
347 $extra_sql[0] = $opt->{'extra_sql'} || '';
348 $extra_param[0] = $opt->{'extra_param'} || [];
349 $order_by[0] = $opt->{'order_by'} || '';
350 $cache[0] = $opt->{'cache_obj'} || '';
351 $addl_from[0] = $opt->{'addl_from'} || '';
352 $debug[0] = $opt->{'debug'} || '';
363 my $cache = $cache[0];
369 foreach my $stable ( @stable ) {
370 #stop altering the caller's hashref
371 my $record = { %{ shift(@record) || {} } };#and be liberal in receipt
372 my $select = shift @select;
373 my $extra_sql = shift @extra_sql;
374 my $extra_param = shift @extra_param;
375 my $order_by = shift @order_by;
376 my $cache = shift @cache;
377 my $addl_from = shift @addl_from;
378 my $debug = shift @debug;
380 #$stable =~ /^([\w\_]+)$/ or die "Illegal table: $table";
382 $stable =~ /^([\w\s\(\)\.\,\=]+)$/ or die "Illegal table: $stable";
385 my $table = $cache ? $cache->table : $stable;
386 my $dbdef_table = dbdef->table($table)
387 or die "No schema for table $table found - ".
388 "do you need to run freeside-upgrade?";
389 my $pkey = $dbdef_table->primary_key;
391 my @real_fields = grep exists($record->{$_}), real_fields($table);
393 my $statement .= "SELECT $select FROM $stable";
394 $statement .= " $addl_from" if $addl_from;
395 if ( @real_fields ) {
396 $statement .= ' WHERE '. join(' AND ',
397 get_real_fields($table, $record, \@real_fields));
400 $statement .= " $extra_sql" if defined($extra_sql);
401 $statement .= " $order_by" if defined($order_by);
403 push @statement, $statement;
405 warn "[debug]$me $statement\n" if $DEBUG > 1 || $debug;
409 grep defined( $record->{$_} ) && $record->{$_} ne '', @real_fields
412 my $value = $record->{$field};
413 my $op = (ref($value) && $value->{op}) ? $value->{op} : '=';
414 $value = $value->{'value'} if ref($value);
415 my $type = dbdef->table($table)->column($field)->type;
417 my $bind_type = _bind_type($type, $value);
421 # %TYPE = map { &{"DBI::$_"}() => $_ } @{ $DBI::EXPORT_TAGS{sql_types} }
423 # warn " bind_param $bind (for field $field), $value, TYPE $TYPE{$TYPE}\n";
427 push @bind_type, $bind_type;
431 foreach my $param ( @$extra_param ) {
432 my $bind_type = { TYPE => SQL_VARCHAR };
435 $value = $param->[0];
436 my $type = $param->[1];
437 $bind_type = _bind_type($type, $value);
440 push @bind_type, $bind_type;
444 my $statement = join( ' ) UNION ( ', @statement );
445 $statement = "( $statement )" if scalar(@statement) > 1;
446 $statement .= " $union_options{order_by}" if $union_options{order_by};
448 my $sth = $dbh->prepare($statement)
449 or croak "$dbh->errstr doing $statement";
452 foreach my $value ( @value ) {
453 my $bind_type = shift @bind_type;
454 $sth->bind_param($bind++, $value, $bind_type );
457 # $sth->execute( map $record->{$_},
458 # grep defined( $record->{$_} ) && $record->{$_} ne '', @fields
459 # ) or croak "Error executing \"$statement\": ". $sth->errstr;
461 my $ok = $sth->execute;
463 my $error = "Error executing \"$statement\"";
464 $error .= ' (' . join(', ', map {"'$_'"} @value) . ')' if @value;
465 $error .= ': '. $sth->errstr;
469 my $table = $stable[0];
471 $table = '' if grep { $_ ne $table } @stable;
472 $pkey = dbdef->table($table)->primary_key if $table;
475 tie %result, "Tie::IxHash";
476 my @stuff = @{ $sth->fetchall_arrayref( {} ) };
477 if ( $pkey && scalar(@stuff) && $stuff[0]->{$pkey} ) {
478 %result = map { $_->{$pkey}, $_ } @stuff;
480 @result{@stuff} = @stuff;
486 if ( eval 'scalar(@FS::'. $table. '::ISA);' ) {
487 if ( eval 'FS::'. $table. '->can(\'new\')' eq \&new ) {
488 #derivied class didn't override new method, so this optimization is safe
491 new_or_cached( "FS::$table", { %{$_} }, $cache )
495 new( "FS::$table", { %{$_} } )
499 #okay, its been tested
500 # warn "untested code (class FS::$table uses custom new method)";
502 eval 'FS::'. $table. '->new( { %{$_} } )';
506 # Check for encrypted fields and decrypt them.
507 ## only in the local copy, not the cached object
508 if ( $conf_encryption
509 && eval '@FS::'. $table . '::encrypted_fields' ) {
510 foreach my $record (@return) {
511 foreach my $field (eval '@FS::'. $table . '::encrypted_fields') {
512 next if $field eq 'payinfo'
513 && ($record->isa('FS::payinfo_transaction_Mixin')
514 || $record->isa('FS::payinfo_Mixin') )
516 && !grep { $record->payby eq $_ } @encrypt_payby;
517 # Set it directly... This may cause a problem in the future...
518 $record->setfield($field, $record->decrypt($record->getfield($field)));
523 cluck "warning: FS::$table not loaded; returning FS::Record objects"
524 unless $nowarn_classload;
526 FS::Record->new( $table, { %{$_} } );
534 Construct the SQL statement and parameter-binding list for qsearch. Takes
535 the qsearch parameters.
537 Returns a hash containing:
538 'table': The primary table name (if there is one).
539 'statement': The SQL statement itself.
540 'bind_type': An arrayref of bind types.
541 'value': An arrayref of parameter values.
542 'cache': The cache object, if one was passed.
547 my( @stable, @record, @cache );
548 my( @select, @extra_sql, @extra_param, @order_by, @addl_from );
551 my %union_options = ();
552 if ( ref($_[0]) eq 'ARRAY' ) {
555 foreach my $href ( @$optlist ) {
556 push @stable, ( $href->{'table'} or die "table name is required" );
557 push @record, ( $href->{'hashref'} || {} );
558 push @select, ( $href->{'select'} || '*' );
559 push @extra_sql, ( $href->{'extra_sql'} || '' );
560 push @extra_param, ( $href->{'extra_param'} || [] );
561 push @order_by, ( $href->{'order_by'} || '' );
562 push @cache, ( $href->{'cache_obj'} || '' );
563 push @addl_from, ( $href->{'addl_from'} || '' );
564 push @debug, ( $href->{'debug'} || '' );
566 die "at least one hashref is required" unless scalar(@stable);
567 } elsif ( ref($_[0]) eq 'HASH' ) {
569 $stable[0] = $opt->{'table'} or die "table name is required";
570 $record[0] = $opt->{'hashref'} || {};
571 $select[0] = $opt->{'select'} || '*';
572 $extra_sql[0] = $opt->{'extra_sql'} || '';
573 $extra_param[0] = $opt->{'extra_param'} || [];
574 $order_by[0] = $opt->{'order_by'} || '';
575 $cache[0] = $opt->{'cache_obj'} || '';
576 $addl_from[0] = $opt->{'addl_from'} || '';
577 $debug[0] = $opt->{'debug'} || '';
588 my $cache = $cache[0];
594 my $result_table = $stable[0];
595 foreach my $stable ( @stable ) {
596 #stop altering the caller's hashref
597 my $record = { %{ shift(@record) || {} } };#and be liberal in receipt
598 my $select = shift @select;
599 my $extra_sql = shift @extra_sql;
600 my $extra_param = shift @extra_param;
601 my $order_by = shift @order_by;
602 my $cache = shift @cache;
603 my $addl_from = shift @addl_from;
604 my $debug = shift @debug;
606 #$stable =~ /^([\w\_]+)$/ or die "Illegal table: $table";
608 $stable =~ /^([\w\s\(\)\.\,\=]+)$/ or die "Illegal table: $stable";
611 $result_table = '' if $result_table ne $stable;
613 my $table = $cache ? $cache->table : $stable;
614 my $dbdef_table = dbdef->table($table)
615 or die "No schema for table $table found - ".
616 "do you need to run freeside-upgrade?";
617 my $pkey = $dbdef_table->primary_key;
619 my @real_fields = grep exists($record->{$_}), real_fields($table);
621 my $statement .= "SELECT $select FROM $stable";
622 $statement .= " $addl_from" if $addl_from;
623 if ( @real_fields ) {
624 $statement .= ' WHERE '. join(' AND ',
625 get_real_fields($table, $record, \@real_fields));
628 $statement .= " $extra_sql" if defined($extra_sql);
629 $statement .= " $order_by" if defined($order_by);
631 push @statement, $statement;
633 warn "[debug]$me $statement\n" if $DEBUG > 1 || $debug;
637 grep defined( $record->{$_} ) && $record->{$_} ne '', @real_fields
640 my $value = $record->{$field};
641 my $op = (ref($value) && $value->{op}) ? $value->{op} : '=';
642 $value = $value->{'value'} if ref($value);
643 my $type = dbdef->table($table)->column($field)->type;
645 my $bind_type = _bind_type($type, $value);
649 # %TYPE = map { &{"DBI::$_"}() => $_ } @{ $DBI::EXPORT_TAGS{sql_types} }
651 # warn " bind_param $bind (for field $field), $value, TYPE $TYPE{$TYPE}\n";
655 push @bind_type, $bind_type;
659 foreach my $param ( @$extra_param ) {
660 my $bind_type = { TYPE => SQL_VARCHAR };
663 $value = $param->[0];
664 my $type = $param->[1];
665 $bind_type = _bind_type($type, $value);
668 push @bind_type, $bind_type;
672 my $statement = join( ' ) UNION ( ', @statement );
673 $statement = "( $statement )" if scalar(@statement) > 1;
674 $statement .= " $union_options{order_by}" if $union_options{order_by};
677 statement => $statement,
678 bind_type => \@bind_type,
680 table => $result_table,
685 # qsearch should eventually use this
687 my ($table, $cache, @hashrefs) = @_;
689 # XXX get rid of these string evals at some point
690 # (when we have time to test it)
691 # my $class = "FS::$table" if $table;
692 # if ( $class and $class->isa('FS::Record') )
693 # if ( $class->can('new') eq \&new )
695 if ( $table && eval 'scalar(@FS::'. $table. '::ISA);' ) {
696 if ( eval 'FS::'. $table. '->can(\'new\')' eq \&new ) {
697 #derivied class didn't override new method, so this optimization is safe
700 new_or_cached( "FS::$table", { %{$_} }, $cache )
704 new( "FS::$table", { %{$_} } )
708 #okay, its been tested
709 # warn "untested code (class FS::$table uses custom new method)";
711 eval 'FS::'. $table. '->new( { %{$_} } )';
715 # Check for encrypted fields and decrypt them.
716 ## only in the local copy, not the cached object
717 if ( $conf_encryption
718 && eval '@FS::'. $table . '::encrypted_fields' ) {
719 foreach my $record (@return) {
720 foreach my $field (eval '@FS::'. $table . '::encrypted_fields') {
721 next if $field eq 'payinfo'
722 && ($record->isa('FS::payinfo_transaction_Mixin')
723 || $record->isa('FS::payinfo_Mixin') )
725 && !grep { $record->payby eq $_ } @encrypt_payby;
726 # Set it directly... This may cause a problem in the future...
727 $record->setfield($field, $record->decrypt($record->getfield($field)));
732 cluck "warning: FS::$table not loaded; returning FS::Record objects"
733 unless $nowarn_classload;
735 FS::Record->new( $table, { %{$_} } );
741 ## makes this easier to read
743 sub get_real_fields {
746 my $real_fields = shift;
748 ## this huge map was previously inline, just broke it out to help read the qsearch method, should be optimized for readability
754 my $type = dbdef->table($table)->column($column)->type;
755 my $value = $record->{$column};
756 $value = $value->{'value'} if ref($value);
757 if ( ref($record->{$_}) ) {
758 $op = $record->{$_}{'op'} if $record->{$_}{'op'};
759 #$op = 'LIKE' if $op =~ /^ILIKE$/i && driver_name ne 'Pg';
760 if ( uc($op) eq 'ILIKE' ) {
762 $record->{$_}{'value'} = lc($record->{$_}{'value'});
763 $column = "LOWER($_)";
765 $record->{$_} = $record->{$_}{'value'}
768 if ( ! defined( $record->{$_} ) || $record->{$_} eq '' ) {
770 if ( driver_name eq 'Pg' ) {
771 if ( $type =~ /(int|numeric|real|float4|(big)?serial)/i ) {
772 qq-( $column IS NULL )-;
774 qq-( $column IS NULL OR $column = '' )-;
777 qq-( $column IS NULL OR $column = "" )-;
779 } elsif ( $op eq '!=' ) {
780 if ( driver_name eq 'Pg' ) {
781 if ( $type =~ /(int|numeric|real|float4|(big)?serial)/i ) {
782 qq-( $column IS NOT NULL )-;
784 qq-( $column IS NOT NULL AND $column != '' )-;
787 qq-( $column IS NOT NULL AND $column != "" )-;
790 if ( driver_name eq 'Pg' ) {
791 qq-( $column $op '' )-;
793 qq-( $column $op "" )-;
796 } elsif ( $op eq '!=' ) {
797 qq-( $column IS NULL OR $column != ? )-;
798 #if this needs to be re-enabled, it needs to use a custom op like
799 #"APPROX=" or something (better name?, not '=', to avoid affecting other
801 #} elsif ( $op eq 'APPROX=' && _is_fs_float( $type, $value ) ) {
802 # ( "$column <= ?", "$column >= ?" );
806 } @{ $real_fields } );
809 =item by_key PRIMARY_KEY_VALUE
811 This is a class method that returns the record with the given primary key
812 value. This method is only useful in FS::Record subclasses. For example:
814 my $cust_main = FS::cust_main->by_key(1); # retrieve customer with custnum 1
818 my $cust_main = qsearchs('cust_main', { 'custnum' => 1 } );
823 my ($class, $pkey_value) = @_;
825 my $table = $class->table
826 or croak "No table for $class found";
828 my $dbdef_table = dbdef->table($table)
829 or die "No schema for table $table found - ".
830 "do you need to create it or run dbdef-create?";
831 my $pkey = $dbdef_table->primary_key
832 or die "No primary key for table $table";
834 return qsearchs($table, { $pkey => $pkey_value });
837 =item jsearch TABLE, HASHREF, SELECT, EXTRA_SQL, PRIMARY_TABLE, PRIMARY_KEY
839 Experimental JOINed search method. Using this method, you can execute a
840 single SELECT spanning multiple tables, and cache the results for subsequent
841 method calls. Interface will almost definately change in an incompatible
849 my($table, $record, $select, $extra_sql, $ptable, $pkey ) = @_;
850 my $cache = FS::SearchCache->new( $ptable, $pkey );
853 grep { !$saw{$_->getfield($pkey)}++ }
854 qsearch($table, $record, $select, $extra_sql, $cache )
858 =item qsearchs PARAMS_HASHREF | TABLE, HASHREF, SELECT, EXTRA_SQL, CACHE_OBJ, ADDL_FROM
860 Same as qsearch, except that if more than one record matches, it B<carp>s but
861 returns the first. If this happens, you either made a logic error in asking
862 for a single item, or your data is corrupted.
866 sub qsearchs { # $result_record = &FS::Record:qsearchs('table',\%hash);
868 my(@result) = qsearch(@_);
869 cluck "warning: Multiple records in scalar search ($table)"
870 if scalar(@result) > 1;
871 #should warn more vehemently if the search was on a primary key?
872 scalar(@result) ? ($result[0]) : ();
883 Returns the table name.
888 # cluck "warning: FS::Record::table deprecated; supply one in subclass!";
895 Returns the DBIx::DBSchema::Table object for the table.
901 my($table)=$self->table;
902 dbdef->table($table);
907 Returns the primary key for the table.
913 my $pkey = $self->dbdef_table->primary_key;
916 =item get, getfield COLUMN
918 Returns the value of the column/field/key COLUMN.
923 my($self,$field) = @_;
924 # to avoid "Use of unitialized value" errors
925 if ( defined ( $self->{Hash}->{$field} ) ) {
926 $self->{Hash}->{$field};
936 =item set, setfield COLUMN, VALUE
938 Sets the value of the column/field/key COLUMN to VALUE. Returns VALUE.
943 my($self,$field,$value) = @_;
944 $self->{'modified'} = 1;
945 $self->{'Hash'}->{$field} = $value;
954 Returns true if the column/field/key COLUMN exists.
959 my($self,$field) = @_;
960 exists($self->{Hash}->{$field});
963 =item AUTLOADED METHODS
965 $record->column is a synonym for $record->get('column');
967 $record->column('value') is a synonym for $record->set('column','value');
969 $record->foreign_table_name calls qsearchs and returns a single
970 FS::foreign_table record (for tables referenced by a column of this table) or
971 qsearch and returns an array of FS::foreign_table records (for tables
972 referenced by a column in the foreign table).
979 my($field)=$AUTOLOAD;
982 #$fk_method_cache{$self->table} ||= fk_methods($self->table);
983 if ( exists($fk_method_cache{$self->table}->{$field}) ) {
985 my $fk_info = $fk_method_cache{$self->table}->{$field};
986 my $method = $fk_info->{method} || 'qsearchs';
987 my $table = $fk_info->{table} || $field;
988 my $column = $fk_info->{column};
989 my $foreign_column = $fk_info->{references} || $column;
991 eval "use FS::$table";
994 my $pkey_value = $self->$column();
995 my %search = ( $foreign_column => $pkey_value );
997 # FS::Record->$method() ? they're actually just subs :/
998 if ( $method eq 'qsearchs' ) {
999 return $pkey_value ? qsearchs( $table, \%search ) : '';
1000 } elsif ( $method eq 'qsearch' ) {
1001 return $pkey_value ? qsearch( $table, \%search ) : ();
1003 die "unknown method $method";
1008 if ( defined($value) ) {
1009 confess "errant AUTOLOAD $field for $self (arg $value)"
1010 unless blessed($self) && $self->can('setfield');
1011 $self->setfield($field,$value);
1013 confess "errant AUTOLOAD $field for $self (no args)"
1014 unless blessed($self) && $self->can('getfield');
1015 $self->getfield($field);
1019 # efficient (also, old, doesn't support FK stuff)
1021 # my $field = $AUTOLOAD;
1022 # $field =~ s/.*://;
1023 # if ( defined($_[1]) ) {
1024 # $_[0]->setfield($field, $_[1]);
1026 # $_[0]->getfield($field);
1035 # foreign keys we reference in other tables
1036 foreach my $fk (dbdef->table($table)->foreign_keys) {
1039 if ( scalar( @{$fk->columns} ) == 1 ) {
1040 if ( ! @{$fk->references} || $fk->columns->[0] eq $fk->references->[0] ){
1041 $method = $fk->table;
1043 #some sort of hint in the table.pm or schema for methods not named
1044 # after their foreign table (well, not a whole lot different than
1045 # just providing a small subroutine...)
1049 $hash{$method} = { #fk_info
1050 'method' => 'qsearchs',
1051 'column' => $fk->columns->[0],
1052 #'references' => $fk->references->[0],
1060 # foreign keys referenced in other tables to us
1061 # (alas. why we're cached. still, might this loop better be done once at
1062 # schema load time insetad of every time we AUTOLOAD a method on a new
1064 foreach my $f_table ( dbdef->tables ) {
1065 foreach my $fk (dbdef->table($f_table)->foreign_keys) {
1067 next unless $fk->table eq $table;
1070 if ( scalar( @{$fk->columns} ) == 1 ) {
1071 if ( ! @{$fk->references} || $fk->columns->[0] eq $fk->references->[0] ){
1074 #some sort of hint in the table.pm or schema for methods not named
1075 # after their foreign table (well, not a whole lot different than
1076 # just providing a small subroutine...)
1080 $hash{$method} = { #fk_info
1081 'method' => 'qsearch',
1082 'column' => $fk->columns->[0], #references||column
1083 #'references' => $fk->column->[0],
1098 Returns a list of the column/value pairs, usually for assigning to a new hash.
1100 To make a distinct duplicate of an FS::Record object, you can do:
1102 $new = new FS::Record ( $old->table, { $old->hash } );
1108 confess $self. ' -> hash: Hash attribute is undefined'
1109 unless defined($self->{'Hash'});
1110 %{ $self->{'Hash'} };
1115 Returns a reference to the column/value hash. This may be deprecated in the
1116 future; if there's a reason you can't just use the autoloaded or get/set
1128 Returns true if any of this object's values have been modified with set (or via
1129 an autoloaded method). Doesn't yet recognize when you retreive a hashref and
1136 $self->{'modified'};
1139 =item select_for_update
1141 Selects this record with the SQL "FOR UPDATE" command. This can be useful as
1146 sub select_for_update {
1148 my $primary_key = $self->primary_key;
1151 'table' => $self->table,
1152 'hashref' => { $primary_key => $self->$primary_key() },
1153 'extra_sql' => 'FOR UPDATE',
1159 Locks this table with a database-driver specific lock method. This is used
1160 as a mutex in order to do a duplicate search.
1162 For PostgreSQL, does "LOCK TABLE tablename IN SHARE ROW EXCLUSIVE MODE".
1164 For MySQL, does a SELECT FOR UPDATE on the duplicate_lock table.
1166 Errors are fatal; no useful return value.
1168 Note: To use this method for new tables other than svc_acct and svc_phone,
1169 edit freeside-upgrade and add those tables to the duplicate_lock list.
1175 my $table = $self->table;
1177 warn "$me locking $table table\n" if $DEBUG;
1179 if ( driver_name =~ /^Pg/i ) {
1181 dbh->do("LOCK TABLE $table IN SHARE ROW EXCLUSIVE MODE")
1184 } elsif ( driver_name =~ /^mysql/i ) {
1186 dbh->do("SELECT * FROM duplicate_lock
1187 WHERE lockname = '$table'
1189 ) or die dbh->errstr;
1193 die "unknown database ". driver_name. "; don't know how to lock table";
1197 warn "$me acquired $table table lock\n" if $DEBUG;
1203 Inserts this record to the database. If there is an error, returns the error,
1204 otherwise returns false.
1212 warn "$self -> insert" if $DEBUG;
1214 my $error = $self->check;
1215 return $error if $error;
1217 #single-field non-null unique keys are given a value if empty
1218 #(like MySQL's AUTO_INCREMENT or Pg SERIAL)
1219 foreach ( $self->dbdef_table->unique_singles) {
1220 next if $self->getfield($_);
1221 next if $self->dbdef_table->column($_)->null eq 'NULL';
1225 #and also the primary key, if the database isn't going to
1226 my $primary_key = $self->dbdef_table->primary_key;
1228 if ( $primary_key ) {
1229 my $col = $self->dbdef_table->column($primary_key);
1232 uc($col->type) =~ /^(BIG)?SERIAL\d?/
1233 || ( driver_name eq 'Pg'
1234 && defined($col->default)
1235 && $col->quoted_default =~ /^nextval\(/i
1237 || ( driver_name eq 'mysql'
1238 && defined($col->local)
1239 && $col->local =~ /AUTO_INCREMENT/i
1241 $self->unique($primary_key) unless $self->getfield($primary_key) || $db_seq;
1244 my $table = $self->table;
1246 # Encrypt before the database
1247 if ( defined(eval '@FS::'. $table . '::encrypted_fields')
1248 && scalar( eval '@FS::'. $table . '::encrypted_fields')
1249 && $conf->exists('encryption')
1251 foreach my $field (eval '@FS::'. $table . '::encrypted_fields') {
1252 next if $field eq 'payinfo'
1253 && ($self->isa('FS::payinfo_transaction_Mixin')
1254 || $self->isa('FS::payinfo_Mixin') )
1256 && !grep { $self->payby eq $_ } @encrypt_payby;
1257 $saved->{$field} = $self->getfield($field);
1258 $self->setfield($field, $self->encrypt($self->getfield($field)));
1262 #false laziness w/delete
1264 grep { defined($self->getfield($_)) && $self->getfield($_) ne "" }
1267 my @values = map { _quote( $self->getfield($_), $table, $_) } @real_fields;
1270 my $statement = "INSERT INTO $table ";
1271 if ( @real_fields ) {
1274 join( ', ', @real_fields ).
1276 join( ', ', @values ).
1280 $statement .= 'DEFAULT VALUES';
1282 warn "[debug]$me $statement\n" if $DEBUG > 1;
1283 my $sth = dbh->prepare($statement) or return dbh->errstr;
1285 local $SIG{HUP} = 'IGNORE';
1286 local $SIG{INT} = 'IGNORE';
1287 local $SIG{QUIT} = 'IGNORE';
1288 local $SIG{TERM} = 'IGNORE';
1289 local $SIG{TSTP} = 'IGNORE';
1290 local $SIG{PIPE} = 'IGNORE';
1292 $sth->execute or return $sth->errstr;
1294 # get inserted id from the database, if applicable & needed
1295 if ( $db_seq && ! $self->getfield($primary_key) ) {
1296 warn "[debug]$me retreiving sequence from database\n" if $DEBUG;
1300 if ( driver_name eq 'Pg' ) {
1302 #my $oid = $sth->{'pg_oid_status'};
1303 #my $i_sql = "SELECT $primary_key FROM $table WHERE oid = ?";
1305 my $default = $self->dbdef_table->column($primary_key)->quoted_default;
1306 unless ( $default =~ /^nextval\(\(?'"?([\w\.]+)"?'/i ) {
1307 dbh->rollback if $FS::UID::AutoCommit;
1308 return "can't parse $table.$primary_key default value".
1309 " for sequence name: $default";
1313 my $i_sql = "SELECT currval('$sequence')";
1314 my $i_sth = dbh->prepare($i_sql) or do {
1315 dbh->rollback if $FS::UID::AutoCommit;
1318 $i_sth->execute() or do { #$i_sth->execute($oid)
1319 dbh->rollback if $FS::UID::AutoCommit;
1320 return $i_sth->errstr;
1322 $insertid = $i_sth->fetchrow_arrayref->[0];
1324 } elsif ( driver_name eq 'mysql' ) {
1326 $insertid = dbh->{'mysql_insertid'};
1327 # work around mysql_insertid being null some of the time, ala RT :/
1328 unless ( $insertid ) {
1329 warn "WARNING: DBD::mysql didn't return mysql_insertid; ".
1330 "using SELECT LAST_INSERT_ID();";
1331 my $i_sql = "SELECT LAST_INSERT_ID()";
1332 my $i_sth = dbh->prepare($i_sql) or do {
1333 dbh->rollback if $FS::UID::AutoCommit;
1336 $i_sth->execute or do {
1337 dbh->rollback if $FS::UID::AutoCommit;
1338 return $i_sth->errstr;
1340 $insertid = $i_sth->fetchrow_arrayref->[0];
1345 dbh->rollback if $FS::UID::AutoCommit;
1346 return "don't know how to retreive inserted ids from ". driver_name.
1347 ", try using counterfiles (maybe run dbdef-create?)";
1351 $self->setfield($primary_key, $insertid);
1356 if ( defined( dbdef->table('h_'. $table) ) && ! $no_history ) {
1357 my $h_statement = $self->_h_statement('insert');
1358 warn "[debug]$me $h_statement\n" if $DEBUG > 2;
1359 $h_sth = dbh->prepare($h_statement) or do {
1360 dbh->rollback if $FS::UID::AutoCommit;
1366 $h_sth->execute or return $h_sth->errstr if $h_sth;
1368 dbh->commit or croak dbh->errstr if $FS::UID::AutoCommit;
1370 # Now that it has been saved, reset the encrypted fields so that $new
1371 # can still be used.
1372 foreach my $field (keys %{$saved}) {
1373 $self->setfield($field, $saved->{$field});
1381 Depriciated (use insert instead).
1386 cluck "warning: FS::Record::add deprecated!";
1387 insert @_; #call method in this scope
1392 Delete this record from the database. If there is an error, returns the error,
1393 otherwise returns false.
1400 my $statement = "DELETE FROM ". $self->table. " WHERE ". join(' AND ',
1402 $self->getfield($_) eq ''
1403 #? "( $_ IS NULL OR $_ = \"\" )"
1404 ? ( driver_name eq 'Pg'
1406 : "( $_ IS NULL OR $_ = \"\" )"
1408 : "$_ = ". _quote($self->getfield($_),$self->table,$_)
1409 } ( $self->dbdef_table->primary_key )
1410 ? ( $self->dbdef_table->primary_key)
1411 : real_fields($self->table)
1413 warn "[debug]$me $statement\n" if $DEBUG > 1;
1414 my $sth = dbh->prepare($statement) or return dbh->errstr;
1417 if ( defined dbdef->table('h_'. $self->table) ) {
1418 my $h_statement = $self->_h_statement('delete');
1419 warn "[debug]$me $h_statement\n" if $DEBUG > 2;
1420 $h_sth = dbh->prepare($h_statement) or return dbh->errstr;
1425 my $primary_key = $self->dbdef_table->primary_key;
1427 local $SIG{HUP} = 'IGNORE';
1428 local $SIG{INT} = 'IGNORE';
1429 local $SIG{QUIT} = 'IGNORE';
1430 local $SIG{TERM} = 'IGNORE';
1431 local $SIG{TSTP} = 'IGNORE';
1432 local $SIG{PIPE} = 'IGNORE';
1434 my $rc = $sth->execute or return $sth->errstr;
1435 #not portable #return "Record not found, statement:\n$statement" if $rc eq "0E0";
1436 $h_sth->execute or return $h_sth->errstr if $h_sth;
1438 dbh->commit or croak dbh->errstr if $FS::UID::AutoCommit;
1440 #no need to needlessly destoy the data either (causes problems actually)
1441 #undef $self; #no need to keep object!
1448 Depriciated (use delete instead).
1453 cluck "warning: FS::Record::del deprecated!";
1454 &delete(@_); #call method in this scope
1457 =item replace OLD_RECORD
1459 Replace the OLD_RECORD with this one in the database. If there is an error,
1460 returns the error, otherwise returns false.
1465 my ($new, $old) = (shift, shift);
1467 $old = $new->replace_old unless defined($old);
1469 warn "[debug]$me $new ->replace $old\n" if $DEBUG;
1471 if ( $new->can('replace_check') ) {
1472 my $error = $new->replace_check($old);
1473 return $error if $error;
1476 return "Records not in same table!" unless $new->table eq $old->table;
1478 my $primary_key = $old->dbdef_table->primary_key;
1479 return "Can't change primary key $primary_key ".
1480 'from '. $old->getfield($primary_key).
1481 ' to ' . $new->getfield($primary_key)
1483 && ( $old->getfield($primary_key) ne $new->getfield($primary_key) );
1485 my $error = $new->check;
1486 return $error if $error;
1488 # Encrypt for replace
1490 if ( $conf->exists('encryption')
1491 && defined(eval '@FS::'. $new->table . '::encrypted_fields')
1492 && scalar( eval '@FS::'. $new->table . '::encrypted_fields')
1494 foreach my $field (eval '@FS::'. $new->table . '::encrypted_fields') {
1495 next if $field eq 'payinfo'
1496 && ($new->isa('FS::payinfo_transaction_Mixin')
1497 || $new->isa('FS::payinfo_Mixin') )
1499 && !grep { $new->payby eq $_ } @encrypt_payby;
1500 $saved->{$field} = $new->getfield($field);
1501 $new->setfield($field, $new->encrypt($new->getfield($field)));
1505 #my @diff = grep $new->getfield($_) ne $old->getfield($_), $old->fields;
1506 my %diff = map { ($new->getfield($_) ne $old->getfield($_))
1507 ? ($_, $new->getfield($_)) : () } $old->fields;
1509 unless (keys(%diff) || $no_update_diff ) {
1510 carp "[warning]$me ". ref($new)."->replace ".
1511 ( $primary_key ? "$primary_key ".$new->get($primary_key) : '' ).
1512 ": records identical"
1513 unless $nowarn_identical;
1517 my $statement = "UPDATE ". $old->table. " SET ". join(', ',
1519 "$_ = ". _quote($new->getfield($_),$old->table,$_)
1520 } real_fields($old->table)
1525 if ( $old->getfield($_) eq '' ) {
1527 #false laziness w/qsearch
1528 if ( driver_name eq 'Pg' ) {
1529 my $type = $old->dbdef_table->column($_)->type;
1530 if ( $type =~ /(int|(big)?serial)/i ) {
1533 qq-( $_ IS NULL OR $_ = '' )-;
1536 qq-( $_ IS NULL OR $_ = "" )-;
1540 "$_ = ". _quote($old->getfield($_),$old->table,$_);
1543 } ( $primary_key ? ( $primary_key ) : real_fields($old->table) )
1546 warn "[debug]$me $statement\n" if $DEBUG > 1;
1547 my $sth = dbh->prepare($statement) or return dbh->errstr;
1550 if ( defined dbdef->table('h_'. $old->table) ) {
1551 my $h_old_statement = $old->_h_statement('replace_old');
1552 warn "[debug]$me $h_old_statement\n" if $DEBUG > 2;
1553 $h_old_sth = dbh->prepare($h_old_statement) or return dbh->errstr;
1559 if ( defined dbdef->table('h_'. $new->table) ) {
1560 my $h_new_statement = $new->_h_statement('replace_new');
1561 warn "[debug]$me $h_new_statement\n" if $DEBUG > 2;
1562 $h_new_sth = dbh->prepare($h_new_statement) or return dbh->errstr;
1567 local $SIG{HUP} = 'IGNORE';
1568 local $SIG{INT} = 'IGNORE';
1569 local $SIG{QUIT} = 'IGNORE';
1570 local $SIG{TERM} = 'IGNORE';
1571 local $SIG{TSTP} = 'IGNORE';
1572 local $SIG{PIPE} = 'IGNORE';
1574 my $rc = $sth->execute or return $sth->errstr;
1575 #not portable #return "Record not found (or records identical)." if $rc eq "0E0";
1576 $h_old_sth->execute or return $h_old_sth->errstr if $h_old_sth;
1577 $h_new_sth->execute or return $h_new_sth->errstr if $h_new_sth;
1579 dbh->commit or croak dbh->errstr if $FS::UID::AutoCommit;
1581 # Now that it has been saved, reset the encrypted fields so that $new
1582 # can still be used.
1583 foreach my $field (keys %{$saved}) {
1584 $new->setfield($field, $saved->{$field});
1592 my( $self ) = shift;
1593 warn "[$me] replace called with no arguments; autoloading old record\n"
1596 my $primary_key = $self->dbdef_table->primary_key;
1597 if ( $primary_key ) {
1598 $self->by_key( $self->$primary_key() ) #this is what's returned
1599 or croak "can't find ". $self->table. ".$primary_key ".
1600 $self->$primary_key();
1602 croak $self->table. " has no primary key; pass old record as argument";
1609 Depriciated (use replace instead).
1614 cluck "warning: FS::Record::rep deprecated!";
1615 replace @_; #call method in this scope
1620 Checks custom fields. Subclasses should still provide a check method to validate
1621 non-custom fields, etc., and call this method via $self->SUPER::check.
1627 foreach my $field ($self->virtual_fields) {
1628 my $error = $self->ut_textn($field);
1629 return $error if $error;
1634 =item virtual_fields [ TABLE ]
1636 Returns a list of virtual fields defined for the table. This should not
1637 be exported, and should only be called as an instance or class method.
1641 sub virtual_fields {
1644 $table = $self->table or confess "virtual_fields called on non-table";
1646 confess "Unknown table $table" unless dbdef->table($table);
1648 return () unless dbdef->table('part_virtual_field');
1650 unless ( $virtual_fields_cache{$table} ) {
1651 my $concat = [ "'cf_'", "name" ];
1652 my $query = "SELECT ".concat_sql($concat).' from part_virtual_field ' .
1653 "WHERE dbtable = '$table'";
1655 my $result = $dbh->selectcol_arrayref($query);
1656 confess "Error executing virtual fields query: $query: ". $dbh->errstr
1658 $virtual_fields_cache{$table} = $result;
1661 @{$virtual_fields_cache{$table}};
1665 =item process_batch_import JOB OPTIONS_HASHREF PARAMS
1667 Processes a batch import as a queued JSRPC job
1669 JOB is an FS::queue entry.
1671 OPTIONS_HASHREF can have the following keys:
1677 Table name (required).
1681 Listref of field names for static fields. They will be given values from the
1682 PARAMS hashref and passed as a "params" hashref to batch_import.
1686 Formats hashref. Keys are field names, values are listrefs that define the
1689 Each listref value can be a column name or a code reference. Coderefs are run
1690 with the row object, data and a FS::Conf object as the three parameters.
1691 For example, this coderef does the same thing as using the "columnname" string:
1694 my( $record, $data, $conf ) = @_;
1695 $record->columnname( $data );
1698 Coderefs are run after all "column name" fields are assigned.
1702 Optional format hashref of types. Keys are field names, values are "csv",
1703 "xls" or "fixedlength". Overrides automatic determination of file type
1706 =item format_headers
1708 Optional format hashref of header lines. Keys are field names, values are 0
1709 for no header, 1 to ignore the first line, or to higher numbers to ignore that
1712 =item format_sep_chars
1714 Optional format hashref of CSV sep_chars. Keys are field names, values are the
1715 CSV separation character.
1717 =item format_fixedlenth_formats
1719 Optional format hashref of fixed length format defintiions. Keys are field
1720 names, values Parse::FixedLength listrefs of field definitions.
1724 Set true to default to CSV file type if the filename does not contain a
1725 recognizable ".csv" or ".xls" extension (and type is not pre-specified by
1730 PARAMS is a base64-encoded Storable string containing the POSTed data as
1731 a hash ref. It normally contains at least one field, "uploaded files",
1732 generated by /elements/file-upload.html and containing the list of uploaded
1733 files. Currently only supports a single file named "file".
1737 use Storable qw(thaw);
1740 sub process_batch_import {
1741 my($job, $opt) = ( shift, shift );
1743 my $table = $opt->{table};
1744 my @pass_params = $opt->{params} ? @{ $opt->{params} } : ();
1745 my %formats = %{ $opt->{formats} };
1747 my $param = thaw(decode_base64(shift));
1748 warn Dumper($param) if $DEBUG;
1750 my $files = $param->{'uploaded_files'}
1751 or die "No files provided.\n";
1753 my (%files) = map { /^(\w+):([\.\w]+)$/ ? ($1,$2):() } split /,/, $files;
1755 my $dir = '%%%FREESIDE_CACHE%%%/cache.'. $FS::UID::datasrc. '/';
1756 my $file = $dir. $files{'file'};
1761 formats => \%formats,
1762 format_types => $opt->{format_types},
1763 format_headers => $opt->{format_headers},
1764 format_sep_chars => $opt->{format_sep_chars},
1765 format_fixedlength_formats => $opt->{format_fixedlength_formats},
1766 format_xml_formats => $opt->{format_xml_formats},
1767 format_asn_formats => $opt->{format_asn_formats},
1768 format_row_callbacks => $opt->{format_row_callbacks},
1773 format => $param->{format},
1774 params => { map { $_ => $param->{$_} } @pass_params },
1776 default_csv => $opt->{default_csv},
1777 postinsert_callback => $opt->{postinsert_callback},
1780 if ( $opt->{'batch_namecol'} ) {
1781 $iopt{'batch_namevalue'} = $param->{ $opt->{'batch_namecol'} };
1782 $iopt{$_} = $opt->{$_} foreach qw( batch_keycol batch_table batch_namecol );
1785 my $error = FS::Record::batch_import( \%iopt );
1789 die "$error\n" if $error;
1792 =item batch_import PARAM_HASHREF
1794 Class method for batch imports. Available params:
1800 =item format - usual way to specify import, with this format string selecting data from the formats and format_* info hashes
1806 =item format_headers
1808 =item format_sep_chars
1810 =item format_fixedlength_formats
1812 =item format_row_callbacks
1814 =item fields - Alternate way to specify import, specifying import fields directly as a listref
1816 =item preinsert_callback
1818 =item postinsert_callback
1824 FS::queue object, will be updated with progress
1830 csv, xls, fixedlength, xml
1842 warn "$me batch_import call with params: \n". Dumper($param)
1845 my $table = $param->{table};
1847 my $job = $param->{job};
1848 my $file = $param->{file};
1849 my $params = $param->{params} || {};
1851 my( $type, $header, $sep_char,
1852 $fixedlength_format, $xml_format, $asn_format,
1853 $row_callback, @fields );
1855 my $postinsert_callback = '';
1856 $postinsert_callback = $param->{'postinsert_callback'}
1857 if $param->{'postinsert_callback'};
1858 my $preinsert_callback = '';
1859 $preinsert_callback = $param->{'preinsert_callback'}
1860 if $param->{'preinsert_callback'};
1862 if ( $param->{'format'} ) {
1864 my $format = $param->{'format'};
1865 my $formats = $param->{formats};
1866 die "unknown format $format" unless exists $formats->{ $format };
1868 $type = $param->{'format_types'}
1869 ? $param->{'format_types'}{ $format }
1870 : $param->{type} || 'csv';
1873 $header = $param->{'format_headers'}
1874 ? $param->{'format_headers'}{ $param->{'format'} }
1877 $sep_char = $param->{'format_sep_chars'}
1878 ? $param->{'format_sep_chars'}{ $param->{'format'} }
1881 $fixedlength_format =
1882 $param->{'format_fixedlength_formats'}
1883 ? $param->{'format_fixedlength_formats'}{ $param->{'format'} }
1887 $param->{'format_xml_formats'}
1888 ? $param->{'format_xml_formats'}{ $param->{'format'} }
1892 $param->{'format_asn_formats'}
1893 ? $param->{'format_asn_formats'}{ $param->{'format'} }
1897 $param->{'format_row_callbacks'}
1898 ? $param->{'format_row_callbacks'}{ $param->{'format'} }
1901 @fields = @{ $formats->{ $format } };
1903 } elsif ( $param->{'fields'} ) {
1905 $type = ''; #infer from filename
1908 $fixedlength_format = '';
1910 @fields = @{ $param->{'fields'} };
1913 die "neither format nor fields specified";
1916 #my $file = $param->{file};
1919 if ( $file =~ /\.(\w+)$/i ) {
1923 warn "can't parse file type from filename $file; defaulting to CSV";
1927 if $param->{'default_csv'} && $type ne 'xls';
1935 my $asn_header_buffer;
1936 if ( $type eq 'csv' || $type eq 'fixedlength' ) {
1938 if ( $type eq 'csv' ) {
1940 my %attr = ( 'binary' => 1, );
1941 $attr{sep_char} = $sep_char if $sep_char;
1942 $parser = new Text::CSV_XS \%attr;
1944 } elsif ( $type eq 'fixedlength' ) {
1946 eval "use Parse::FixedLength;";
1948 $parser = Parse::FixedLength->new($fixedlength_format);
1952 die "Unknown file type $type\n";
1955 @buffer = split(/\r?\n/, slurp($file) );
1956 splice(@buffer, 0, ($header || 0) );
1957 $count = scalar(@buffer);
1959 } elsif ( $type eq 'xls' ) {
1961 eval "use Spreadsheet::ParseExcel;";
1964 eval "use DateTime::Format::Excel;";
1965 #for now, just let the error be thrown if it is used, since only CDR
1966 # formats bill_west and troop use it, not other excel-parsing things
1969 my $excel = Spreadsheet::ParseExcel::Workbook->new->Parse($file);
1971 $parser = $excel->{Worksheet}[0]; #first sheet
1973 $count = $parser->{MaxRow} || $parser->{MinRow};
1976 $row = $header || 0;
1978 } elsif ( $type eq 'xml' ) {
1981 eval "use XML::Simple;";
1983 my $xmlrow = $xml_format->{'xmlrow'};
1984 $parser = $xml_format->{'xmlkeys'};
1985 die 'no xmlkeys specified' unless ref $parser eq 'ARRAY';
1986 my $data = XML::Simple::XMLin(
1988 'SuppressEmpty' => '', #sets empty values to ''
1992 $rows = $rows->{$_} foreach @$xmlrow;
1993 $rows = [ $rows ] if ref($rows) ne 'ARRAY';
1994 $count = @buffer = @$rows;
1996 } elsif ( $type eq 'asn.1' ) {
1998 eval "use Convert::ASN1";
2001 my $asn = Convert::ASN1->new;
2002 $asn->prepare( $asn_format->{'spec'} ) or die $asn->error;
2004 $parser = $asn->find( $asn_format->{'macro'} ) or die $asn->error;
2006 my $data = slurp($file);
2007 my $asn_output = $parser->decode( $data )
2008 or return "No ". $asn_format->{'macro'}. " found\n";
2010 $asn_header_buffer = &{ $asn_format->{'header_buffer'} }( $asn_output );
2012 my $rows = &{ $asn_format->{'arrayref'} }( $asn_output );
2013 $count = @buffer = @$rows;
2016 die "Unknown file type $type\n";
2021 local $SIG{HUP} = 'IGNORE';
2022 local $SIG{INT} = 'IGNORE';
2023 local $SIG{QUIT} = 'IGNORE';
2024 local $SIG{TERM} = 'IGNORE';
2025 local $SIG{TSTP} = 'IGNORE';
2026 local $SIG{PIPE} = 'IGNORE';
2028 my $oldAutoCommit = $FS::UID::AutoCommit;
2029 local $FS::UID::AutoCommit = 0;
2032 #my $params = $param->{params} || {};
2033 if ( $param->{'batch_namecol'} && $param->{'batch_namevalue'} ) {
2034 my $batch_col = $param->{'batch_keycol'};
2036 my $batch_class = 'FS::'. $param->{'batch_table'};
2037 my $batch = $batch_class->new({
2038 $param->{'batch_namecol'} => $param->{'batch_namevalue'}
2040 my $error = $batch->insert;
2042 $dbh->rollback if $oldAutoCommit;
2043 return "can't insert batch record: $error";
2045 #primary key via dbdef? (so the column names don't have to match)
2046 my $batch_value = $batch->get( $param->{'batch_keycol'} );
2048 $params->{ $batch_col } = $batch_value;
2051 #my $job = $param->{job};
2054 my( $last, $min_sec ) = ( time, 5 ); #progressbar foo
2058 my %hash = %$params;
2059 if ( $type eq 'csv' ) {
2061 last unless scalar(@buffer);
2062 $line = shift(@buffer);
2064 next if $line =~ /^\s*$/; #skip empty lines
2066 $line = &{$row_callback}($line) if $row_callback;
2068 next if $line =~ /^\s*$/; #skip empty lines
2070 $parser->parse($line) or do {
2071 $dbh->rollback if $oldAutoCommit;
2072 return "can't parse: ". $parser->error_input() . " " . $parser->error_diag;
2074 @columns = $parser->fields();
2076 } elsif ( $type eq 'fixedlength' ) {
2078 last unless scalar(@buffer);
2079 $line = shift(@buffer);
2081 @columns = $parser->parse($line);
2083 } elsif ( $type eq 'xls' ) {
2085 last if $row > ($parser->{MaxRow} || $parser->{MinRow})
2086 || ! $parser->{Cells}[$row];
2088 my @row = @{ $parser->{Cells}[$row] };
2089 @columns = map $_->{Val}, @row;
2092 #warn $z++. ": $_\n" for @columns;
2094 } elsif ( $type eq 'xml' ) {
2096 # $parser = [ 'Column0Key', 'Column1Key' ... ]
2097 last unless scalar(@buffer);
2098 my $row = shift @buffer;
2099 @columns = @{ $row }{ @$parser };
2101 } elsif ( $type eq 'asn.1' ) {
2103 last unless scalar(@buffer);
2104 my $row = shift @buffer;
2105 &{ $asn_format->{row_callback} }( $row, $asn_header_buffer )
2106 if $asn_format->{row_callback};
2107 foreach my $key ( keys %{ $asn_format->{map} } ) {
2108 $hash{$key} = &{ $asn_format->{map}{$key} }( $row, $asn_header_buffer );
2112 die "Unknown file type $type\n";
2117 foreach my $field ( @fields ) {
2119 my $value = shift @columns;
2121 if ( ref($field) eq 'CODE' ) {
2122 #&{$field}(\%hash, $value);
2123 push @later, $field, $value;
2125 #??? $hash{$field} = $value if length($value);
2126 $hash{$field} = $value if defined($value) && length($value);
2131 #my $table = $param->{table};
2132 my $class = "FS::$table";
2134 my $record = $class->new( \%hash );
2137 while ( scalar(@later) ) {
2138 my $sub = shift @later;
2139 my $data = shift @later;
2141 &{$sub}($record, $data, $conf, $param); # $record->&{$sub}($data, $conf)
2144 $dbh->rollback if $oldAutoCommit;
2145 return "can't insert record". ( $line ? " for $line" : '' ). ": $@";
2147 last if exists( $param->{skiprow} );
2149 next if exists( $param->{skiprow} );
2151 if ( $preinsert_callback ) {
2152 my $error = &{$preinsert_callback}($record, $param);
2154 $dbh->rollback if $oldAutoCommit;
2155 return "preinsert_callback error". ( $line ? " for $line" : '' ).
2158 next if exists $param->{skiprow} && $param->{skiprow};
2161 my $error = $record->insert;
2164 $dbh->rollback if $oldAutoCommit;
2165 return "can't insert record". ( $line ? " for $line" : '' ). ": $error";
2171 if ( $postinsert_callback ) {
2172 my $error = &{$postinsert_callback}($record, $param);
2174 $dbh->rollback if $oldAutoCommit;
2175 return "postinsert_callback error". ( $line ? " for $line" : '' ).
2180 if ( $job && time - $min_sec > $last ) { #progress bar
2181 $job->update_statustext( int(100 * $imported / $count) );
2187 unless ( $imported || $param->{empty_ok} ) {
2188 $dbh->rollback if $oldAutoCommit;
2189 return "Empty file!";
2192 $dbh->commit or die $dbh->errstr if $oldAutoCommit;
2199 my( $self, $action, $time ) = @_;
2203 my %nohistory = map { $_=>1 } $self->nohistory_fields;
2206 grep { defined($self->get($_)) && $self->get($_) ne "" && ! $nohistory{$_} }
2207 real_fields($self->table);
2210 # If we're encrypting then don't store the payinfo in the history
2211 if ( $conf && $conf->exists('encryption') && $self->table ne 'banned_pay' ) {
2212 @fields = grep { $_ ne 'payinfo' } @fields;
2215 my @values = map { _quote( $self->getfield($_), $self->table, $_) } @fields;
2217 "INSERT INTO h_". $self->table. " ( ".
2218 join(', ', qw(history_date history_usernum history_action), @fields ).
2221 $FS::CurrentUser::CurrentUser->usernum,
2222 dbh->quote($action),
2231 B<Warning>: External use is B<deprecated>.
2233 Replaces COLUMN in record with a unique number, using counters in the
2234 filesystem. Used by the B<insert> method on single-field unique columns
2235 (see L<DBIx::DBSchema::Table>) and also as a fallback for primary keys
2236 that aren't SERIAL (Pg) or AUTO_INCREMENT (mysql).
2238 Returns the new value.
2243 my($self,$field) = @_;
2244 my($table)=$self->table;
2246 croak "Unique called on field $field, but it is ",
2247 $self->getfield($field),
2249 if $self->getfield($field);
2251 #warn "table $table is tainted" if is_tainted($table);
2252 #warn "field $field is tainted" if is_tainted($field);
2254 my($counter) = new File::CounterFile "$table.$field",0;
2256 my $index = $counter->inc;
2257 $index = $counter->inc while qsearchs($table, { $field=>$index } );
2259 $index =~ /^(\d*)$/;
2262 $self->setfield($field,$index);
2266 =item ut_float COLUMN
2268 Check/untaint floating point numeric data: 1.1, 1, 1.1e10, 1e10. May not be
2269 null. If there is an error, returns the error, otherwise returns false.
2274 my($self,$field)=@_ ;
2275 ($self->getfield($field) =~ /^\s*(\d+\.\d+)\s*$/ ||
2276 $self->getfield($field) =~ /^\s*(\d+)\s*$/ ||
2277 $self->getfield($field) =~ /^\s*(\d+\.\d+e\d+)\s*$/ ||
2278 $self->getfield($field) =~ /^\s*(\d+e\d+)\s*$/)
2279 or return "Illegal or empty (float) $field: ". $self->getfield($field);
2280 $self->setfield($field,$1);
2283 =item ut_floatn COLUMN
2285 Check/untaint floating point numeric data: 1.1, 1, 1.1e10, 1e10. May be
2286 null. If there is an error, returns the error, otherwise returns false.
2290 #false laziness w/ut_ipn
2292 my( $self, $field ) = @_;
2293 if ( $self->getfield($field) =~ /^()$/ ) {
2294 $self->setfield($field,'');
2297 $self->ut_float($field);
2301 =item ut_sfloat COLUMN
2303 Check/untaint signed floating point numeric data: 1.1, 1, 1.1e10, 1e10.
2304 May not be null. If there is an error, returns the error, otherwise returns
2310 my($self,$field)=@_ ;
2311 ($self->getfield($field) =~ /^\s*(-?\d+\.\d+)\s*$/ ||
2312 $self->getfield($field) =~ /^\s*(-?\d+)\s*$/ ||
2313 $self->getfield($field) =~ /^\s*(-?\d+\.\d+[eE]-?\d+)\s*$/ ||
2314 $self->getfield($field) =~ /^\s*(-?\d+[eE]-?\d+)\s*$/)
2315 or return "Illegal or empty (float) $field: ". $self->getfield($field);
2316 $self->setfield($field,$1);
2319 =item ut_sfloatn COLUMN
2321 Check/untaint signed floating point numeric data: 1.1, 1, 1.1e10, 1e10. May be
2322 null. If there is an error, returns the error, otherwise returns false.
2327 my( $self, $field ) = @_;
2328 if ( $self->getfield($field) =~ /^()$/ ) {
2329 $self->setfield($field,'');
2332 $self->ut_sfloat($field);
2336 =item ut_snumber COLUMN
2338 Check/untaint signed numeric data (whole numbers). If there is an error,
2339 returns the error, otherwise returns false.
2344 my($self, $field) = @_;
2345 $self->getfield($field) =~ /^\s*(-?)\s*(\d+)\s*$/
2346 or return "Illegal or empty (numeric) $field: ". $self->getfield($field);
2347 $self->setfield($field, "$1$2");
2351 =item ut_snumbern COLUMN
2353 Check/untaint signed numeric data (whole numbers). If there is an error,
2354 returns the error, otherwise returns false.
2359 my($self, $field) = @_;
2360 $self->getfield($field) =~ /^\s*(-?)\s*(\d*)\s*$/
2361 or return "Illegal (numeric) $field: ". $self->getfield($field);
2363 return "Illegal (numeric) $field: ". $self->getfield($field)
2366 $self->setfield($field, "$1$2");
2370 =item ut_number COLUMN
2372 Check/untaint simple numeric data (whole numbers). May not be null. If there
2373 is an error, returns the error, otherwise returns false.
2378 my($self,$field)=@_;
2379 $self->getfield($field) =~ /^\s*(\d+)\s*$/
2380 or return "Illegal or empty (numeric) $field: ". $self->getfield($field);
2381 $self->setfield($field,$1);
2385 =item ut_numbern COLUMN
2387 Check/untaint simple numeric data (whole numbers). May be null. If there is
2388 an error, returns the error, otherwise returns false.
2393 my($self,$field)=@_;
2394 $self->getfield($field) =~ /^\s*(\d*)\s*$/
2395 or return "Illegal (numeric) $field: ". $self->getfield($field);
2396 $self->setfield($field,$1);
2400 =item ut_money COLUMN
2402 Check/untaint monetary numbers. May be negative. Set to 0 if null. If there
2403 is an error, returns the error, otherwise returns false.
2408 my($self,$field)=@_;
2410 if ( $self->getfield($field) eq '' ) {
2411 $self->setfield($field, 0);
2412 } elsif ( $self->getfield($field) =~ /^\s*(\-)?\s*(\d*)(\.\d{1})\s*$/ ) {
2413 #handle one decimal place without barfing out
2414 $self->setfield($field, ( ($1||''). ($2||''). ($3.'0') ) || 0);
2415 } elsif ( $self->getfield($field) =~ /^\s*(\-)?\s*(\d*)(\.\d{2})?\s*$/ ) {
2416 $self->setfield($field, ( ($1||''). ($2||''). ($3||'') ) || 0);
2418 return "Illegal (money) $field: ". $self->getfield($field);
2424 =item ut_moneyn COLUMN
2426 Check/untaint monetary numbers. May be negative. If there
2427 is an error, returns the error, otherwise returns false.
2432 my($self,$field)=@_;
2433 if ($self->getfield($field) eq '') {
2434 $self->setfield($field, '');
2437 $self->ut_money($field);
2440 =item ut_currencyn COLUMN
2442 Check/untaint currency indicators, such as USD or EUR. May be null. If there
2443 is an error, returns the error, otherwise returns false.
2448 my($self, $field) = @_;
2449 if ($self->getfield($field) eq '') { #can be null
2450 $self->setfield($field, '');
2453 $self->ut_currency($field);
2456 =item ut_currency COLUMN
2458 Check/untaint currency indicators, such as USD or EUR. May not be null. If
2459 there is an error, returns the error, otherwise returns false.
2464 my($self, $field) = @_;
2465 my $value = uc( $self->getfield($field) );
2466 if ( code2currency($value) ) {
2467 $self->setfield($value);
2469 return "Unknown currency $value";
2475 =item ut_text COLUMN
2477 Check/untaint text. Alphanumerics, spaces, and the following punctuation
2478 symbols are currently permitted: ! @ # $ % & ( ) - + ; : ' " , . ? / = [ ] < >
2479 May not be null. If there is an error, returns the error, otherwise returns
2485 my($self,$field)=@_;
2486 #warn "msgcat ". \&msgcat. "\n";
2487 #warn "notexist ". \¬exist. "\n";
2488 #warn "AUTOLOAD ". \&AUTOLOAD. "\n";
2489 $self->getfield($field)
2490 =~ /^([\wô \!\@\#\$\%\&\(\)\-\+\;\:\'\"\,\.\?\/\=\[\]\<\>$money_char]+)$/
2491 or return gettext('illegal_or_empty_text'). " $field: ".
2492 $self->getfield($field);
2493 $self->setfield($field,$1);
2497 =item ut_textn COLUMN
2499 Check/untaint text. Alphanumerics, spaces, and the following punctuation
2500 symbols are currently permitted: ! @ # $ % & ( ) - + ; : ' " , . ? / = [ ] < >
2501 May be null. If there is an error, returns the error, otherwise returns false.
2506 my($self,$field)=@_;
2507 return $self->setfield($field, '') if $self->getfield($field) =~ /^$/;
2508 $self->ut_text($field);
2511 =item ut_alpha COLUMN
2513 Check/untaint alphanumeric strings (no spaces). May not be null. If there is
2514 an error, returns the error, otherwise returns false.
2519 my($self,$field)=@_;
2520 $self->getfield($field) =~ /^(\w+)$/
2521 or return "Illegal or empty (alphanumeric) $field: ".
2522 $self->getfield($field);
2523 $self->setfield($field,$1);
2527 =item ut_alphan COLUMN
2529 Check/untaint alphanumeric strings (no spaces). May be null. If there is an
2530 error, returns the error, otherwise returns false.
2535 my($self,$field)=@_;
2536 $self->getfield($field) =~ /^(\w*)$/
2537 or return "Illegal (alphanumeric) $field: ". $self->getfield($field);
2538 $self->setfield($field,$1);
2542 =item ut_alphasn COLUMN
2544 Check/untaint alphanumeric strings, spaces allowed. May be null. If there is
2545 an error, returns the error, otherwise returns false.
2550 my($self,$field)=@_;
2551 $self->getfield($field) =~ /^([\w ]*)$/
2552 or return "Illegal (alphanumeric) $field: ". $self->getfield($field);
2553 $self->setfield($field,$1);
2558 =item ut_alpha_lower COLUMN
2560 Check/untaint lowercase alphanumeric strings (no spaces). May not be null. If
2561 there is an error, returns the error, otherwise returns false.
2565 sub ut_alpha_lower {
2566 my($self,$field)=@_;
2567 $self->getfield($field) =~ /[[:upper:]]/
2568 and return "Uppercase characters are not permitted in $field";
2569 $self->ut_alpha($field);
2572 =item ut_phonen COLUMN [ COUNTRY ]
2574 Check/untaint phone numbers. May be null. If there is an error, returns
2575 the error, otherwise returns false.
2577 Takes an optional two-letter ISO country code; without it or with unsupported
2578 countries, ut_phonen simply calls ut_alphan.
2583 my( $self, $field, $country ) = @_;
2584 return $self->ut_alphan($field) unless defined $country;
2585 my $phonen = $self->getfield($field);
2586 if ( $phonen eq '' ) {
2587 $self->setfield($field,'');
2588 } elsif ( $country eq 'US' || $country eq 'CA' ) {
2590 $phonen = $conf->config('cust_main-default_areacode').$phonen
2591 if length($phonen)==7 && $conf->config('cust_main-default_areacode');
2592 $phonen =~ /^(\d{3})(\d{3})(\d{4})(\d*)$/
2593 or return gettext('illegal_phone'). " $field: ". $self->getfield($field);
2594 $phonen = "$1-$2-$3";
2595 $phonen .= " x$4" if $4;
2596 $self->setfield($field,$phonen);
2598 warn "warning: don't know how to check phone numbers for country $country";
2599 return $self->ut_textn($field);
2606 Check/untaint hexadecimal values.
2611 my($self, $field) = @_;
2612 $self->getfield($field) =~ /^([\da-fA-F]+)$/
2613 or return "Illegal (hex) $field: ". $self->getfield($field);
2614 $self->setfield($field, uc($1));
2618 =item ut_hexn COLUMN
2620 Check/untaint hexadecimal values. May be null.
2625 my($self, $field) = @_;
2626 $self->getfield($field) =~ /^([\da-fA-F]*)$/
2627 or return "Illegal (hex) $field: ". $self->getfield($field);
2628 $self->setfield($field, uc($1));
2632 =item ut_mac_addr COLUMN
2634 Check/untaint mac addresses. May be null.
2639 my($self, $field) = @_;
2641 my $mac = $self->get($field);
2644 $self->set($field, $mac);
2646 my $e = $self->ut_hex($field);
2649 return "Illegal (mac address) $field: ". $self->getfield($field)
2650 unless length($self->getfield($field)) == 12;
2656 =item ut_mac_addrn COLUMN
2658 Check/untaint mac addresses. May be null.
2663 my($self, $field) = @_;
2664 ($self->getfield($field) eq '') ? '' : $self->ut_mac_addr($field);
2669 Check/untaint ip addresses. IPv4 only for now, though ::1 is auto-translated
2675 my( $self, $field ) = @_;
2676 $self->setfield($field, '127.0.0.1') if $self->getfield($field) eq '::1';
2677 $self->getfield($field) =~ /^(\d+)\.(\d+)\.(\d+)\.(\d+)$/
2678 or return "Illegal (IP address) $field: ". $self->getfield($field);
2679 for ( $1, $2, $3, $4 ) { return "Illegal (IP address) $field" if $_ > 255; }
2680 $self->setfield($field, "$1.$2.$3.$4");
2686 Check/untaint ip addresses. IPv4 only for now, though ::1 is auto-translated
2687 to 127.0.0.1. May be null.
2692 my( $self, $field ) = @_;
2693 if ( $self->getfield($field) =~ /^()$/ ) {
2694 $self->setfield($field,'');
2697 $self->ut_ip($field);
2701 =item ut_ip46 COLUMN
2703 Check/untaint IPv4 or IPv6 address.
2708 my( $self, $field ) = @_;
2709 my $ip = NetAddr::IP->new($self->getfield($field))
2710 or return "Illegal (IP address) $field: ".$self->getfield($field);
2711 $self->setfield($field, lc($ip->addr));
2717 Check/untaint IPv6 or IPv6 address. May be null.
2722 my( $self, $field ) = @_;
2723 if ( $self->getfield($field) =~ /^$/ ) {
2724 $self->setfield($field, '');
2727 $self->ut_ip46($field);
2730 =item ut_coord COLUMN [ LOWER [ UPPER ] ]
2732 Check/untaint coordinates.
2733 Accepts the following forms:
2743 The "DDD MM SS" and "DDD MM MMM" are potentially ambiguous.
2744 The latter form (that is, the MMM are thousands of minutes) is
2745 assumed if the "MMM" is exactly three digits or two digits > 59.
2747 To be safe, just use the DDD.DDDDD form.
2749 If LOWER or UPPER are specified, then the coordinate is checked
2750 for lower and upper bounds, respectively.
2755 my ($self, $field) = (shift, shift);
2758 if ( $field =~ /latitude/ ) {
2759 $lower = $lat_lower;
2761 } elsif ( $field =~ /longitude/ ) {
2763 $upper = $lon_upper;
2766 my $coord = $self->getfield($field);
2767 my $neg = $coord =~ s/^(-)//;
2769 my ($d, $m, $s) = (0, 0, 0);
2772 (($d) = ($coord =~ /^(\s*\d{1,3}(?:\.\d+)?)\s*$/)) ||
2773 (($d, $m) = ($coord =~ /^(\s*\d{1,3})\s+(\d{1,2}(?:\.\d+))\s*$/)) ||
2774 (($d, $m, $s) = ($coord =~ /^(\s*\d{1,3})\s+(\d{1,2})\s+(\d{1,3})\s*$/))
2776 $s = (((($s =~ /^\d{3}$/) or $s > 59) ? ($s / 1000) : ($s / 60)) / 60);
2779 return "Invalid (coordinate with minutes > 59) $field: "
2780 . $self->getfield($field);
2783 $coord = ($neg ? -1 : 1) * sprintf('%.8f', $d + $m + $s);
2785 if (defined($lower) and ($coord < $lower)) {
2786 return "Invalid (coordinate < $lower) $field: "
2787 . $self->getfield($field);;
2790 if (defined($upper) and ($coord > $upper)) {
2791 return "Invalid (coordinate > $upper) $field: "
2792 . $self->getfield($field);;
2795 $self->setfield($field, $coord);
2799 return "Invalid (coordinate) $field: " . $self->getfield($field);
2803 =item ut_coordn COLUMN [ LOWER [ UPPER ] ]
2805 Same as ut_coord, except optionally null.
2811 my ($self, $field) = (shift, shift);
2813 if ($self->getfield($field) =~ /^\s*$/) {
2816 return $self->ut_coord($field, @_);
2821 =item ut_domain COLUMN
2823 Check/untaint host and domain names. May not be null.
2828 my( $self, $field ) = @_;
2829 #$self->getfield($field) =~/^(\w+\.)*\w+$/
2830 $self->getfield($field) =~/^(([\w\-]+\.)*\w+)$/
2831 or return "Illegal (hostname) $field: ". $self->getfield($field);
2832 $self->setfield($field,$1);
2836 =item ut_domainn COLUMN
2838 Check/untaint host and domain names. May be null.
2843 my( $self, $field ) = @_;
2844 if ( $self->getfield($field) =~ /^()$/ ) {
2845 $self->setfield($field,'');
2848 $self->ut_domain($field);
2852 =item ut_name COLUMN
2854 Check/untaint proper names; allows alphanumerics, spaces and the following
2855 punctuation: , . - '
2862 my( $self, $field ) = @_;
2863 # warn "ut_name allowed alphanumerics: +(sort grep /\w/, map { chr() } 0..255), "\n";
2864 $self->getfield($field) =~ /^([\w \,\.\-\']+)$/
2865 or return gettext('illegal_name'). " $field: ". $self->getfield($field);
2870 $self->setfield($field, $name);
2874 =item ut_namen COLUMN
2876 Check/untaint proper names; allows alphanumerics, spaces and the following
2877 punctuation: , . - '
2884 my( $self, $field ) = @_;
2885 return $self->setfield($field, '') if $self->getfield($field) =~ /^$/;
2886 $self->ut_name($field);
2891 Check/untaint zip codes.
2895 my @zip_reqd_countries = qw( AU CA US ); #CA, US implicit...
2898 my( $self, $field, $country ) = @_;
2900 if ( $country eq 'US' ) {
2902 $self->getfield($field) =~ /^\s*(\d{5}(\-\d{4})?)\s*$/
2903 or return gettext('illegal_zip'). " $field for country $country: ".
2904 $self->getfield($field);
2905 $self->setfield($field, $1);
2907 } elsif ( $country eq 'CA' ) {
2909 $self->getfield($field) =~ /^\s*([A-Z]\d[A-Z])\s*(\d[A-Z]\d)\s*$/i
2910 or return gettext('illegal_zip'). " $field for country $country: ".
2911 $self->getfield($field);
2912 $self->setfield($field, "$1 $2");
2916 if ( $self->getfield($field) =~ /^\s*$/
2917 && ( !$country || ! grep { $_ eq $country } @zip_reqd_countries )
2920 $self->setfield($field,'');
2922 $self->getfield($field) =~ /^\s*(\w[\w\-\s]{0,8}\w)\s*$/
2923 or return gettext('illegal_zip'). " $field: ". $self->getfield($field);
2924 $self->setfield($field,$1);
2932 =item ut_country COLUMN
2934 Check/untaint country codes. Country names are changed to codes, if possible -
2935 see L<Locale::Country>.
2940 my( $self, $field ) = @_;
2941 unless ( $self->getfield($field) =~ /^(\w\w)$/ ) {
2942 if ( $self->getfield($field) =~ /^([\w \,\.\(\)\']+)$/
2943 && country2code($1) ) {
2944 $self->setfield($field,uc(country2code($1)));
2947 $self->getfield($field) =~ /^(\w\w)$/
2948 or return "Illegal (country) $field: ". $self->getfield($field);
2949 $self->setfield($field,uc($1));
2953 =item ut_anything COLUMN
2955 Untaints arbitrary data. Be careful.
2960 my( $self, $field ) = @_;
2961 $self->getfield($field) =~ /^(.*)$/s
2962 or return "Illegal $field: ". $self->getfield($field);
2963 $self->setfield($field,$1);
2967 =item ut_enum COLUMN CHOICES_ARRAYREF
2969 Check/untaint a column, supplying all possible choices, like the "enum" type.
2974 my( $self, $field, $choices ) = @_;
2975 foreach my $choice ( @$choices ) {
2976 if ( $self->getfield($field) eq $choice ) {
2977 $self->setfield($field, $choice);
2981 return "Illegal (enum) field $field: ". $self->getfield($field);
2984 =item ut_enumn COLUMN CHOICES_ARRAYREF
2986 Like ut_enum, except the null value is also allowed.
2991 my( $self, $field, $choices ) = @_;
2992 $self->getfield($field)
2993 ? $self->ut_enum($field, $choices)
2997 =item ut_flag COLUMN
2999 Check/untaint a column if it contains either an empty string or 'Y'. This
3000 is the standard form for boolean flags in Freeside.
3005 my( $self, $field ) = @_;
3006 my $value = uc($self->getfield($field));
3007 if ( $value eq '' or $value eq 'Y' ) {
3008 $self->setfield($field, $value);
3011 return "Illegal (flag) field $field: $value";
3014 =item ut_foreign_key COLUMN FOREIGN_TABLE FOREIGN_COLUMN
3016 Check/untaint a foreign column key. Call a regular ut_ method (like ut_number)
3017 on the column first.
3021 sub ut_foreign_key {
3022 my( $self, $field, $table, $foreign ) = @_;
3023 return '' if $no_check_foreign;
3024 qsearchs($table, { $foreign => $self->getfield($field) })
3025 or return "Can't find ". $self->table. ".$field ". $self->getfield($field).
3026 " in $table.$foreign";
3030 =item ut_foreign_keyn COLUMN FOREIGN_TABLE FOREIGN_COLUMN
3032 Like ut_foreign_key, except the null value is also allowed.
3036 sub ut_foreign_keyn {
3037 my( $self, $field, $table, $foreign ) = @_;
3038 $self->getfield($field)
3039 ? $self->ut_foreign_key($field, $table, $foreign)
3043 =item ut_agentnum_acl COLUMN [ NULL_RIGHT | NULL_RIGHT_LISTREF ]
3045 Checks this column as an agentnum, taking into account the current users's
3046 ACLs. NULL_RIGHT or NULL_RIGHT_LISTREF, if specified, indicates the access
3047 right or rights allowing no agentnum.
3051 sub ut_agentnum_acl {
3052 my( $self, $field ) = (shift, shift);
3053 my $null_acl = scalar(@_) ? shift : [];
3054 $null_acl = [ $null_acl ] unless ref($null_acl);
3056 my $error = $self->ut_foreign_keyn($field, 'agent', 'agentnum');
3057 return "Illegal agentnum: $error" if $error;
3059 my $curuser = $FS::CurrentUser::CurrentUser;
3061 if ( $self->$field() ) {
3063 return "Access denied"
3064 unless $curuser->agentnum($self->$field());
3068 return "Access denied"
3069 unless grep $curuser->access_right($_), @$null_acl;
3077 =item fields [ TABLE ]
3079 This is a wrapper for real_fields. Code that called
3080 fields before should probably continue to call fields.
3085 my $something = shift;
3087 if($something->isa('FS::Record')) {
3088 $table = $something->table;
3090 $table = $something;
3091 $something = "FS::$table";
3093 return (real_fields($table));
3097 =item encrypt($value)
3099 Encrypts the credit card using a combination of PK to encrypt and uuencode to armour.
3101 Returns the encrypted string.
3103 You should generally not have to worry about calling this, as the system handles this for you.
3108 my ($self, $value) = @_;
3109 my $encrypted = $value;
3111 if ($conf->exists('encryption')) {
3112 if ($self->is_encrypted($value)) {
3113 # Return the original value if it isn't plaintext.
3114 $encrypted = $value;
3117 if (ref($rsa_encrypt) =~ /::RSA/) { # We Can Encrypt
3118 # RSA doesn't like the empty string so let's pack it up
3119 # The database doesn't like the RSA data so uuencode it
3120 my $length = length($value)+1;
3121 $encrypted = pack("u*",$rsa_encrypt->encrypt(pack("Z$length",$value)));
3123 die ("You can't encrypt w/o a valid RSA engine - Check your installation or disable encryption");
3130 =item is_encrypted($value)
3132 Checks to see if the string is encrypted and returns true or false (1/0) to indicate it's status.
3138 my ($self, $value) = @_;
3139 # could be more precise about it, but this will do for now
3140 $value =~ /^M/ && length($value) > 80;
3143 =item decrypt($value)
3145 Uses the private key to decrypt the string. Returns the decryoted string or undef on failure.
3147 You should generally not have to worry about calling this, as the system handles this for you.
3152 my ($self,$value) = @_;
3153 my $decrypted = $value; # Will return the original value if it isn't encrypted or can't be decrypted.
3154 if ($conf->exists('encryption') && $self->is_encrypted($value)) {
3156 if (ref($rsa_decrypt) =~ /::RSA/) {
3157 my $encrypted = unpack ("u*", $value);
3158 $decrypted = unpack("Z*", eval{$rsa_decrypt->decrypt($encrypted)});
3159 if ($@) {warn "Decryption Failed"};
3167 #Initialize the Module
3168 $rsa_module = 'Crypt::OpenSSL::RSA'; # The Default
3170 if ($conf->exists('encryptionmodule') && $conf->config('encryptionmodule') ne '') {
3171 $rsa_module = $conf->config('encryptionmodule');
3175 eval ("require $rsa_module"); # No need to import the namespace
3178 # Initialize Encryption
3179 if ($conf->exists('encryptionpublickey') && $conf->config('encryptionpublickey') ne '') {
3180 my $public_key = join("\n",$conf->config('encryptionpublickey'));
3181 $rsa_encrypt = $rsa_module->new_public_key($public_key);
3184 # Intitalize Decryption
3185 if ($conf->exists('encryptionprivatekey') && $conf->config('encryptionprivatekey') ne '') {
3186 my $private_key = join("\n",$conf->config('encryptionprivatekey'));
3187 $rsa_decrypt = $rsa_module->new_private_key($private_key);
3191 =item h_search ACTION
3193 Given an ACTION, either "insert", or "delete", returns the appropriate history
3194 record corresponding to this record, if any.
3199 my( $self, $action ) = @_;
3201 my $table = $self->table;
3204 my $primary_key = dbdef->table($table)->primary_key;
3207 'table' => "h_$table",
3208 'hashref' => { $primary_key => $self->$primary_key(),
3209 'history_action' => $action,
3217 Given an ACTION, either "insert", or "delete", returns the timestamp of the
3218 appropriate history record corresponding to this record, if any.
3223 my($self, $action) = @_;
3224 my $h = $self->h_search($action);
3225 $h ? $h->history_date : '';
3228 =item scalar_sql SQL [ PLACEHOLDER, ... ]
3230 A class or object method. Executes the sql statement represented by SQL and
3231 returns a scalar representing the result: the first column of the first row.
3233 Dies on bogus SQL. Returns an empty string if no row is returned.
3235 Typically used for statments which return a single value such as "SELECT
3236 COUNT(*) FROM table WHERE something" OR "SELECT column FROM table WHERE key = ?"
3241 my($self, $sql) = (shift, shift);
3242 my $sth = dbh->prepare($sql) or die dbh->errstr;
3244 or die "Unexpected error executing statement $sql: ". $sth->errstr;
3245 my $row = $sth->fetchrow_arrayref or return '';
3246 my $scalar = $row->[0];
3247 defined($scalar) ? $scalar : '';
3250 =item count [ WHERE ]
3252 Convenience method for the common case of "SELECT COUNT(*) FROM table",
3253 with optional WHERE. Must be called as method on a class with an
3259 my($self, $where) = (shift, shift);
3260 my $table = $self->table or die 'count called on object of class '.ref($self);
3261 my $sql = "SELECT COUNT(*) FROM $table";
3262 $sql .= " WHERE $where" if $where;
3263 $self->scalar_sql($sql);
3272 =item real_fields [ TABLE ]
3274 Returns a list of the real columns in the specified table. Called only by
3275 fields() and other subroutines elsewhere in FS::Record.
3282 my($table_obj) = dbdef->table($table);
3283 confess "Unknown table $table" unless $table_obj;
3284 $table_obj->columns;
3287 =item pvf FIELD_NAME
3289 Returns the FS::part_virtual_field object corresponding to a field in the
3290 record (specified by FIELD_NAME).
3295 my ($self, $name) = (shift, shift);
3297 if(grep /^$name$/, $self->virtual_fields) {
3299 my $concat = [ "'cf_'", "name" ];
3300 return qsearchs({ table => 'part_virtual_field',
3301 hashref => { dbtable => $self->table,
3304 select => 'vfieldpart, dbtable, length, label, '.concat_sql($concat).' as name',
3310 =item _quote VALUE, TABLE, COLUMN
3312 This is an internal function used to construct SQL statements. It returns
3313 VALUE DBI-quoted (see L<DBI/"quote">) unless VALUE is a number and the column
3314 type (see L<DBIx::DBSchema::Column>) does not end in `char' or `binary'.
3319 my($value, $table, $column) = @_;
3320 my $column_obj = dbdef->table($table)->column($column);
3321 my $column_type = $column_obj->type;
3322 my $nullable = $column_obj->null;
3324 warn " $table.$column: $value ($column_type".
3325 ( $nullable ? ' NULL' : ' NOT NULL' ).
3326 ")\n" if $DEBUG > 2;
3328 if ( $value eq '' && $nullable ) {
3330 } elsif ( $value eq '' && $column_type =~ /^(int|numeric)/ ) {
3331 cluck "WARNING: Attempting to set non-null integer $table.$column null; ".
3334 } elsif ( $value =~ /^\d+(\.\d+)?$/ &&
3335 ! $column_type =~ /(char|binary|text)$/i ) {
3337 } elsif (( $column_type =~ /^bytea$/i || $column_type =~ /(blob|varbinary)/i )
3338 && driver_name eq 'Pg'
3342 # dbh->quote($value, { pg_type => PG_BYTEA() }); # doesn't work right
3343 # Pg binary string quoting: convert each character to 3-digit octal prefixed with \\,
3344 # single-quote the whole mess, and put an "E" in front.
3345 return ("E'" . join('', map { sprintf('\\\\%03o', ord($_)) } split(//, $value) ) . "'");
3353 This is deprecated. Don't use it.
3355 It returns a hash-type list with the fields of this record's table set true.
3360 carp "warning: hfields is deprecated";
3363 foreach (fields($table)) {
3372 "$_: ". $self->getfield($_). "|"
3373 } (fields($self->table)) );
3376 sub DESTROY { return; }
3380 # #use Carp qw(cluck);
3381 # #cluck "DESTROYING $self";
3382 # warn "DESTROYING $self";
3386 # return ! eval { join('',@_), kill 0; 1; };
3389 =item str2time_sql [ DRIVER_NAME ]
3391 Returns a function to convert to unix time based on database type, such as
3392 "EXTRACT( EPOCH FROM" for Pg or "UNIX_TIMESTAMP(" for mysql. See
3393 the str2time_sql_closing method to return a closing string rather than just
3394 using a closing parenthesis as previously suggested.
3396 You can pass an optional driver name such as "Pg", "mysql" or
3397 $dbh->{Driver}->{Name} to return a function for that database instead of
3398 the current database.
3403 my $driver = shift || driver_name;
3405 return 'UNIX_TIMESTAMP(' if $driver =~ /^mysql/i;
3406 return 'EXTRACT( EPOCH FROM ' if $driver =~ /^Pg/i;
3408 warn "warning: unknown database type $driver; guessing how to convert ".
3409 "dates to UNIX timestamps";
3410 return 'EXTRACT(EPOCH FROM ';
3414 =item str2time_sql_closing [ DRIVER_NAME ]
3416 Returns the closing suffix of a function to convert to unix time based on
3417 database type, such as ")::integer" for Pg or ")" for mysql.
3419 You can pass an optional driver name such as "Pg", "mysql" or
3420 $dbh->{Driver}->{Name} to return a function for that database instead of
3421 the current database.
3425 sub str2time_sql_closing {
3426 my $driver = shift || driver_name;
3428 return ' )::INTEGER ' if $driver =~ /^Pg/i;
3432 =item regexp_sql [ DRIVER_NAME ]
3434 Returns the operator to do a regular expression comparison based on database
3435 type, such as '~' for Pg or 'REGEXP' for mysql.
3437 You can pass an optional driver name such as "Pg", "mysql" or
3438 $dbh->{Driver}->{Name} to return a function for that database instead of
3439 the current database.
3444 my $driver = shift || driver_name;
3446 return '~' if $driver =~ /^Pg/i;
3447 return 'REGEXP' if $driver =~ /^mysql/i;
3449 die "don't know how to use regular expressions in ". driver_name." databases";
3453 =item not_regexp_sql [ DRIVER_NAME ]
3455 Returns the operator to do a regular expression negation based on database
3456 type, such as '!~' for Pg or 'NOT REGEXP' for mysql.
3458 You can pass an optional driver name such as "Pg", "mysql" or
3459 $dbh->{Driver}->{Name} to return a function for that database instead of
3460 the current database.
3464 sub not_regexp_sql {
3465 my $driver = shift || driver_name;
3467 return '!~' if $driver =~ /^Pg/i;
3468 return 'NOT REGEXP' if $driver =~ /^mysql/i;
3470 die "don't know how to use regular expressions in ". driver_name." databases";
3474 =item concat_sql [ DRIVER_NAME ] ITEMS_ARRAYREF
3476 Returns the items concatenated based on database type, using "CONCAT()" for
3477 mysql and " || " for Pg and other databases.
3479 You can pass an optional driver name such as "Pg", "mysql" or
3480 $dbh->{Driver}->{Name} to return a function for that database instead of
3481 the current database.
3486 my $driver = ref($_[0]) ? driver_name : shift;
3489 if ( $driver =~ /^mysql/i ) {
3490 'CONCAT('. join(',', @$items). ')';
3492 join('||', @$items);
3497 =item midnight_sql DATE
3499 Returns an SQL expression to convert DATE (a unix timestamp) to midnight
3500 on that day in the system timezone, using the default driver name.
3505 my $driver = driver_name;
3507 if ( $driver =~ /^mysql/i ) {
3508 "UNIX_TIMESTAMP(DATE(FROM_UNIXTIME($expr)))";
3511 "EXTRACT( EPOCH FROM DATE(TO_TIMESTAMP($expr)) )";
3519 This module should probably be renamed, since much of the functionality is
3520 of general use. It is not completely unlike Adapter::DBI (see below).
3522 Exported qsearch and qsearchs should be deprecated in favor of method calls
3523 (against an FS::Record object like the old search and searchs that qsearch
3524 and qsearchs were on top of.)
3526 The whole fields / hfields mess should be removed.
3528 The various WHERE clauses should be subroutined.
3530 table string should be deprecated in favor of DBIx::DBSchema::Table.
3532 No doubt we could benefit from a Tied hash. Documenting how exists / defined
3533 true maps to the database (and WHERE clauses) would also help.
3535 The ut_ methods should ask the dbdef for a default length.
3537 ut_sqltype (like ut_varchar) should all be defined
3539 A fallback check method should be provided which uses the dbdef.
3541 The ut_money method assumes money has two decimal digits.
3543 The Pg money kludge in the new method only strips `$'.
3545 The ut_phonen method only checks US-style phone numbers.
3547 The _quote function should probably use ut_float instead of a regex.
3549 All the subroutines probably should be methods, here or elsewhere.
3551 Probably should borrow/use some dbdef methods where appropriate (like sub
3554 As of 1.14, DBI fetchall_hashref( {} ) doesn't set fetchrow_hashref NAME_lc,
3555 or allow it to be set. Working around it is ugly any way around - DBI should
3556 be fixed. (only affects RDBMS which return uppercase column names)
3558 ut_zip should take an optional country like ut_phone.
3562 L<DBIx::DBSchema>, L<FS::UID>, L<DBI>
3564 Adapter::DBI from Ch. 11 of Advanced Perl Programming by Sriram Srinivasan.