diff options
Diffstat (limited to 'rhimport-btl')
-rwxr-xr-x | rhimport-btl | 20 |
1 files changed, 9 insertions, 11 deletions
diff --git a/rhimport-btl b/rhimport-btl index e212793..939d763 100755 --- a/rhimport-btl +++ b/rhimport-btl @@ -22,7 +22,6 @@ # use strict; -use DBI; use Date::Calc; use XML::Feed; use XML::Feed::Entry; @@ -37,11 +36,6 @@ use LWP::Simple; use lib '/usr/local/share/rhimport/'; use rhimport; -my $DBHOST = "airplay"; -my $DBUSER = "rivendell"; -my $DBPW = "lldriven"; -my $DB = "rivendell"; - my $STAT_FILE = $ENV{'HOME'} . "/rhimport-btl.stat"; my $RSS_URL = "http://www.btlonline.org/rss/btl128.xml"; my $PV_ID = '221'; @@ -58,12 +52,16 @@ my $user = `/usr/bin/id -un`; $user =~ s/\n//; my $group = "betweenlin"; -my $dbh = DBI->connect("DBI:mysql:$DB:$DBHOST","$DBUSER","$DBPW") or die "Database Error: $DBI::errstr"; +my ($dbh, $errorstring) = rhimport::opendb(); +if(!defined $dbh) { + print "$errorstring\n"; + exit 1; +} my @allowed_dbs = rhimport::get_dropboxes($dbh, $user, $group); if(scalar(@allowed_dbs) != 1) { print "found more or less than one Dropbox for this group?!\n"; - $dbh->disconnect(); + rhimport::closedb($dbh); exit 1; } my $dropbox = $allowed_dbs[0]->{'PATH'}; @@ -108,13 +106,13 @@ for my $entry ($feed->entries) { my $current_file = `cat $STAT_FILE`; if($current_file eq $ff->output_file) { print "Already downloaded file of today\n"; - $dbh->disconnect(); + rhimport::closedb($dbh); exit 0; } $out_file = $ff->output_file; if(!rhimport::check_file_extension($out_file)) { print "\n\nThe extension of the matching file '". $out_file . "' seems to be wrong - manual import necessary!!!\n"; - $dbh->disconnect(); + rhimport::closedb($dbh); exit 1; } @@ -156,7 +154,7 @@ my $import_log; ($ret, $import_log) = rhimport::import_single($file, $dropbox, $user, 0, $error_cb); $log .= $import_log; -$dbh->disconnect(); +rhimport::closedb($dbh); unlink $file; |