config.py

   1 import ConfigParser, os
   2 import re
   3 from ConfigParser import NoOptionError
   4
   5 BLACKLIST_169 = ('540', '649')
   6
   7 config = ConfigParser.ConfigParser()
   8 p = os.path.dirname(__file__)
   9 config.read(os.path.join(p, 'pyTivo.conf'))
  10
  11 def getGUID():
  12     if config.has_option('Server', 'GUID'):
  13         guid = config.get('Server', 'GUID')
  14     else:
  15         guid = '123456'
  16     return guid
  17
  18 def getBeaconAddresses():
  19     if config.has_option('Server', 'beacon'):
  20         beacon_ips = config.get('Server', 'beacon')
  21     else:
  22         beacon_ips = '255.255.255.255'
  23     return beacon_ips
  24
  25 def getPort():
  26     return config.get('Server', 'Port')
  27
  28 def get169Setting(tsn):
  29     if not tsn:
  30         return True
  31
  32     if config.has_section('_tivo_' + tsn):
  33         if config.has_option('_tivo_' + tsn, 'aspect169'):
  34             if config.get('_tivo_' + tsn, 'aspect169').lower() == 'true':
  35                 return True
  36             else:
  37                 return False
  38
  39     if tsn[:3] in BLACKLIST_169:
  40         return False
  41
  42     return True
  43
  44 def getShares():
  45     shares = [ (section, dict(config.items(section))) for section in config.sections() if not(section.startswith('_tivo_') or section == 'Server') ]
  46
  47     for name, data in shares:
  48         if not data.get('auto_subshares', 'False').lower() == 'true':
  49             continue
  50
  51         base_path = data['path']
  52         for item in os.listdir(base_path):
  53             item_path = os.path.join(base_path, item)
  54             if not os.path.isdir(item_path):
  55                 continue
  56
  57             new_name = name + '/' + item
  58             new_data = dict(data)
  59             new_data['path'] = item_path
  60
  61             shares.append( (new_name, new_data) )
  62
  63     return shares
  64
  65
  66 def getDebug():
  67     try:
  68         debug = config.get('Server', 'debug')
  69         if debug.lower() == 'true':
  70             return True
  71         else:
  72             return False
  73     except NoOptionError:
  74         return False
  75
  76 def getHack83():
  77     try:
  78         debug = config.get('Server', 'hack83')
  79         if debug.lower() == 'true':
  80             return True
  81         else:
  82             return False
  83     except NoOptionError:
  84         return False
  85
  86 def getOptres():
  87     try:
  88         optres = config.get('Server', 'optres')
  89         if optres.lower() == 'true':
  90             return True
  91         else:
  92             return False
  93     except NoOptionError:
  94         return False
  95
  96 def get(section, key):
  97     return config.get(section, key)
  98
  99 def getFFMPEGTemplate(tsn):
 100     if tsn and config.has_section('_tivo_' + tsn):
 101         try:
 102             return config.get('_tivo_' + tsn, 'ffmpeg_prams', raw = True)
 103         except NoOptionError:
 104             pass
 105
 106     try:
 107         return config.get('Server', 'ffmpeg_prams', raw = True)
 108     except NoOptionError: #default
 109         return '-vcodec mpeg2video -r 29.97 -b %(video_br)s -maxrate %(max_video_br)s -bufsize %(buff_size)s %(aspect_ratio)s -comment pyTivo.py -ac 2 -ab %(audio_br)s -ar 44100 -f vob -'
 110
 111 def getValidWidths():
 112     return [1920, 1440, 1280, 720, 704, 544, 480, 352]
 113
 114 def getValidHeights():
 115     return [1080, 720, 480] # Technically 240 is also supported
 116
 117 # Return the number in list that is nearest to x
 118 # if two values are equidistant, return the larger
 119 def nearest(x, list):
 120     return reduce(lambda a, b: closest(x,a,b), list)
 121
 122 def closest(x,a, b):
 123     if abs(x-a) < abs(x-b) or (abs(x-a) == abs(x-b)and a>b):
 124         return a
 125     else:
 126         return b
 127
 128 def nearestTivoHeight(height):
 129     return nearest(height, getValidHeights())
 130
 131 def nearestTivoWidth(width):
 132     return nearest(width, getValidWidths())
 133
 134 def getTivoHeight(tsn):
 135     if tsn and config.has_section('_tivo_' + tsn):
 136         try:
 137             height = int(config.get('_tivo_' + tsn, 'height'))
 138             return nearestTivoHeight(height)
 139         except NoOptionError:
 140             pass
 141
 142     try:
 143         height = int(config.get('Server', 'height'))
 144         return nearestTivoHeight(height)
 145     except NoOptionError: #default
 146         return 480
 147
 148 def getTivoWidth(tsn):
 149     if tsn and config.has_section('_tivo_' + tsn):
 150         try:
 151             width = int(config.get('_tivo_' + tsn, 'width'))
 152             return nearestTivoWidth(width)
 153         except NoOptionError:
 154             pass
 155
 156     try:
 157         width = int(config.get('Server', 'width'))
 158         return nearestTivoWidth(width)
 159     except NoOptionError: #default
 160         return 544
 161
 162 def getAudioBR(tsn = None):
 163     if tsn and config.has_section('_tivo_' + tsn):
 164         try:
 165             return config.get('_tivo_' + tsn, 'audio_br')
 166         except NoOptionError:
 167             pass
 168
 169     try:
 170         return config.get('Server', 'audio_br')
 171     except NoOptionError: #default to 192
 172         return '192K'
 173
 174 def getVideoBR(tsn = None):
 175     if tsn and config.has_section('_tivo_' + tsn):
 176         try:
 177             return config.get('_tivo_' + tsn, 'video_br')
 178         except NoOptionError:
 179             pass
 180
 181     try:
 182         return config.get('Server', 'video_br')
 183     except NoOptionError: #default to 4096K
 184         return '4096K'
 185
 186 def getMaxVideoBR():
 187     try:
 188         return config.get('Server', 'max_video_br')
 189     except NoOptionError: #default to 17M
 190         return '17408k'
 191
 192 def getBuffSize():
 193     try:
 194         return config.get('Server', 'bufsize')
 195     except NoOptionError: #default 1024k
 196         return '1024k'