Fixed a bug that didn't save scans with nothing in them

[ndwebbie.git] / scripts / scans.pl

diff --git a/scripts/scans.pl b/scripts/scans.pl
index 7eff622938190c8b66f6e5bac9a78f55f5ce04cd..e9795f7eba5e87e49b131481289fdc9671494fa4 100755 (executable)
--- a/scripts/scans.pl
+++ b/scripts/scans.pl
@@ -26,7 +26,7 @@ use CGI;
 use DBI;
 use DBD::Pg qw(:pg_types);
 use LWP::Simple;
-use lib qw{/var/www/ndawn/};
+use lib qw{/var/www/ndawn/lib/};
 use ND::DB;
 
 our $dbh = ND::DB::DB();
@@ -78,7 +78,11 @@ my $addpdata = $dbh->prepare(q{INSERT INTO planet_data (id,tick,scan,rid,amount)
 $dbh->begin_work or die 'No transaction';
 $newscans->execute or die $dbh->errstr;
 $dbh->pg_savepoint('scans') or die "No savepoint";
+
+my %production = (None => 0, Low => 35, Medium => 65, High => 100);
 while (my $scan = $newscans->fetchrow_hashref){
+       $dbh->pg_release('scans') or die "Couldn't save";
+       $dbh->pg_savepoint('scans') or die "Couldn't save";
        my $file = get("http://game.planetarion.com/showscan.pl?scan_id=$scan->{scan_id}");
        next unless defined $file;
        if ($file =~ /((?:\w| )*) (?:Scan|Probe) on (\d+):(\d+):(\d+) in tick (\d+)/){
@@ -115,14 +119,14 @@ while (my $scan = $newscans->fetchrow_hashref){
                                $addpdata->execute($planet,$tick,$scan->{id}
                                        ,'resource',$1, $res) or die $dbh->errstr;
                        }
-                       #if($file =~ m{"center">([^<]+)</td><td class="center">([^<]+)</td><td class="center">([^<]+)</td>}){
-                       #       $addpdata->execute($planet,$tick,$scan->{id}
-                       #               ,'planet','Light Usage', $1) or die $dbh->errstr;
-                       #       $addpdata->execute($planet,$tick,$scan->{id}
-                       #               ,'planet','Medium Usage', $2) or die $dbh->errstr;
-                       #       $addpdata->execute($planet,$tick,$scan->{id}
-                       #               ,'planet','Heavy Usage', $3) or die $dbh->errstr;
-                       #}
+                       if($file =~ m{<td class="center">([A-Z][a-z]+)</td><td class="center">([A-Z][a-z]+)</td><td class="center">([A-Z][a-z]+)</td>}){
+                               $addpdata->execute($planet,$tick,$scan->{id}
+                                       ,'planet','Light Usage', $production{$1}) or die $dbh->errstr;
+                               $addpdata->execute($planet,$tick,$scan->{id}
+                                       ,'planet','Medium Usage', $production{$2}) or die $dbh->errstr;
+                               $addpdata->execute($planet,$tick,$scan->{id}
+                                       ,'planet','Heavy Usage', $production{$3}) or die $dbh->errstr;
+                       }
                        if($file =~ m{<span class="superhighlight">([\d,]+)</span>}){
                                my $res = $1;
                                $res =~ s/,//g;
@@ -190,8 +194,6 @@ while (my $scan = $newscans->fetchrow_hashref){
                        print "Something wrong with scan $scan->{id} type $type at tick $tick http://game.planetarion.com/showscan.pl?scan_id=$scan->{scan_id}";
                }
                $parsedscan->execute($tick,$type,$planet,$scan->{id}) or die $dbh->errstr;
-               $dbh->pg_release('scans') or die "Couldn't save";
-               $dbh->pg_savepoint('scans') or die "Couldn't save";
                #$dbh->rollback;
                };
                if ($@) {