FS/bin/freeside-cdr-sql.import

   1 #!/usr/bin/perl
   2
   3 use strict;
   4 use vars qw( $DEBUG );
   5 use Date::Parse 'str2time';
   6 use Date::Format 'time2str';
   7 use FS::UID qw(adminsuidsetup dbh);
   8 use FS::cdr;
   9 use DBI;
  10 use Getopt::Std;
  11
  12 my %opt;
  13 getopts('e:H:U:P:D:T:', \%opt);
  14 my $user = shift or die &usage;
  15
  16 my $engine = $opt{e} || 'mysql';
  17
  18 my $dsn = "dbi:$engine";
  19 $dsn .= ":database=$opt{D}" if $opt{D};
  20 $dsn .= ":host=$opt{H}" if $opt{H};
  21
  22 my $dbi = DBI->connect($dsn, $opt{U}, $opt{P})
  23   or die $DBI::errstr;
  24
  25 adminsuidsetup $user;
  26
  27 my $fsdbh = FS::UID::dbh;
  28
  29 # check for existence of freesidestatus
  30 my $table = $opt{T} || 'cdr';
  31 my $status = $dbi->selectall_arrayref("SHOW COLUMNS FROM $table WHERE Field = 'freesidestatus'");
  32 if( ! @$status ) {
  33   print "Adding freesidestatus column...\n";
  34   $dbi->do("ALTER TABLE $table ADD COLUMN freesidestatus varchar(32)")
  35     or die $dbi->errstr;
  36 }
  37 else {
  38   print "freesidestatus column present\n";
  39 }
  40
  41 my @cols = ( qw(
  42 calldate clid src dst dcontext channel lastapp lastdata duration
  43     billsec disposition amaflags accountcode uniqueid userfield) );
  44 my $sql = 'SELECT '.join(',', @cols). " FROM $table WHERE freesidestatus IS NULL";
  45 my $sth = $sql->prepare($sql);
  46 $sth->execute;
  47 print "Importing ".$sth->rows." records...\n";
  48
  49 my $cdr_batch = new FS::cdr_batch({
  50     'cdrbatch' => 'sql-import-'. time2str('%Y/%m/%d-%T',time),
  51   });
  52 my $error = $cdr_batch->insert;
  53 die $error if $error;
  54 my $cdrbatchnum = $cdr_batch->cdrbatchnum;
  55 my $imports = 0;
  56 my $updates = 0;
  57
  58 my $row;
  59 while ( $row = $sth->fetchrow_hashref ) {
  60   my $cdr = FS::cdr->new($row);
  61   $cdr->startdate(str2time($cdr->calldate));
  62   $cdr->cdrbatchnum($cdrbatchnum);
  63   my $error = $cdr->insert;
  64   if($error) {
  65     print "failed import: $error\n";
  66   }
  67   else {
  68     $imports++;
  69     if( $dbi->do("UPDATE cdr SET freesidestatus = 'done'
  70         WHERE calldate = ? AND src = ? AND dst = ?",
  71                 undef,
  72                 $row->{'calldate'},
  73                 $row->{'src'},
  74                 $row->{'dst'},
  75
  76       ) ) {
  77         $updates++;
  78     }
  79     else {
  80       print "failed to set status: ".$dbi->errstr."\n";
  81     }
  82   }
  83 }
  84 print "Done.\nImported $imports CDRs, marked $updates CDRs as done.\n";
  85 $dbi->disconnect;
  86
  87 sub usage {
  88   "Usage: \n  freeside-cdr-sql.import\n\t-e mysql|Pg|... [ -H host ]\n\t-D database\n\t-U user\n\t-P password\n\tfreesideuser\n";
  89 }
  90