src/rtv_convert.py

   1 #!/usr/bin/python
   2
   3 import os
   4 import os.path
   5 import re
   6 import shutil
   7 import sys
   8 import rtv_utils, rtv_programmeinfo
   9
  10 allowed_chars_re = re.compile( "[a-zA-z0-9_-]" )
  11
  12 def _sanitise_filename( fn, max_len ):
  13         fn = str( fn )
  14         san_fn = ""
  15         for ch in fn:
  16                 if allowed_chars_re.match( ch ):
  17                         san_fn += ch
  18                 else:
  19                         san_fn += "_"
  20         return san_fn[:max_len]
  21
  22 def _get_flv_file_path( rtvinfo, fallback_filename ):
  23         if not os.path.isfile( rtvinfo ):
  24                 print "Error: rtvinfo file '%s' not found." % rtvinfo
  25                 sys.stdout.flush()
  26                 return ( "", fallback_filename )
  27
  28         proginfo = rtv_programmeinfo.ProgrammeInfo()
  29         proginfo.load( rtvinfo )
  30
  31         startTime = str( proginfo.startTime )
  32
  33         if proginfo.sub_title:
  34                 unsan_file = startTime[:10] + "_" + str( proginfo.sub_title )
  35         elif proginfo.channel_pretty:
  36                 unsan_file = startTime + "_" + str( proginfo.channel_pretty )
  37         else:
  38                 unsan_file = startTime + "_" + str( proginfo.channel )
  39
  40         progname_dir = _sanitise_filename( proginfo.title, 40 )
  41
  42         if proginfo.destination and proginfo.destination != "":
  43                 ret_dir = os.path.join(
  44                         _sanitise_filename( proginfo.destination, 40 ),
  45                         progname_dir )
  46         else:
  47                 ret_dir = progname_dir
  48
  49         ans = ( ret_dir,
  50                 os.path.join( ret_dir, _sanitise_filename( unsan_file, 40 ) ) )
  51
  52         return ans
  53
  54 def _delete_empty_dirs( config ):
  55         for ( dirpath, dirnames, filenames ) in os.walk(
  56                         config.converted_progs_dir ):
  57                 if len( filenames ) == 0 and len( dirnames ) == 0:
  58
  59                         # Don't delete the "deleted" dir
  60                         if ( dirpath.endswith( "deleted" ) ):
  61                                 continue
  62
  63                         empty_dir_name = os.path.join(
  64                                 config.converted_progs_dir, dirpath )
  65                         print "Removing empty directory '%s'" % empty_dir_name
  66                         os.rmdir( empty_dir_name )
  67
  68
  69 def convert( config ):
  70
  71         for fn in os.listdir( config.recorded_progs_dir ):
  72                 if fn[-3:] != ".ts":
  73                         continue
  74
  75                 fn_stem = fn[ : fn.rfind( "." ) ]
  76
  77                 fn = os.path.join( config.recorded_progs_dir, fn )
  78
  79                 rtvinfo = os.path.join( config.recorded_progs_dir,
  80                         fn_stem + ".rtvinfo" )
  81
  82                 (flv_dir, flv_stem) = _get_flv_file_path( rtvinfo, fn_stem )
  83
  84                 rtv_utils.ensure_dir_exists(
  85                         os.path.join( config.converted_progs_dir, flv_dir ) )
  86
  87                 flv_before = os.path.join( config.converted_progs_dir,
  88                         flv_stem + "_before_flvtool2.flv" )
  89
  90                 flv_after = os.path.join( config.converted_progs_dir,
  91                         flv_stem + ".flv" )
  92
  93                 rtvinfo_new = flv_stem + ".rtvinfo"
  94
  95                 old_dir = os.path.join( config.recorded_progs_dir, "old" )
  96
  97                 sys.stdout.flush()
  98                 retval = os.system( config.convert_command % (
  99                         fn, flv_before, flv_after, old_dir ) )
 100
 101                 if retval != 0:
 102                         print "Conversion of '%s' returned an error." % fn
 103                         continue
 104
 105                 if os.path.isfile( rtvinfo ):
 106                         shutil.move( rtvinfo,
 107                                 os.path.join( config.converted_progs_dir, rtvinfo_new ) )
 108
 109         _delete_empty_dirs( config )
 110
 111
 112 def rename( config ):
 113         for ( dirpath, dirnames, filenames ) in os.walk( config.converted_progs_dir ):
 114                 for fn in filenames:
 115                         if fn[-4:] != ".flv":
 116                                 continue
 117
 118                         fn_stem = os.path.join( dirpath, fn[ : fn.rfind( "." ) ] )
 119
 120                         rtvinfo_orig = os.path.join( config.converted_progs_dir,
 121                                 fn_stem + ".rtvinfo" )
 122
 123                         flv_orig = os.path.join( config.converted_progs_dir,
 124                                 fn_stem + ".flv" )
 125
 126                         (flv_dir, flv_stem) = _get_flv_file_path( rtvinfo_orig, fn_stem )
 127
 128                         rtv_utils.ensure_dir_exists(
 129                                 os.path.join( config.converted_progs_dir, flv_dir ) )
 130
 131                         flv_after = os.path.join( config.converted_progs_dir,
 132                                 flv_stem + ".flv" )
 133
 134                         rtvinfo_after = os.path.join( config.converted_progs_dir,
 135                                 flv_stem + ".rtvinfo" )
 136
 137                         if flv_orig != flv_after:
 138                                 print "Renaming %s -> %s" % ( flv_orig, flv_after )
 139                                 shutil.move( flv_orig, flv_after )
 140
 141                         if rtvinfo_orig != rtvinfo_after:
 142                                 print "Renaming %s -> %s" % ( rtvinfo_orig, rtvinfo_after )
 143                                 shutil.move( rtvinfo_orig, rtvinfo_after )
 144
 145         _delete_empty_dirs( config )
 146