1 import ConfigParser
, os
3 from ConfigParser
import NoOptionError
5 BLACKLIST_169
= ('540', '649')
7 config
= ConfigParser
.ConfigParser()
8 p
= os
.path
.dirname(__file__
)
9 config_file
= os
.path
.join(p
, 'pyTivo.conf')
10 config
.read(config_file
)
15 config
= ConfigParser
.ConfigParser()
16 config
.read(config_file
)
19 if config
.has_option('Server', 'GUID'):
20 guid
= config
.get('Server', 'GUID')
25 def getBeaconAddresses():
26 if config
.has_option('Server', 'beacon'):
27 beacon_ips
= config
.get('Server', 'beacon')
29 beacon_ips
= '255.255.255.255'
33 return config
.get('Server', 'Port')
35 def get169Setting(tsn
):
39 if config
.has_section('_tivo_' + tsn
):
40 if config
.has_option('_tivo_' + tsn
, 'aspect169'):
42 return config
.getboolean('_tivo_' + tsn
, 'aspect169')
46 if tsn
[:3] in BLACKLIST_169
:
51 def getShares(tsn
=''):
52 shares
= [(section
, dict(config
.items(section
)))
53 for section
in config
.sections()
54 if not(section
.startswith('_tivo_') or section
== 'Server')]
56 if config
.has_section('_tivo_' + tsn
):
57 if config
.has_option('_tivo_' + tsn
, 'shares'):
58 # clean up leading and trailing spaces & make sure ref is valid
60 for x
in config
.get('_tivo_' + tsn
, 'shares').split(','):
61 y
= x
.lstrip().rstrip()
62 if config
.has_section(y
):
63 tsnshares
+= [(y
, dict(config
.items(y
)))]
67 for name
, data
in shares
:
68 if not data
.get('auto_subshares', 'False').lower() == 'true':
71 base_path
= data
['path']
73 for item
in os
.listdir(base_path
):
74 item_path
= os
.path
.join(base_path
, item
)
75 if not os
.path
.isdir(item_path
):
78 new_name
= name
+ '/' + item
80 new_data
['path'] = item_path
82 shares
.append((new_name
, new_data
))
89 if config
.has_option('Server', 'debug'):
91 return str2tuple(config
.get('Server', 'debug')+',,')[ref
]
94 return str2tuple('False,,')[ref
]
98 debug
= config
.get('Server', 'hack83')
99 if debug
.lower() == 'true':
103 except NoOptionError
:
106 def getOptres(tsn
= None):
107 if tsn
and config
.has_section('_tivo_' + tsn
):
109 return config
.getboolean('_tivo_' + tsn
, 'optres')
110 except NoOptionError
, ValueError:
113 return config
.getboolean('Server', 'optres')
114 except NoOptionError
, ValueError:
117 def get(section
, key
):
118 return config
.get(section
, key
)
120 def getFFmpegTemplate(tsn
):
121 if tsn
and config
.has_section('_tivo_' + tsn
):
123 return config
.get('_tivo_' + tsn
, 'ffmpeg_tmpl', raw
=True)
124 except NoOptionError
:
127 return config
.get('Server', 'ffmpeg_tmpl', raw
=True)
128 except NoOptionError
: #default
129 return '%(video_codec)s %(video_fps)s %(video_br)s %(max_video_br)s \
130 %(buff_size)s %(aspect_ratio)s -comment pyTivo.py %(audio_br)s \
131 %(audio_fr)s %(audio_ch)s %(audio_codec)s %(ffmpeg_pram)s %(format)s'
133 def getFFmpegPrams(tsn
):
134 if tsn
and config
.has_section('_tivo_' + tsn
):
136 return config
.get('_tivo_' + tsn
, 'ffmpeg_pram', raw
=True)
137 except NoOptionError
:
140 return config
.get('Server', 'ffmpeg_pram', raw
=True)
141 except NoOptionError
:
144 def isHDtivo(tsn
): # tsn's of High Definition Tivo's
145 return tsn
!= '' and tsn
[:3] in ['648', '652']
147 def getValidWidths():
148 return [1920, 1440, 1280, 720, 704, 544, 480, 352]
150 def getValidHeights():
151 return [1080, 720, 480] # Technically 240 is also supported
153 # Return the number in list that is nearest to x
154 # if two values are equidistant, return the larger
155 def nearest(x
, list):
156 return reduce(lambda a
, b
: closest(x
, a
, b
), list)
158 def closest(x
, a
, b
):
159 if abs(x
- a
) < abs(x
- b
) or (abs(x
- a
) == abs(x
- b
) and a
> b
):
164 def nearestTivoHeight(height
):
165 return nearest(height
, getValidHeights())
167 def nearestTivoWidth(width
):
168 return nearest(width
, getValidWidths())
170 def getTivoHeight(tsn
):
171 if tsn
and config
.has_section('_tivo_' + tsn
):
173 height
= config
.getint('_tivo_' + tsn
, 'height')
174 return nearestTivoHeight(height
)
175 except NoOptionError
:
178 height
= config
.getint('Server', 'height')
179 return nearestTivoHeight(height
)
180 except NoOptionError
: #defaults for S3/S2 TiVo
186 def getTivoWidth(tsn
):
187 if tsn
and config
.has_section('_tivo_' + tsn
):
189 width
= config
.getint('_tivo_' + tsn
, 'width')
190 return nearestTivoWidth(width
)
191 except NoOptionError
:
194 width
= config
.getint('Server', 'width')
195 return nearestTivoWidth(width
)
196 except NoOptionError
: #defaults for S3/S2 TiVo
202 def getAudioBR(tsn
= None):
203 #convert to non-zero multiple of 64 to ensure ffmpeg compatibility
204 #compare audio_br to max_audio_br and return lowest
205 if tsn
and config
.has_section('_tivo_' + tsn
):
207 audiobr
= int(max(int(strtod(config
.get('_tivo_' + tsn
, 'audio_br'))/1000), 64)/64)*64
208 return str(min(audiobr
, getMaxAudioBR(tsn
))) + 'k'
209 except NoOptionError
:
212 audiobr
= int(max(int(strtod(config
.get('Server', 'audio_br'))/1000), 64)/64)*64
213 return str(min(audiobr
, getMaxAudioBR(tsn
))) + 'k'
214 except NoOptionError
:
215 return str(min(384, getMaxAudioBR(tsn
))) + 'k'
217 def getVideoBR(tsn
= None):
218 if tsn
and config
.has_section('_tivo_' + tsn
):
220 return config
.get('_tivo_' + tsn
, 'video_br')
221 except NoOptionError
:
224 return config
.get('Server', 'video_br')
225 except NoOptionError
: #defaults for S3/S2 TiVo
233 return str(int(strtod(config
.get('Server', 'max_video_br'))/1000)) + 'k'
234 except NoOptionError
: #default to 17Mi
239 return str(int(strtod(config
.get('Server', 'bufsize'))))
240 except NoOptionError
: #default 1024k
243 def getMaxAudioBR(tsn
= None):
244 #convert to non-zero multiple of 64 for ffmpeg compatibility
245 if tsn
and config
.has_section('_tivo_' + tsn
):
247 return int(int(strtod(config
.get('_tivo_' + tsn
, 'max_audio_br'))/1000)/64)*64
248 except NoOptionError
:
251 return int(int(strtod(config
.get('Server', 'max_audio_br'))/1000)/64)*64
252 except NoOptionError
:
253 return int(448) #default to 448
255 def getAudioCodec(tsn
= None):
256 if tsn
and config
.has_section('_tivo_' + tsn
):
258 return config
.get('_tivo_' + tsn
, 'audio_codec')
259 except NoOptionError
:
262 return config
.get('Server', 'audio_codec')
263 except NoOptionError
:
266 def getAudioCH(tsn
= None):
267 if tsn
and config
.has_section('_tivo_' + tsn
):
269 return config
.get('_tivo_' + tsn
, 'audio_ch')
270 except NoOptionError
:
273 return config
.get('Server', 'audio_ch')
274 except NoOptionError
:
277 def getAudioFR(tsn
= None):
278 if tsn
and config
.has_section('_tivo_' + tsn
):
280 return config
.get('_tivo_' + tsn
, 'audio_fr')
281 except NoOptionError
:
284 return config
.get('Server', 'audio_fr')
285 except NoOptionError
:
288 def getVideoFPS(tsn
= None):
289 if tsn
and config
.has_section('_tivo_' + tsn
):
291 return config
.get('_tivo_' + tsn
, 'video_fps')
292 except NoOptionError
:
295 return config
.get('Server', 'video_fps')
296 except NoOptionError
:
299 def getVideoCodec(tsn
= None):
300 if tsn
and config
.has_section('_tivo_' + tsn
):
302 return config
.get('_tivo_' + tsn
, 'video_codec')
303 except NoOptionError
:
306 return config
.get('Server', 'video_codec')
307 except NoOptionError
:
310 def getFormat(tsn
= None):
311 if tsn
and config
.has_section('_tivo_' + tsn
):
313 return config
.get('_tivo_' + tsn
, 'force_format')
314 except NoOptionError
:
317 return config
.get('Server', 'force_format')
318 except NoOptionError
:
323 L
= [x
.strip() for x
in items
]
326 # Parse a bitrate using the SI/IEEE suffix values as if by ffmpeg
327 # For example, 2K==2000, 2Ki==2048, 2MB==16000000, 2MiB==16777216
328 # Algorithm: http://svn.mplayerhq.hu/ffmpeg/trunk/libavcodec/eval.c
330 prefixes
= {'y': -24, 'z': -21, 'a': -18, 'f': -15, 'p': -12,
331 'n': -9, 'u': -6, 'm': -3, 'c': -2, 'd': -1,
332 'h': 2, 'k': 3, 'K': 3, 'M': 6, 'G': 9,
333 'T': 12, 'P': 15, 'E': 18, 'Z': 21, 'Y': 24}
334 p
= re
.compile(r
'^(\d+)(?:([yzafpnumcdhkKMGTPEZY])(i)?)?([Bb])?$')
337 raise SyntaxError('Invalid bit value syntax')
338 (coef
, prefix
, power
, byte
) = m
.groups()
342 exponent
= float(prefixes
[prefix
])
345 value
= float(coef
) * pow(2.0, exponent
/ 0.3)
348 value
= float(coef
) * pow(10.0, exponent
)
349 if byte
== 'B': # B == Byte, b == bit