10 from ConfigParser
import NoOptionError
13 def getboolean(self
, x
):
14 return self
.get(x
, 'False').lower() in ('1', 'yes', 'true', 'on')
26 p
= os
.path
.dirname(__file__
)
27 config_files
= ['/etc/pyTivo.conf', os
.path
.join(p
, 'pyTivo.conf')]
30 opts
, _
= getopt
.getopt(argv
, 'c:e:', ['config=', 'extraconf='])
31 except getopt
.GetoptError
, msg
:
34 for opt
, value
in opts
:
35 if opt
in ('-c', '--config'):
36 config_files
= [value
]
37 elif opt
in ('-e', '--extraconf'):
38 config_files
.append(value
)
50 config
= ConfigParser
.ConfigParser()
51 configs_found
= config
.read(config_files
)
53 print ('WARNING: pyTivo.conf does not exist.\n' +
54 'Assuming default values.')
55 configs_found
= config_files
[-1:]
57 for section
in config
.sections():
58 if section
.startswith('_tivo_'):
60 if tsn
.upper() not in ['SD', 'HD', '4K']:
62 tivos
[tsn
] = Bdict(config
.items(section
))
64 for section
in ['Server', '_tivo_SD', '_tivo_HD', '_tivo_4K']:
65 if not config
.has_section(section
):
66 config
.add_section(section
)
69 f
= open(configs_found
[-1], 'w')
73 def tivos_by_ip(tivoIP
):
74 for key
, value
in tivos
.items():
75 if value
['address'] == tivoIP
:
78 def get_server(name
, default
=None):
79 if config
.has_option('Server', name
):
80 return config
.get('Server', name
)
90 dest_ip
= tivos
[tsn
]['address']
93 s
= socket
.socket(socket
.AF_INET
, socket
.SOCK_DGRAM
)
94 s
.connect((dest_ip
, 123))
95 return s
.getsockname()[0]
98 opt
= get_server('zeroconf', 'auto').lower()
101 for section
in config
.sections():
102 if section
.startswith('_tivo_'):
103 if config
.has_option(section
, 'shares'):
104 logger
= logging
.getLogger('pyTivo.config')
105 logger
.info('Shares security in use -- zeroconf disabled')
107 elif opt
in ['false', 'no', 'off']:
112 def getBeaconAddresses():
113 return get_server('beacon', '255.255.255.255')
116 return get_server('port', '9032')
118 def get169Blacklist(tsn
): # tivo does not pad 16:9 video
119 return tsn
and not isHDtivo(tsn
) and not get169Letterbox(tsn
)
120 # verified Blacklist Tivo's are ('130', '240', '540')
121 # It is assumed all remaining non-HD and non-Letterbox tivos are Blacklist
123 def get169Letterbox(tsn
): # tivo pads 16:9 video for 4:3 display
124 return tsn
and tsn
[:3] in ['649']
126 def get169Setting(tsn
):
130 tsnsect
= '_tivo_' + tsn
131 if config
.has_section(tsnsect
):
132 if config
.has_option(tsnsect
, 'aspect169'):
134 return config
.getboolean(tsnsect
, 'aspect169')
138 if get169Blacklist(tsn
) or get169Letterbox(tsn
):
143 def getAllowedClients():
144 return get_server('allowedips', '').split()
146 def getIsExternal(tsn
):
147 tsnsect
= '_tivo_' + tsn
148 if tsnsect
in config
.sections():
149 if config
.has_option(tsnsect
, 'external'):
151 return config
.getboolean(tsnsect
, 'external')
157 def isTsnInConfig(tsn
):
158 return ('_tivo_' + tsn
) in config
.sections()
160 def getShares(tsn
=''):
161 shares
= [(section
, Bdict(config
.items(section
)))
162 for section
in config
.sections()
163 if not (section
.startswith(('_tivo_', 'logger_', 'handler_',
165 or section
in ('Server', 'loggers', 'handlers',
170 tsnsect
= '_tivo_' + tsn
171 if config
.has_section(tsnsect
) and config
.has_option(tsnsect
, 'shares'):
172 # clean up leading and trailing spaces & make sure ref is valid
174 for x
in config
.get(tsnsect
, 'shares').split(','):
176 if config
.has_section(y
):
177 tsnshares
.append((y
, Bdict(config
.items(y
))))
182 if get_server('nosettings', 'false').lower() in ['false', 'no', 'off']:
183 shares
.append(('Settings', {'type': 'settings'}))
184 if get_server('tivo_mak') and get_server('togo_path'):
185 shares
.append(('ToGo', {'type': 'togo'}))
191 return config
.getboolean('Server', 'debug')
195 def getOptres(tsn
=None):
197 return config
.getboolean('_tivo_' + tsn
, 'optres')
200 return config
.getboolean(get_section(tsn
), 'optres')
203 return config
.getboolean('Server', 'optres')
210 logger
= logging
.getLogger('pyTivo.config')
212 if fname
in bin_paths
:
213 return bin_paths
[fname
]
215 if config
.has_option('Server', fname
):
216 fpath
= config
.get('Server', fname
)
217 if os
.path
.exists(fpath
) and os
.path
.isfile(fpath
):
218 bin_paths
[fname
] = fpath
221 logger
.error('Bad %s path: %s' % (fname
, fpath
))
223 if sys
.platform
== 'win32':
228 for path
in ([os
.path
.join(os
.path
.dirname(__file__
), 'bin')] +
229 os
.getenv('PATH').split(os
.pathsep
)):
230 fpath
= os
.path
.join(path
, fname
+ fext
)
231 if os
.path
.exists(fpath
) and os
.path
.isfile(fpath
):
232 bin_paths
[fname
] = fpath
235 logger
.warn('%s not found' % fname
)
239 if config
.has_option('Server', 'ffmpeg_wait'):
240 return max(int(float(config
.get('Server', 'ffmpeg_wait'))), 1)
244 def getFFmpegPrams(tsn
):
245 return get_tsn('ffmpeg_pram', tsn
, True)
247 def isHDtivo(tsn
): # TSNs of High Definition TiVos
248 return bool(tsn
and tsn
[0] >= '6' and tsn
[:3] != '649')
250 def is4Ktivo(tsn
): # TSNs of 4K TiVos
251 return bool(tsn
[:3] in ('849', '8F9'))
254 return get_server('ts', 'auto').lower()
256 def is_ts_capable(tsn
): # tsn's of Tivos that support transport streams
257 return bool(tsn
and (tsn
[0] >= '7' or tsn
.startswith('663')))
259 def getValidWidths():
260 return [1920, 1440, 1280, 720, 704, 544, 480, 352]
262 def getValidHeights():
263 return [1080, 720, 480] # Technically 240 is also supported
265 # Return the number in list that is nearest to x
266 # if two values are equidistant, return the larger
267 def nearest(x
, list):
268 return reduce(lambda a
, b
: closest(x
, a
, b
), list)
270 def closest(x
, a
, b
):
273 if da
< db
or (da
== db
and a
> b
):
278 def nearestTivoHeight(height
):
279 return nearest(height
, getValidHeights())
281 def nearestTivoWidth(width
):
282 return nearest(width
, getValidWidths())
284 def getTivoHeight(tsn
):
288 return [480, 1080][isHDtivo(tsn
)]
290 def getTivoWidth(tsn
):
294 return [544, 1920][isHDtivo(tsn
)]
297 return max(int(strtod(i
)) / 64000, 1) * 64
299 def getAudioBR(tsn
=None):
300 rate
= get_tsn('audio_br', tsn
)
303 # convert to non-zero multiple of 64 to ensure ffmpeg compatibility
304 # compare audio_br to max_audio_br and return lowest
305 return str(min(_trunc64(rate
), getMaxAudioBR(tsn
))) + 'k'
308 return str(int(strtod(i
)) / 1000) + 'k'
310 def getVideoBR(tsn
=None):
311 rate
= get_tsn('video_br', tsn
)
315 return getMaxVideoBR(tsn
)
317 return ['4096K', '16384K'][isHDtivo(tsn
)]
319 def getMaxVideoBR(tsn
=None):
320 rate
= get_tsn('max_video_br', tsn
)
325 def getBuffSize(tsn
=None):
326 size
= get_tsn('bufsize', tsn
)
332 return ['1024k', '4096k'][isHDtivo(tsn
)]
334 def getMaxAudioBR(tsn
=None):
335 rate
= get_tsn('max_audio_br', tsn
)
336 # convert to non-zero multiple of 64 for ffmpeg compatibility
338 return _trunc64(rate
)
341 def get_section(tsn
):
345 return ['_tivo_SD', '_tivo_HD'][isHDtivo(tsn
)]
347 def get_tsn(name
, tsn
=None, raw
=False):
349 return config
.get('_tivo_' + tsn
, name
, raw
)
352 return config
.get(get_section(tsn
), name
, raw
)
355 return config
.get('Server', name
, raw
)
359 # Parse a bitrate using the SI/IEEE suffix values as if by ffmpeg
360 # For example, 2K==2000, 2Ki==2048, 2MB==16000000, 2MiB==16777216
361 # Algorithm: http://svn.mplayerhq.hu/ffmpeg/trunk/libavcodec/eval.c
363 prefixes
= {'y': -24, 'z': -21, 'a': -18, 'f': -15, 'p': -12,
364 'n': -9, 'u': -6, 'm': -3, 'c': -2, 'd': -1,
365 'h': 2, 'k': 3, 'K': 3, 'M': 6, 'G': 9,
366 'T': 12, 'P': 15, 'E': 18, 'Z': 21, 'Y': 24}
367 p
= re
.compile(r
'^(\d+)(?:([yzafpnumcdhkKMGTPEZY])(i)?)?([Bb])?$')
370 raise SyntaxError('Invalid bit value syntax')
371 (coef
, prefix
, power
, byte
) = m
.groups()
375 exponent
= float(prefixes
[prefix
])
378 value
= float(coef
) * pow(2.0, exponent
/ 0.3)
381 value
= float(coef
) * pow(10.0, exponent
)
382 if byte
== 'B': # B == Byte, b == bit
387 if (config
.has_section('loggers') and
388 config
.has_section('handlers') and
389 config
.has_section('formatters')):
391 logging
.config
.fileConfig(config_files
)
394 logging
.basicConfig(level
=logging
.DEBUG
)
396 logging
.basicConfig(level
=logging
.INFO
)