src/rtv_selection.py

   1 #!/usr/bin/python
   2
   3 import re, os
   4 import rtv_programmeinfo, rtv_utils, rtv_favourite
   5
   6 fn_re = re.compile(
   7         "^(\\S*)-(\\d{4}-\\d{2}-\\d{2}_\\d{2}_\\d{2})-(\S*)$" )
   8
   9 FILENAME_DATETIME_FORMAT = "%Y-%m-%d_%H_%M"
  10
  11 class Selection( object ):
  12
  13         def __init__( self ):
  14                 self.title = None
  15                 self.channel = None
  16                 self.startTime = None
  17                 self.deleteAfterDays = None
  18                 self.priority = 100
  19
  20         def load( self, filename ):
  21                 m = fn_re.match( filename )
  22                 if not m:
  23                         print( "Selection not recognised - filename '%s'" % filename
  24                                 + "is not of the recognised format." )
  25                 else:
  26                         self.title = m.group( 1 )
  27                         self.channel = m.group( 3 )
  28                         self.startTime = rtv_utils.parse_datetime( m.group( 2 ),
  29                                 FILENAME_DATETIME_FORMAT )
  30
  31         def matches( self, pi ):
  32
  33                 return ( self.title == rtv_utils.prepare_filename( pi.title ) and
  34                                  self.channel == pi.channel and
  35                                  self.startTime == pi.startTime )
  36
  37
  38
  39 def read_favs_and_selections( config, record_only = False ):
  40         ans = []
  41
  42         rtv_utils.ensure_dir_exists( config.favourites_dir )
  43
  44         for fn in os.listdir( config.favourites_dir ):
  45
  46                 if fn[-7:] == ".rtvfav":
  47                         full_fn = os.path.join( config.favourites_dir, fn )
  48
  49                         fav = rtv_favourite.Favourite()
  50                         fav.load( full_fn )
  51
  52                         if ( not record_only ) or ( fav.record == "yes" ):
  53                                 ans.append( fav )
  54
  55         rtv_utils.ensure_dir_exists( config.selections_dir )
  56
  57         for fn in os.listdir( config.selections_dir ):
  58
  59                 if fn[-7:] == ".rtvsel":
  60                         # Note: no need for full path, as the filename itself
  61                         # contains all the information we need, so we don't need
  62                         # to read the file at all.
  63
  64                         sel = Selection()
  65                         sel.load( fn[:-7] )
  66
  67                         ans.append( sel )
  68
  69         return ans
  70