[BACK]Return to updatedb.pl CVS log [TXT] [DIR] Up to [Local Repository] / db / prgsrc

Diff for /db/prgsrc/updatedb.pl between versions 1.19 and 1.48

version 1.19, 2000/11/11 04:05:52 version 1.48, 2010/04/24 18:13:03
Line 6  updatedb.pl - a script for creation of n Line 6  updatedb.pl - a script for creation of n
   
 =head1 SYNOPSIS  =head1 SYNOPSIS
   
 updatedb.pl I<file1> I<file2>....  updatedb.pl B<[-i]> I<file1> I<file2>....
   
   
 =head1 DESCRIPTION  =head1 DESCRIPTION
   
 Updates information in the B<chgk> databse. Uses file  Updates information in the B<chgk> databse. Uses file
   
   =head1 OPTIONS
   
   =item B<-i> 
   
   Ask about ParentId. 
   
   
 =head1 BUGS  =head1 BUGS
   
Line 26  Dmitry Rubinstein Line 32  Dmitry Rubinstein
   
 =cut  =cut
   
   use vars qw($opt_i, $opt_n);
   
   use Getopt::Std;
   my $unsortedname="../dump/unsorted";
   getopts('in');
   #open STDERR, ">errors";
   my $Interactive=$opt_i || 0;
   my $newOnly = $opt_n ||0;
   my $DUMPDIR = $ENV{DUMPDIR} || "../dump";
   
 my (%RevMonths) =   my (%RevMonths) = 
     ('Jan', '1', 'Feb', '2', 'Mar', '3', 'Apr', '4', 'May', '5', 'Jun', '6',      ('Jan', '1', 'Feb', '2', 'Mar', '3', 'Apr', '4', 'May', '5', 'Jun', '6',
      'Jul', '7', 'Aug', '8', 'Sep', '9', 'Oct', '10', 'Nov', '11',       'Jul', '7', 'Aug', '8', 'Sep', '9', 'Oct', '10', 'Nov', '11',
Line 44  my ($sth); Line 60  my ($sth);
   
 use DBI;  use DBI;
 use strict;  use strict;
   use Data::Dumper;
   my $isunsorted=0;
 sub UpdateParents {  sub UpdateParents {
     my ($dbh, $ParentId, $all_qnum) = @_;      my ($dbh, $ParentId, $all_qnum,$CreatedAt) = @_;
     if ($ParentId) {      if ($ParentId) {
         my ($sth1) = $dbh->prepare("SELECT QuestionsNum, ParentId           my ($sth1) = $dbh->prepare("SELECT QuestionsNum, ParentId, CreatedAt
 FROM Tournaments WHERE Id = $ParentId");  FROM Tournaments WHERE Id = $ParentId");
         $sth1->execute;          $sth1->execute;
         my ($q, $p) = ($sth1->fetchrow)[0, 1];          my ($q, $p,$c) = ($sth1->fetchrow)[0, 1, 2];
           $c=$CreatedAt if $CreatedAt && ($CreatedAt gt $c);
           my $qc=$dbh->quote($c);
         $dbh->do("UPDATE Tournaments SET           $dbh->do("UPDATE Tournaments SET 
                   QuestionsNum=$q + $all_qnum                     QuestionsNum=$q + $all_qnum, CreatedAt=$qc
                   WHERE Id = $ParentId");                    WHERE Id = $ParentId");
         &UpdateParents($dbh, $p, $all_qnum);          &UpdateParents($dbh, $p, $all_qnum,$c);
     }      }
 }  }
   
   sub parseDate {
     my $value = shift;
     my ($from, $to) = split /\s+\-+\s+/, $value;
     $from =~ s/^(.*)-(.*)-(.*)$/$3-$2-$1/;
     my($month) = $RevMonths{$2} || '01';
     $from =~ s/-(.*)-/-$month-/;
     $from =~ s/-00*$/-01/;
     if ($to) {
       $to =~ s/^(.*)-(.*)-(.*)$/$3-$2-$1/;
       $month = $RevMonths{$2} || '01';    
       $to =~ s/-(.*)-/-$month-/;
       $to =~ s/-00*$/-01/;
     }
     return ($from, $to);
     
   }
   
 sub getField {  sub getField {
     my($desc, $dbh) = @_;      my($desc, $dbh) = @_;
     my($key);      my($key);
     my($value) = ('');      my($value) = ('');
     while (<$desc>) {      while (<$desc>) {
         s/[ ]//g;          s/[
   ]//g;
         if ($key && /^\s*$/) {          if ($key && /^\s*$/) {
             chomp $value;              chomp $value;
               $value =~ s/\s+$//;
             chomp $key;              chomp $key;
             if ($key eq 'Дата') {              if ($key eq 'Дата') {
                 $value =~ s/^(.*)-(.*)-(.*)$/$3-$2-$1/;                my ($from, $to) = parseDate($value);
                 my($month) = $RevMonths{$2} || '000';                $value = {'PlayedAt'=>$from, 'PlayedAt2'=>$to};
                 $value =~ s/$2/$month/;  
             }              }
             $value = $dbh->quote($value);              if ($key eq 'Автор') {$value=~s/\.$//;}
             return ($key, $value);              return ($key, $value);
         }          }
         next if (/^\s*$/);          next if (/^\s*$/);
Line 84  sub getField { Line 121  sub getField {
             next;              next;
         }          }
         if ($key) {          if ($key) {
             $value .= $_;              $value .= $_."\n";
             next;              next;
         }          }
     }      }
     if ($key && $value) {      if ($key && $value) {
         $value = $dbh->quote($value);          $value=~s/\s+$//sm;
         return ($key, $value);                return ($key, $value);
     }      }
     return (0, 0);      return (0, 0);
 }  }
Line 98  sub getField { Line 135  sub getField {
 sub SelectGroup {  sub SelectGroup {
     my ($dbh, $source, $TourName) = @_;      my ($dbh, $source, $TourName) = @_;
     my ($sth, $ParentId, $i, @arr);      my ($sth, $ParentId, $i, @arr);
       
     $sth = $dbh->prepare("SELECT Id, Title FROM      if ($Interactive) {    
           $sth = $dbh->prepare("SELECT Id, Title FROM
                 Tournaments WHERE Type = 'Г'");                  Tournaments WHERE Type = 'Г'");
     $sth->execute;          $sth->execute;
     print "Выберите группу для турнира:\n$TourName, файл $source\n\n";          print "Выберите группу для турнира:\n$TourName, файл $source\n\n";
     while (@arr=$sth->fetchrow) {          while (@arr=$sth->fetchrow) {
         print "[$arr[0]] $arr[1]\n";                  print "[$arr[0]] $arr[1]\n";
     }          }
     $ParentId = <STDIN>;          $ParentId = <STDIN>;
     chomp $ParentId;          chomp $ParentId;
     if (!$ParentId) {          if (!$ParentId) {
         print "Пропускаем файл $source\n";                  print "Пропускаем файл $source\n";
         print STDERR "Файл $source отвергнут оператором\n";                  print STDERR "Файл $source отвергнут оператором\n";
         return (0,0);                  return (0,0);
     } else {          } else {
         print "Вы выбрали турнир: $ParentId\n";                  print "Вы выбрали турнир: $ParentId\n";
         $sth = $dbh->prepare("INSERT INTO Tournaments                  $sth = $dbh->prepare("INSERT INTO Tournaments
                               (Title, Type, ParentId, FileName)                                 (Title, Type, ParentId, FileName) 
                                VALUES ($TourName, 'Ч', $ParentId,                                  VALUES ($TourName, 'Ч', $ParentId, 
                                        $source)");                                         $source)");
         $sth->execute;                  $sth->execute;
         my $TournamentId = $sth->{mysql_insertid};                  my $TournamentId = $sth->{mysql_insertid};
         return ($TournamentId,$ParentId);                  return ($TournamentId,$ParentId);
     }          }
       } else {
   # Теперь, если файла нет в дереве турниров, никаких вопросов не 
   # задаётся, а вместо этого он добавляется в группу 9999
                   $ParentId = 9999;
                   my $tempsource=$source;
                   my $temptname=$TourName;
                   $tempsource=~s/^\'(.*)\'$/$1/;
                   $temptname=~s/^\'(.*)\'$/$1/;
                   print UNSORTED "$tempsource".((12 -length($source))x' ')."\t$temptname\n";
                   $isunsorted=1;
                   $sth = $dbh->prepare("INSERT INTO Tournaments
                                 (Title, Type, ParentId, FileName,CreatedAt) 
                                  VALUES (".$dbh->quote($TourName).", 'Ч', $ParentId, 
                                          ?,NOW())");
                   $sth->execute($source);
                   my $TournamentId = $sth->{mysql_insertid};
                   return ($TournamentId,$ParentId);
           }
                                   
           
 }  }
   
 sub UpdateTournament {  sub UpdateTournament {
     my ($dbh, $TournamentId, $field, $value) = @_;      my ($dbh, $TournamentId, $field, $value) = @_;
     $dbh->do("UPDATE Tournaments SET $field=$value WHERE Id=$TournamentId")      if (ref $value eq 'HASH') {
         or die $dbh->errstr;              # It means that $value contains information about several fields
               foreach my $k(keys %$value) {
                 if ($value->{$k}) {
             &UpdateTournament($dbh, $TournamentId, $k, $value->{$k});
           }
         }
       } else {    
         my $v = $dbh->quote($value);
         $dbh->do("UPDATE Tournaments SET $field=$v WHERE Id=$TournamentId")
               or die $dbh->errstr;
       }
 }  }
   
 sub UpdateQuestion {  sub UpdateQuestion {
     my ($dbh, $QuestionId, $field, $value) = @_;      my ($dbh, $QuestionId, $field, $value) = @_;
     $dbh->do("UPDATE Questions SET $field=$value       
       if (($field eq 'Type') && ($value eq "Д")) {
            $value = "ЧД";
       }
       my $v = $dbh->quote($value);
       $dbh->do("UPDATE Questions SET $field=$v
                 WHERE QuestionId=$QuestionId")                  WHERE QuestionId=$QuestionId")
         or die $dbh->errstr;          or die $dbh->errstr;
 }  }
Line 142  sub UpdateQuestion { Line 213  sub UpdateQuestion {
 sub CheckFile {  sub CheckFile {
     my ($dbh, $source, $title) = @_;      my ($dbh, $source, $title) = @_;
     my $sth = $dbh->prepare("SELECT Id,ParentId,QuestionsNum FROM Tournaments      my $sth = $dbh->prepare("SELECT Id,ParentId,QuestionsNum FROM Tournaments
                              WHERE FileName=$source AND Type='Ч'");                               WHERE FileName=? AND Type='Ч'");
     $sth->execute or die $dbh->errstr;      $sth->execute($source) or die $dbh->errstr;
     my @arr = $sth->fetchrow;      my @arr = $sth->fetchrow;
     if (! scalar @arr) {      if (! scalar @arr) {
         return SelectGroup($dbh,$source,$title);          return SelectGroup($dbh,$source,$title);
     }      }
     my($Id,$ParentId,$QuestionsNum)=@arr;      my($Id,$ParentId,$QuestionsNum)=@arr;
     if($QuestionsNum) {       if($QuestionsNum) { 
           if ($newOnly) {
             return (0,0);
           }
         print "Файл $source с данными $title уже существует. ",          print "Файл $source с данными $title уже существует. ",
         "Заменить?[y/N]\n";          "Заменить?[y/N]\n";
         my $answer = <STDIN>;          my $answer = <STDIN>;
Line 199  sub GetTours { Line 273  sub GetTours {
 }  }
   
 sub CreateTour {  sub CreateTour {
     my ($dbh,$title,$ParentId,$TourNum,$rh_defaults)=@_;      my ($dbh,$title,$ParentId,$TourNum,$rh_defaults)=@_;   
     my $sth = $dbh->prepare("INSERT INTO Tournaments      my $sth = $dbh->prepare("INSERT INTO Tournaments
                              (Title, Type, ParentId, Number)                                (Title, Type, ParentId, Number,CreatedAt) 
                              VALUES ($title, 'Т', $ParentId, $TourNum)");                               VALUES (?, 'Т', $ParentId, $TourNum,NOW())");
     $sth->execute;      $sth->execute($title);
     my $TourId = $sth->{mysql_insertid};      my $TourId = $sth->{mysql_insertid};
     while (my ($key,$value)=each %$rh_defaults) {      while (my ($key,$value)=each %$rh_defaults) {
         &UpdateTournament($dbh, $TourId, $key, $value);              &UpdateTournament($dbh, $TourId, $key, $value);
     }      }
     return $TourId;      return $TourId;
 }  }
Line 222  MAIN: Line 296  MAIN:
                       'Инфо' => 'Info', 'URL' => 'URL',                        'Инфо' => 'Info', 'URL' => 'URL',
                       'Ссылка' => 'URL', 'Редактор' => 'Editors',                        'Ссылка' => 'URL', 'Редактор' => 'Editors',
                       'Обработан'=> 'EnteredBy',                        'Обработан'=> 'EnteredBy',
                       'Дата'=>'PlayedAt');                        'Дата'=>'PlayedAt', 'PlayedAt2'=>'PlayedAt2');
     #      #
     # Inherited fields for a Question      # Inherited fields for a Question
     #      #
     my %QuestionFields = ('Тип'=> 'Type', 'Вид'=> 'Type',       my %QuestionFields = ('Тип'=> 'Type', 'Вид'=> 'Type', 
                           'Автор' => 'Authors', 'Рейтинг'=>'Rating',                             'Автор' => 'Authors', 'Рейтинг'=>'Rating', 
                           'Источник' => 'Sources');                            'Источник' => 'Sources',
                             'Тема' => 'Topic');
                                                       
                                               
     my($source);      my($source);
           
     my($dbh) = DBI->connect("DBI:mysql:chgk", "piataev", "")      my($dbh) = DBI->connect("DBI:mysql:chgk", "piataev", "")
         or die "Can't connect to DB chgk\n";          or die "Can't connect to DB chgk\n";
       if ($dbh->get_info( 18 )=~/^(5|(4.1))/)  {$dbh->do("SET NAMES 'koi8r'");}
       my @sources;        
       open UNSORTED, ">$unsortedname";
     while ($source = shift) {      while ($source = shift) {
          push @sources,glob($source);
       }
       foreach $source(@sources) {
         my $TourNum=0;          my $TourNum=0;
         my($PlayedAt) = '';          my($PlayedAt) = '';
         my($QuestionId, $TourId, $TournamentId, $ParentId) = (0, 0, 0, 0);          my($QuestionId, $TourId, $TournamentId, $ParentId) = (0, 0, 0, 0);
         my($tournum, $qnum, $all_qnum) = (0, 0, 0);          my($tournum, $qnum, $all_qnum) = (0, 0, 0);
         my (@d) = (localtime((stat($source))[9]))[5,4,3];          my (@d) = (localtime((stat($source))[9]))[5,4,3];
         $d[1]++;          $d[1]++;
           $d[1]=sprintf("%02d",$d[1]);
           $d[2]=sprintf("%02d",$d[2]);
         $d[0]+=1900;          $d[0]+=1900;
         my ($CreatedAt) = $dbh->quote( join('-', @d));          my $UnquotedCreated=join('-', @d);
           my ($CreatedAt) = $UnquotedCreated;
   
         open INFD, $source           open INFD, $source 
             or die "Can't open input file: $!\n";              or die "Can't open input file: $!\n";
                   
         $source =~ s/^.*\/([^\/]*)$/$1/;          $source =~ s/^.*\/([^\/]*)$/$1/;
         $source = $dbh->quote($source);          my $unquotedsource=$source;
           $unquotedsource=~s/\.txt\s*$//;
         print STDERR "Файл: $source, дата: $CreatedAt ";          print STDERR "Файл: $source, дата: $CreatedAt ";
         my %TourDefaults=('CreatedAt'=>$CreatedAt);          my %TourDefaults=('CreatedAt'=>$CreatedAt);
         my %QuestionDefaults=();          my %QuestionDefaults=();
         my %QuestionGlobalDefaults=('Type'=>$dbh->quote('Ч'));          my %QuestionGlobalDefaults=('Type'=>'Ч');
         while (($key, $value) = getField(\*INFD, $dbh)) {          while (($key, $value) = getField(\*INFD, $dbh)) {
             last if (!$key);            
       last if (!$key);
                           
             if ($key =~ /Мета/) {      if ($key =~ /Мета/) {
                 next;   # This is obsolete                    next;   # This is obsolete
             }      }
             if ($key =~ /Чемпионат/ || $key =~ /Пакет/) {                     if ($key =~ /Чемпионат/ || $key =~ /Пакет/) {               
                 ($TournamentId, $ParentId) = CheckFile($dbh,$source,$value);                    ($TournamentId, $ParentId) = CheckFile($dbh,$source,$value);
                 if (!$TournamentId)  {                    if (!$TournamentId)  {
                     last;                      last;
                 }                           }     
                 $sth = $dbh->prepare("UPDATE Tournaments SET  
                                      Title=$value, Type='Ч',                     $sth = $dbh->prepare("UPDATE Tournaments SET
                                      ParentId=$ParentId,                                        Title=?, Type='Ч', 
                                      FileName=$source,                                        ParentId=?, 
                                      CreatedAt=$CreatedAt                                       FileName=?, 
                                        CreatedAt=?
                                      WHERE                                       WHERE
                                      Id=$TournamentId");                                       Id=?");
                 $sth->execute;                    $sth->execute($value,$ParentId,$source,$CreatedAt,$TournamentId);
                 next;                    next;
             }      }
             if ($key =~ /Тур/) {      if ($key =~ /Тур/) {      
                 if ($TourId) {                    if ($TourId) {
                     $dbh->do("UPDATE Tournaments SET QuestionsNum=$qnum                          $dbh->do("UPDATE Tournaments SET QuestionsNum=$qnum
                               WHERE Id=$TourId");                                WHERE Id=$TourId");
                 }                    }
                 $qnum = 0;                    $qnum = 0;
                 $TourNum++;                    $TourNum++;
                 $TourId=CreateTour($dbh,$value,$TournamentId,$TourNum,                    $TourDefaults{'FileName'}= "$unquotedsource.$TourNum";
                                                   
   
                     $TourId=CreateTour($dbh,$value,$TournamentId,$TourNum,
                                    \%TourDefaults);                                     \%TourDefaults);
                 %QuestionDefaults=%QuestionGlobalDefaults;                    %QuestionDefaults=%QuestionGlobalDefaults;
                 $QuestionId=0;                    $QuestionId=0;
                 next;                       next; 
             }      }
             if ($key =~ /Вопрос/) {      if ($key =~ /Вопрос/) {
                 if (!$TourId) {                    if (!$TourId) {
                     $qnum = 0;                      $qnum = 0;
                     $TourNum++;                      $TourNum++;     
                     $TourId=CreateTour($dbh,'1',$TournamentId,$TourNum,                      $TourId=CreateTour($dbh,'1',$TournamentId,$TourNum,
                                        \%TourDefaults);                                         \%TourDefaults);
                     %QuestionDefaults=%QuestionGlobalDefaults;          %QuestionDefaults=%QuestionGlobalDefaults;
                 }                  }
                 my $query = "INSERT INTO Questions                   my $query = "INSERT INTO Questions 
                              (ParentId, Number)                                (ParentId, Number) 
                              VALUES ($TourId, $qnum+1)";                               VALUES ($TourId, $qnum+1)";
                 $sth = $dbh->prepare($query);                  $sth = $dbh->prepare($query);
                 $sth->execute or print $query;;                  $sth->execute or print $query;;
                 $QuestionId = $sth->{mysql_insertid};                  $QuestionId = $sth->{mysql_insertid};
                 &UpdateQuestion($dbh, $QuestionId, "Question", $value);                  &UpdateQuestion($dbh, $QuestionId, "Question", $value);
                 while (my ($key,$value)=each %QuestionDefaults) {                  while (my ($key,$value)=each %QuestionDefaults) {
                     &UpdateQuestion($dbh, $QuestionId, $key, $value);                      &UpdateQuestion($dbh, $QuestionId, $key, $value);
                 }                                 }               
                 $qnum++;                  $qnum++;
                 $all_qnum++;                  $all_qnum++;
                 next;                  next;
             }      }
       if ($key =~ /Ответ/) {
             if ($key =~ /Ответ/) {        &UpdateQuestion($dbh, $QuestionId, "Answer", $value);
                 &UpdateQuestion($dbh, $QuestionId, "Answer", $value);                    next;
                 next;      }
             }  
       if ($key =~ /Зач[её]т/) {
             if ($key =~ /Комментари/) {                    &UpdateQuestion($dbh, $QuestionId, "PassCriteria", $value);
                 &UpdateQuestion($dbh, $QuestionId, "Comments", $value);                    next;
                 next;      }
             }  
                   if ($key =~ /Комментари/) {
             my @Fields = grep { $key =~ /$_/ } keys %QuestionFields;                  &UpdateQuestion($dbh, $QuestionId, "Comments", $value);
                   next;
             if (scalar @Fields) {      }
                 my $word = shift @Fields;     
                 my $field = $QuestionFields{$word};      my @Fields = grep { $key =~ /$_/ } keys %QuestionFields;
                 if ($QuestionId) {  
                     &UpdateQuestion($dbh, $QuestionId, $field, $value);      if (scalar @Fields) {
                 } elsif ($TourId) {                    my $word = shift @Fields;
                     my $field = $QuestionFields{$word};
                     if ($QuestionId) {
             &UpdateQuestion($dbh, $QuestionId, $field, $value);
                     } elsif ($TourId) {
                     $QuestionDefaults{$field}=$value;                      $QuestionDefaults{$field}=$value;
                 } else {                    } else {
                     $QuestionGlobalDefaults{$field}=$value;                      $QuestionGlobalDefaults{$field}=$value;
                 }                    }
                 next;                    next;
             }      }
   
             @Fields = grep { $key =~ /$_/ } keys %TourFields;      @Fields = grep { $key =~ /$_/ } keys %TourFields;
   
             if (scalar @Fields) {      if (scalar @Fields) {
                 my $word = shift @Fields;                  my $word = shift @Fields;
                 my $field = $TourFields{$word};                  my $field = $TourFields{$word};
                 if ($QuestionId) {                  my $updateId;
                   if ($QuestionId) {
                     print STDERR "ОШИБКА: $key $value недопустимы после",                      print STDERR "ОШИБКА: $key $value недопустимы после",
                     " начала вопросов\n";                      " начала вопросов\n";
                 } elsif ($TourId) {                  } else {
                     &UpdateTournament($dbh, $TourId, $field, $value);                    if ($TourId) {
                 } else {                      $updateId = $TourId;
                     &UpdateTournament($dbh, $TournamentId, $field, $value);          } else {
                     $TourDefaults{$field}=$value;            $updateId = $TournamentId;
                 }            $TourDefaults{$field}=$value;
                 next;          }
             }          &UpdateTournament($dbh, $updateId, $field, $value);
                   }
                   next;
       }
   
                           
             #              #
Line 365  MAIN: Line 462  MAIN:
                         WHERE Id=$TourId");                          WHERE Id=$TourId");
         $dbh->do("UPDATE Tournaments SET QuestionsNum=$all_qnum          $dbh->do("UPDATE Tournaments SET QuestionsNum=$all_qnum
                         WHERE Id=$TournamentId");                          WHERE Id=$TournamentId");
         &UpdateParents($dbh, $ParentId, $all_qnum);                       &UpdateParents($dbh, $ParentId, $all_qnum,$UnquotedCreated);            
         print STDERR "Всего вопросов: $all_qnum \n";          print STDERR "Всего вопросов: $all_qnum \n";
     }      }
       close UNSORTED;
       unlink $unsortedname unless $isunsorted;
     $dbh->disconnect;      $dbh->disconnect;
 }  }
Removed from v.1.19  
changed lines
  Added in v.1.48

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>