Ticket #2233: dvbradioexport.2.pl

#!/usr/bin/perl -w
# dvbradioexport.pl v2.0
# By Justin Hornsby 22 October 2006
# 
# A MythTV user job script for exporting DVB radio recordings to MP3 format
#
# Usage info will be output if the script is run with no arguments (or insufficient arguments)
#
# Contains elements of mythtv.pl by Nigel Pearson
# Initial idea from nuvexport
# 
# requirements: id3tag, ffmpeg with mp3 exporting enabled, PERL and the DBI & DBD::mysql modules
#

# PERL MODULES WE WILL BE USING
use DBI;
use DBD::mysql

#
$host     = '127.0.0.1';
$database = 'mythconverg';
$user     = 'mythtv';
$pass     = $user;

$hostname = 'mythback';
$exportdir = '/home/mythtv/';
$maxbitrate = '256';
$bitrate = '192';
$prefix = '/usr/bin';

$dir = '/video/videorec/';    # Where the recordings are

$db = undef;
$connect = undef;
$debug = 0;
######################################
#                                    #
#        SUBROUTINES ARE HERE        #
#                                    #
######################################


# Try to find the database
#
sub findDatabase()
{
    my $found = 0;
    my @mysql = ('/usr/local/share/mythtv/mysql.txt',
        '/usr/share/mythtv/mysql.txt',
        '/etc/mythtv/mysql.txt',
        '/usr/local/etc/mythtv/mysql.txt',
        "$ENV{HOME}/.mythtv/mysql.txt",
        'mysql.txt'
    );
    foreach my $file (@mysql) {
        next unless (-e $file);
        $found = 1;
        open(CONF, $file) or die "Unable to open $file:  $!\n\n";
        while (my $line = <CONF>) {
            # Cleanup
            next if ($line =~ /^\s*#/);
            $line =~ s/^str //;
            chomp($line);

            # Split off the var=val pairs
            my ($var, $val) = split(/\=/, $line, 2);
            next unless ($var && $var =~ /\w/);
            if ($var eq 'DBHostName') {
                $host = $val;
            } elsif ($var eq 'DBUserName') {
                $user = $val;
            } elsif ($var eq 'DBName') {
                $database = $val;
            } elsif ($var eq 'DBPassword') {
                $pass = $val;
            }
        }
        close CONF;
    }

if ( ! $found )
    {   die "Unable to locate mysql.txt:  $!\n\n"  }
}


#  DB connect subroutine
# =======================
# openDatabase() - Connect or die
#

sub openDatabase()
{
    $db = DBI->connect("dbi:mysql:database=$database" .
                               ":host=$host",
                               $user, $pass)
              or die "Cannot connect to database: $!\n\n";

    return $db;
}

##################################
#                                #
#    Main code starts here !!    #
#                                #
##################################
#
#
# FIND THE DATABASE SETTINGS

$usage = "\nHow to use dvbradioexport.pl \n\ndvbradioexport.pl exportdir=/foo/bar starttime=%STARTTIME% 
chanid=%CHANID maxbitrate=x debug\n"
        ."\n%CHANID% = channel ID associated with the recording to export\n"
        ."%STARTTIME% = recording start time in either 'yyyy-mm-dd hh:mm:ss' or 'yyyymmddhhmmss' format\n"
        ."exportdir = dir to export completed MP3 files to (note the user the script runs as must have write permission on that dir\n"
        ."maxbitrate = maximum bitrate for the export to use.  If more than the original file's bitrate, the original 
bitrate will be used \n"
        ."debug = enable debugging information - outputs which commands would be run etc\n";

findDatabase();

# get this script's ARGS
#

$num = $#ARGV + 1;

# if user hasn't passed enough arguments, die and print the usage info

if ($num le "2") {
              die "$usage";
            }

#
# Get all the arguments
#

foreach (@ARGV){
                if ($_ =~ m/debug/) {
                                  $debug = 1;
                                 }
                elsif ($_ =~ m/maxbitrate/) {
                                 $maxbitrate = (split(/\=/,$_))[1];
                                 }
                elsif ($_ =~ m/starttime/) {
                                  $starttime = (split(/\=/,$_))[1];
                                  }
                elsif ($_ =~ m/chanid/) {
                                  $chanid = (split(/\=/,$_))[1];
                                  }
                elsif ($_ =~ m/exportdir/) {
                                  $exportdir = (split(/\=/,$_))[1];
                                  }
    }

#
# connect to database
#
$connect = openDatabase(); 

# PREPARE THE QUERY
$query = "SELECT title, subtitle FROM recorded WHERE chanid=$chanid AND starttime='$starttime'";

$query_handle = $connect->prepare($query);


# EXECUTE THE QUERY
$query_handle->execute() || die "Cannot connect to database \n";

# BIND TABLE COLUMNS TO VARIABLES
$query_handle->bind_columns(undef, \$title, \$subtitle);

# LOOP THROUGH RESULTS
$query_handle->fetch();

# FIND WHERE THE RECORDINGS LIVE
$query = "SELECT data FROM settings WHERE value='RecordFilePrefix' AND hostname='$hostname'";

$query_handle = $connect->prepare($query);
$query_handle->execute()  || die "Unable to query settings table";

my ($dir) = $query_handle->fetchrow_array;

# FIND OUT THE CHANNEL NAME
$query = "SELECT name FROM channel WHERE chanid=$chanid";
$query_handle = $connect->prepare($query);
$query_handle->execute()  || die "Unable to query settings table";

my ($channame) = $query_handle->fetchrow_array;

# replace whitespace in channame with dashes
$channame =~ s/\s+/-/g;

# replace whitespace in title with underscores
$title =~ s/\W+/_/g;
# replace whitespace in subtitle with underscores
$subtitle =~ s/\W+/_/g;

#
# Remove non alphanumeric chars from $starttime & $endtime
#
 
$newstarttime = $starttime;

$newstarttime =~ s/[|^\W|\s|-|]//g;

$year = substr($newstarttime, 0, 4);

$filename = $dir."/".$chanid."_".$newstarttime.".mpg";

$newfilename = $exportdir."/".$channame."_".$title."_".$subtitle."_".$newstarttime.".mp3";

if ($debug) {
                          print "\n\n Source filename:$filename \nDestination filename:$newfilename\n \n";
                        }
#
# Now run ffmpeg to find out what bitrate the stream is
#

$output = `ffmpeg -i $filename 2>&1`;

@words = split(/Audio/, $output);
@br = split(/\,/, $words[1]);
@out = split(/kb/, $br[3]);

$origbitrate = $out[0];

#
# If maximum bitrate is less than the source bitrate, allow the max bitrate to remain
#

$bitrate = $maxbitrate;

if ($maxbitrate > $origbitrate) {
    print "maxbitrate is greater than original.  Using source bitrate to save space\n";
    $bitrate = $origbitrate;
    }

#
# Now run ffmpeg to get mp3 out of the dvb radio recording
#

$command = $prefix."/nice -n19 ffmpeg -i $filename -ab $bitrate -acodec mp3 -f mp3 '$newfilename' 2>&1";

if ($debug) {
                      print "\n\nUSING $command \n\n";
                    }

system "$command";

# Now tag the finished MP3 file

$command = "id3tag --song='$title' --album='$subtitle' --artist='$channame' --comment='Transcoded by MythTV' --year=$year 
'$newfilename'";

if ($debug) {
                           print "\n\nUsing $command \n\n";
                         }
system "$command";