src/rtv_download.py

   1 #!/usr/bin/python
   2
   3 import os, time
   4 import rtv_utils
   5
   6 def delete_old_xmltv_files( config ):
   7
   8         for fn in os.listdir( config.xmltv_files_dir ):
   9                 if fn[-6:] == ".xmltv":
  10
  11                         full_fn = os.path.join( config.xmltv_files_dir, fn )
  12
  13                         #print "Deleting file '%s'" % full_fn
  14
  15                         os.unlink( full_fn )
  16
  17 def download( config ):
  18         rtv_utils.ensure_dir_exists( config.xmltv_configs_dir )
  19         rtv_utils.ensure_dir_exists( config.xmltv_files_dir )
  20
  21         # NOTE: alternatively we could move them to a different dir?
  22         delete_old_xmltv_files( config )
  23
  24         if not os.path.isfile( config.xmltv_config_file ):
  25
  26                 if config.options.interactive:
  27                         rtv_utils.run_command( config.xmltv_config_command )
  28
  29                         if not os.path.isfile( config.xmltv_config_file ):
  30                                 raise Exception( "Tried to configure XMLTV, but the "
  31                                         + "config file '" + config.xmltv_config_file
  32                                         + "' still does not exist." )
  33                 else:
  34                         raise Exception( "The XMLTV config file '%s' does not exist!"
  35                                 % config.xmltv_config_file )
  36
  37         date_filename = time.strftime( "tv-%Y-%m-%d.xmltv", time.localtime() )
  38         rtv_utils.run_command_write_output( config.xmltv_download_command,
  39                 os.path.join( config.xmltv_files_dir, date_filename ) )
  40
  41
  42
  43
  44
  45