mirror of https://github.com/ekimekim/wubloader
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import os
|
|
import urllib
|
|
|
|
|
|
def generate_master(playlists):
|
|
"""Generate master playlist. Playlists arg should be a map {name: url}.
|
|
Little validation or encoding is done - please try to keep the names valid
|
|
without escaping.
|
|
"""
|
|
lines = ["#EXTM3U"]
|
|
for name, url in playlists.items():
|
|
lines += [
|
|
# We name each variant with a VIDEO rendition with no url
|
|
'#EXT-X-MEDIA:TYPE=VIDEO,GROUP-ID="{name}",NAME="{name}",AUTOSELECT=YES,DEFAULT=YES'.format(name=name),
|
|
'#EXT-X-STREAM-INF:VIDEO="{name}"'.format(name=name),
|
|
url,
|
|
]
|
|
return "\n".join(lines) + '\n'
|
|
|
|
|
|
def generate_media(segments, base_url):
|
|
"""Generate a media playlist from a list of segments as returned by common.get_best_segments().
|
|
Segments are specified as hour/name.ts relative to base_url.
|
|
"""
|
|
lines = [
|
|
"#EXTM3U",
|
|
"#EXT-X-TARGETDURATION:6",
|
|
]
|
|
for segment in segments:
|
|
# TODO handle missing bits, stream endings, other stuff
|
|
if segment is not None:
|
|
path = '/'.join(segment.path.split('/')[-2:])
|
|
lines.append("#EXTINF:{:.3f},live".format(segment.duration.total_seconds()))
|
|
lines.append(urllib.quote(os.path.join(base_url, path)))
|
|
return "\n".join(lines) + '\n'
|