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

Diff for /db/prgsrc/updatedb.pl between versions 1.1 and 1.45

version 1.1, 2000/10/17 01:50:39 version 1.45, 2009/05/21 11:33:26
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);
   
   use Getopt::Std;
   my $unsortedname="../dump/unsorted";
   getopts('i');
   #open STDERR, ">errors";
   my $Interactive=$opt_i || 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',
         'Dec', '12',        'Dec', '12', 
          'Янв', '0', 'Фев', 1, 'Мар', 2, 'Апр', 3, 'Май', '4',       'JAN', '1', 'FEB', '2', 'MAR', '3', 'APR', '4', 'MAY', '5', 'JUN', '6',
          'Июн', '5', 'Июл', 6, 'Авг', '7', 'Сен', '8',        'JUL', '7', 'AUG', '8', 'SEP', '9', 'OCT', '10', 'NOV', '11',
          'Окт', '9', 'Ноя', '19', 'Дек', '11');       'DEC', '12', 
        'Янв', '1', 'Фев', '2', 'Мар', '3', 'Апр', '4', 'Май', '5',
        'Июн', '6', 'Июл', '7', 'Авг', '8', 'Сен', '9', 
        'Окт', '10', 'Ноя', '11', 'Дек', '12');
 my ($sth);  my ($sth);
   
   
   
   
   
 use DBI;  use DBI;
 use strict;  use strict;
   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("    my ($sth1) = $dbh->prepare("SELECT QuestionsNum, ParentId, CreatedAt
                         SELECT QuestionsNum, ParentId FROM Tournaments WHERE Id = $ParentId  FROM Tournaments WHERE Id = $ParentId");
                 ");    $sth1->execute;
                 $sth1->execute;    my ($q, $p,$c) = ($sth1->fetchrow)[0, 1, 2];
                 my ($q, $p) = ($sth1->fetchrow)[0, 1];    $c=$CreatedAt if $CreatedAt && ($CreatedAt gt $c);
                 $dbh->do("    my $qc=$dbh->quote($c);
                         UPDATE Tournaments SET QuestionsNum=$q + $all_qnum WHERE Id =    $dbh->do("UPDATE Tournaments SET 
                         $ParentId");                    QuestionsNum=$q + $all_qnum, CreatedAt=$qc
                 &UpdateParents($dbh, $p, $all_qnum);                    WHERE Id = $ParentId");
         }    &UpdateParents($dbh, $p, $all_qnum,$c);
       }
 }  }
   
   
 sub getField {  sub getField {
         my($desc, $dbh) = @_;      my($desc, $dbh) = @_;
         my($key);      my($key);
         my($value) = ('');      my($value) = ('');
         while (<$desc>) {      while (<$desc>) {
                 s/ //;    s/[ ]//g;
                 if ($key && /^\s*$/) {    if ($key && /^\s*$/) {
                         chomp $value;        chomp $value;
                         chomp $key;              $value =~ s/\s+$//;
                         if ($key eq 'Дата') {        chomp $key;
                                 $value =~ s/^(.*)-(.*)-(.*)$/$3-$2-$1/;        if ($key eq 'Дата') {
                                 my($month) = $RevMonths{$2};      $value =~ s/^(.*)-(.*)-(.*)$/$3-$2-$1/;
                                 $value =~ s/$2/$month/;      my($month) = $RevMonths{$2} || '01';
                         }      $value =~ s/-(.*)-/-$month-/;
                         $value = $dbh->quote($value);      $value =~ s/-00*$/-01/;
                         return ($key, $value);        }
                 }        if ($key eq 'Автор') {$value=~s/\.$//;}
                 next if (/^\s*$/);        $value = $dbh->quote($value);
         return ($key, $value);
                 if (/^(.*):\s*$/ && !$key) {    }
                         $key = $1;    next if (/^\s*$/);
                         next;    
                 }    if (!$key && /^(.*?)[:\.]\s*(.*)$/s) {
                 if ($key) {        $key = $1;
                         $value .= $_;        $value=$2;
                         next;        next;
                 }    }
         }    if ($key) {
         if ($key && $value) {        $value .= $_;
                 $value = $dbh->quote($value);        next;
                 return ($key, $value);    }
         }      }
         return (0, 0);      if ($key && $value) {
           $value=~s/\s+$//;
     $value = $dbh->quote($value);
     return ($key, $value);
       }
       return (0, 0);
 }  }
   
 sub SelectGroup {  sub SelectGroup {
         my ($dbh, $TourName, $sth, $ParentId, $i, @arr) = @_;      my ($dbh, $source, $TourName) = @_;
       my ($sth, $ParentId, $i, @arr);
   
         $sth = $dbh->prepare("SELECT Id, Title FROM      if ($Interactive) {    
                 Tournaments WHERE Type = 'Г'");        $sth = $dbh->prepare("SELECT Id, Title FROM
         $sth->execute;      Tournaments WHERE Type = 'Г'");
         print "Выберите группу для турнира:\n$TourName\n\n";        $sth->execute;
         for ($i = 0; $i < $sth->numrows; $i++) {        print "Выберите группу для турнира:\n$TourName, файл $source\n\n";
                 @arr = $sth->fetchrow;        while (@arr=$sth->fetchrow) {
                 print "[$arr[0]] $arr[1]\n";      print "[$arr[0]] $arr[1]\n";
         }        }
         $ParentId = <STDIN>;        $ParentId = <STDIN>;
         print "Вы выбрали турнир: $ParentId\n";        chomp $ParentId;
         return $ParentId;        if (!$ParentId) {
       print "Пропускаем файл $source\n";
       print STDERR "Файл $source отвергнут оператором\n";
       return (0,0);
         } else {
       print "Вы выбрали турнир: $ParentId\n";
       $sth = $dbh->prepare("INSERT INTO Tournaments
               (Title, Type, ParentId, FileName) 
                VALUES ($TourName, 'Ч', $ParentId, 
                                          $source)");
       $sth->execute;
       my $TournamentId = $sth->{mysql_insertid};
       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 ($TourName, 'Ч', $ParentId, 
                                          $source,NOW())");
                   $sth->execute;
       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")      $dbh->do("UPDATE Tournaments SET $field=$value WHERE Id=$TournamentId")
                 or die $dbh->errstr;    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       
                 WHERE QuestionId=$QuestionId")      if (($field eq 'Type') && ($value eq "'Д'")) {
                         or die $dbh->errstr;           $value = "'ЧД'";
       }
       $dbh->do("UPDATE Questions SET $field=$value 
       WHERE QuestionId=$QuestionId")
     or die $dbh->errstr;
 }  }
   
 MAIN:   sub CheckFile {
 {      my ($dbh, $source, $title) = @_;
         my($key, $value, $addition);      my $sth = $dbh->prepare("SELECT Id,ParentId,QuestionsNum FROM Tournaments
                                WHERE FileName=$source AND Type='Ч'");
       $sth->execute or die $dbh->errstr;
       my @arr = $sth->fetchrow;
       if (! scalar @arr) {
     return SelectGroup($dbh,$source,$title);
       }
       my($Id,$ParentId,$QuestionsNum)=@arr;
       if($QuestionsNum) { 
     print "Файл $source с данными $title уже существует. ",
     "Заменить?[y/N]\n";
     my $answer = <STDIN>;
     if ($answer !~ /^[yY]/) {
         return (0,0);
     } else {
         DeleteTournament($dbh,$Id,$ParentId,$QuestionsNum,0);
     }
       } 
       return($Id,$ParentId);  
   }
   
   
   sub DeleteTournament {
       my ($dbh,$Id,$ParentId,$QuestionsNum,$DeleteMyself) = @_;
       if ($QuestionsNum) {
     UpdateParents($dbh,$ParentId,-$QuestionsNum);
       }
       my (@Tours) = &GetTours($dbh, $Id);
       foreach my $Tour (@Tours) {
     DeleteTournament($dbh,$Tour,$Id,0,1);
       }
       my $sth = $dbh->prepare("DELETE FROM Questions
                                WHERE ParentId=$Id");
       $sth->execute or die $dbh->errstr;
       if($DeleteMyself) {
     $sth = $dbh->prepare("DELETE FROM Tournaments
                                WHERE Id=$Id");
     $sth->execute or die $dbh->errstr;
       }
   }
   
   sub GetTours {
     my ($dbh, $ParentId) = @_;
     my (@arr, @Tours);
   
     my ($sth) = $dbh->prepare("SELECT Id FROM Tournaments
       WHERE ParentId=$ParentId ORDER BY Id");
   
         my($source);    $sth->execute;
   
         my($dbh) = DBI->connect("DBI:mysql:chgk", "piataev", "")    while (@arr = $sth->fetchrow) {
                 or die "Can't connect to DB chgk\n";      push @Tours, $arr[0];
     }
   
         while ($source = shift) {    return @Tours;
                 my($PlayedAt) = '';  }
                 my($QuestionId, $TourId, $TournamentId, $ParentId) = (0, 0, 0, 0);  
                 my($tournum, $qnum, $all_qnum, $qtype) = (0, 0, 0, 'Ч');  sub CreateTour {
                 my (@d) = (localtime((stat($source))[9]))[5,4,3];      my ($dbh,$title,$ParentId,$TourNum,$rh_defaults)=@_;
                 $d[1]++;      my $sth = $dbh->prepare("INSERT INTO Tournaments
                 my ($CreatedAt) = $dbh->quote( join('-', @d));             (Title, Type, ParentId, Number,CreatedAt) 
                 print "File created on: $CreatedAt\n";             VALUES ($title, 'Т', $ParentId, $TourNum,NOW())");
               $sth->execute;
                 open INFD, $source       my $TourId = $sth->{mysql_insertid};
                         or die "Can't open input file: $!\n";      while (my ($key,$value)=each %$rh_defaults) {
             &UpdateTournament($dbh, $TourId, $key, $value);
                 $source =~ s/^.*\/([^\/]*)$/$1/;      }
                 $source = $dbh->quote($source);      return $TourId;
                 print "Processing file: $source \n";  }
       
                 while (($key, $value) = getField(\*INFD, $dbh)) {  
                         last if (!$key);  MAIN: 
           {
                         if ($key =~ /Мета/) {      my($key, $value, $addition);
                                 $value =~ s/[^\d]*//g;      #
                                 $sth = $dbh->prepare("SELECT Id FROM Tournaments WHERE      # Inherited fields for a Tour or Tournament
                                         MetaId=$value");      #
                                 $sth->execute       my %TourFields = ('Копирайт' => 'Copyright',
                                         or die "Invalid Meta field: $value";            'Инфо' => 'Info', 'URL' => 'URL',
                                 $ParentId = ($sth->fetchrow)[0];            'Ссылка' => 'URL', 'Редактор' => 'Editors',
                                 next;            'Обработан'=> 'EnteredBy',
                         }            'Дата'=>'PlayedAt');
                         if ($key =~ /Чемпионат/) {      #
                                 $ParentId = &SelectGroup($dbh, $value)      # Inherited fields for a Question
                                         unless ($ParentId);      #
                                 $sth = $dbh->prepare("INSERT INTO Tournaments      my %QuestionFields = ('Тип'=> 'Type', 'Вид'=> 'Type', 
                                         (Title, Type, ParentId, FileName, CreatedAt)           'Автор' => 'Authors', 'Рейтинг'=>'Rating', 
                                         VALUES ($value, 'Ч', $ParentId, $source,          'Источник' => 'Sources',
                                         $CreatedAt)");          'Тема' => 'Topic');
                                 $sth->execute;          
                                 $TournamentId = $sth->{insertid};            
                                 next;      my($source);
                         }      
                         if ($key =~ /Тур/) {      my($dbh) = DBI->connect("DBI:mysql:chgk", "piataev", "")
                                 if ($TourId) {    or die "Can't connect to DB chgk\n";
                                         $dbh->do("UPDATE Tournaments SET QuestionsNum=$qnum  if ($dbh->get_info( 18 )=~/^(5|(4.1))/)  {$dbh->do("SET NAMES 'koi8r'");}
                                                 WHERE Id=$TourId");      my @sources;  
                                 }      open UNSORTED, ">$unsortedname";
                                 $qnum = 0;      while ($source = shift) {
                                 $qtype = "'Ч'";         push @sources,glob($source);
                                 $sth = $dbh->prepare("INSERT INTO Tournaments      }
                                         (Title, Type, ParentId, CreatedAt)       foreach $source(@sources) {
                                         VALUES ($value, 'Т', $TournamentId, $CreatedAt)");    my $TourNum=0;
                                 $sth->execute;    my($PlayedAt) = '';
                                 $TourId = $sth->{insertid};    my($QuestionId, $TourId, $TournamentId, $ParentId) = (0, 0, 0, 0);
                                 next;    my($tournum, $qnum, $all_qnum) = (0, 0, 0);
                         }    my (@d) = (localtime((stat($source))[9]))[5,4,3];
                         if ($key =~ /Вид/) {    $d[1]++;
                                 $qtype = $value;    $d[1]=sprintf("%02d",$d[1]);
                                 next;    $d[2]=sprintf("%02d",$d[2]);
                         }    $d[0]+=1900;
                         if ($key =~ /Вопрос/) {    my $UnquotedCreated=join('-', @d);
                                 $sth = $dbh->prepare("INSERT INTO Questions     my ($CreatedAt) = $dbh->quote($UnquotedCreated);
                                         (ParentId, Number, Type)   
                                         VALUES ($TourId, $qnum+1, $qtype)");    open INFD, $source 
                                 $sth->execute;        or die "Can't open input file: $!\n";
                                 $QuestionId = $sth->{insertid};    
                                 &UpdateQuestion($dbh, $QuestionId, "Question", $value);    $source =~ s/^.*\/([^\/]*)$/$1/;
                                 $qnum++;    my $unquotedsource=$source;
                                 $all_qnum++;    $unquotedsource=~s/\.txt\s*$//;
                                 next;    $source = $dbh->quote($source);
                         }    print STDERR "Файл: $source, дата: $CreatedAt ";
                         &UpdateQuestion($dbh, $QuestionId, "Answer", $value)     my %TourDefaults=('CreatedAt'=>$CreatedAt);
                                 if ($key =~ /Ответ/);    my %QuestionDefaults=();
             my %QuestionGlobalDefaults=('Type'=>$dbh->quote('Ч'));
                         &UpdateQuestion($dbh, $QuestionId, "Authors", $value)     while (($key, $value) = getField(\*INFD, $dbh)) {
                                 if ($key =~ /Автор/);        last if (!$key);
                 
                         &UpdateQuestion($dbh, $QuestionId, "Sources", $value)         if ($key =~ /Мета/) {
                                 if ($key =~ /Источник/);      next;   # This is obsolete
                 }
                         &UpdateQuestion($dbh, $QuestionId, "Comments", $value)         if ($key =~ /Чемпионат/ || $key =~ /Пакет/) {   
                                 if ($key =~ /Комментарий/);      ($TournamentId, $ParentId) = CheckFile($dbh,$source,$value);
               if (!$TournamentId)  {
                         &UpdateTournament($dbh, $TournamentId, "URL", $value)          last;
                                 if ($key =~ /URL/);      } 
       $sth = $dbh->prepare("UPDATE Tournaments SET
                         &UpdateTournament($dbh, $TournamentId, "Copyright", $value)                     Title=$value, Type='Ч', 
                                 if ($key =~ /Копирайт/);                                       ParentId=$ParentId, 
                                        FileName=$source, 
                         &UpdateTournament($dbh, $TournamentId, "Info", $value)                                       CreatedAt=$CreatedAt
                                 if ($key =~ /Инфо/);                                       WHERE
                                        Id=$TournamentId");
                         if ($key =~ /Дата/) {      $sth->execute;
                                 if ($TourId) {      next;
                                         &UpdateTournament($dbh, $TourId, "PlayedAt", $value);        }
                                 } else {        if ($key =~ /Тур/) {
                                         &UpdateTournament($dbh, $TournamentId, "PlayedAt", $value);      if ($TourId) {
                                 }  
                         }            $dbh->do("UPDATE Tournaments SET QuestionsNum=$qnum
                 }              WHERE Id=$TourId");
                 $dbh->do("UPDATE Tournaments SET QuestionsNum=$qnum      }
                         WHERE Id=$TourId");      $qnum = 0;
                 $dbh->do("UPDATE Tournaments SET QuestionsNum=$all_qnum      $TourNum++;
                         WHERE Id=$TournamentId");      $TourDefaults{'FileName'}=$dbh->quote(
                 &UpdateParents($dbh, $ParentId, $all_qnum);                                             "$unquotedsource.$TourNum"
         }              );
         $dbh->disconnect;      $TourId=CreateTour($dbh,$value,$TournamentId,$TourNum,
              \%TourDefaults);
       %QuestionDefaults=%QuestionGlobalDefaults;
       $QuestionId=0;
       next; 
         }
         if ($key =~ /Вопрос/) {
       if (!$TourId) {
           $qnum = 0;
           $TourNum++;
           $TourId=CreateTour($dbh,'1',$TournamentId,$TourNum,
                  \%TourDefaults);
           %QuestionDefaults=%QuestionGlobalDefaults;
       }
       my $query = "INSERT INTO Questions 
              (ParentId, Number) 
              VALUES ($TourId, $qnum+1)";
       $sth = $dbh->prepare($query);
       $sth->execute or print $query;;
       $QuestionId = $sth->{mysql_insertid};
       &UpdateQuestion($dbh, $QuestionId, "Question", $value);
       while (my ($key,$value)=each %QuestionDefaults) {
           &UpdateQuestion($dbh, $QuestionId, $key, $value);
       }   
       $qnum++;
       $all_qnum++;
       next;
         }
   
         if ($key =~ /Ответ/) {
       &UpdateQuestion($dbh, $QuestionId, "Answer", $value);
       next;
         }
   
   
         if ($key =~ /Зач[её]т/) {
       &UpdateQuestion($dbh, $QuestionId, "PassCriteria", $value);
       next;
         }
   
   
   
         if ($key =~ /Комментари/) {
       &UpdateQuestion($dbh, $QuestionId, "Comments", $value);
       next;
         }
   
   
         
         my @Fields = grep { $key =~ /$_/ } keys %QuestionFields;
   
         if (scalar @Fields) {
       my $word = shift @Fields;
       my $field = $QuestionFields{$word};
       if ($QuestionId) {
           &UpdateQuestion($dbh, $QuestionId, $field, $value);
       } elsif ($TourId) {
           $QuestionDefaults{$field}=$value;
       } else {
           $QuestionGlobalDefaults{$field}=$value;
       }
       next;
         }
   
         @Fields = grep { $key =~ /$_/ } keys %TourFields;
   
         if (scalar @Fields) {
       my $word = shift @Fields;
       my $field = $TourFields{$word};
       if ($QuestionId) {
           print STDERR "ОШИБКА: $key $value недопустимы после",
           " начала вопросов\n";
       } elsif ($TourId) {
           &UpdateTournament($dbh, $TourId, $field, $value);
       } else {
           &UpdateTournament($dbh, $TournamentId, $field, $value);
           $TourDefaults{$field}=$value;
       }
       next;
         }
   
         
         #
         # If we are here, something got wrong!
         #
         print STDERR "\nЯ НЕ ПОНИМАЮ: $key, $value!\n";
         
     }
     $dbh->do("UPDATE Tournaments SET QuestionsNum=$qnum
         WHERE Id=$TourId");
     $dbh->do("UPDATE Tournaments SET QuestionsNum=$all_qnum
         WHERE Id=$TournamentId");
     &UpdateParents($dbh, $ParentId, $all_qnum,$UnquotedCreated);    
     print STDERR "Всего вопросов: $all_qnum \n";
       }
       close UNSORTED;
       unlink $unsortedname unless $isunsorted;
       $dbh->disconnect;
 }  }
Removed from v.1.1  
changed lines
  Added in v.1.45

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