X-Git-Url: https://git.distorted.org.uk/~mdw/epls/blobdiff_plain/4ae03d6ce334171419ea4c23d36384ed74c5f500..017188769528344839a7699c4528ad7cd26e443e:/mkm3u

diff --git a/mkm3u b/mkm3u
index d704856..9c50c3e 100755
--- a/mkm3u
+++ b/mkm3u
@@ -44,6 +44,10 @@ def getbool(s):
   elif s == "nil": return False
   else: raise ExpectedError("bad boolean `%s'" % s)
 
+def quote(s):
+  if s is None: return "-"
+  else: return '"' + s.replace("\\", "\\\\").replace('"', '\\"') + '"'
+
 class Words (object):
   def __init__(me, s):
     me._s = s
@@ -77,7 +81,7 @@ URL_SAFE_P = 256*[False]
 for ch in \
     b"ABCDEFGHIJKLMNOPQRSTUVWXYZ" \
     b"abcdefghijklmnopqrstuvwxyz" \
-    b"0123456789" b"!$%-.,/":
+    b"0123456789" b"!$%_-.,/":
   URL_SAFE_P[ch] = True
 def urlencode(s):
   return "".join((URL_SAFE_P[ch] and chr(ch) or "%%%02x" % ch
@@ -412,6 +416,7 @@ class Episode (object):
     me.chapters = []
     me.source, me.tno = src, tno
     me.series_title_p = series_title_p
+    me.tno, me.start_chapter, me.end_chapter = tno, startch, endch
     me.url, me.duration = src.url_and_duration(tno, startch, endch)
   def add_chapter(me, title, j):
     ch = Chapter(me, title, j)
@@ -485,9 +490,10 @@ class MovieSeason (BaseSeason):
     return label
 
 class Series (object):
-  def __init__(me, playlist, name, title = None, wantedp = True):
+  def __init__(me, playlist, name, title = None,
+               full_title = None, wantedp = True):
     me.playlist = playlist
-    me.name, me.title = name, title
+    me.name, me.title, me.full_title = name, title, full_title
     me.cur_season = None
     me.wantedp = wantedp
   def _add_season(me, season):
@@ -509,6 +515,8 @@ class Playlist (object):
     me.episodes = []
     me.epname, me.epnames = "Episode", "Episodes"
     me.nseries = 0
+    me.single_series_p = False
+    me.series_title = None
 
   def add_episode(me, episode):
     me.episodes.append(episode)
@@ -534,7 +542,33 @@ class Playlist (object):
         else:
           for ch in ep.chapters:
             f.write("#EXTINF:%d,,%s: %s\n%s\n" %
-                    (ch.duration, label, ch.title, ch.url))
+                      (ch.duration, label, ch.title, ch.url))
+
+  def dump(me, f):
+    if opts.list_name is not None: f.write("LIST %s\n" % opts.list_name)
+    if me.series_title is not None and \
+       me.nseries > 1 and not me.single_series_p:
+      raise ExpectedError("can't force series name for multi-series list")
+    series = set()
+    if me.single_series_p:
+      f.write("SERIES - %s\n" % quote(me.series_title))
+    for season in me.seasons:
+      for ep in season:
+        label = ep.label()
+        title = ep.season.series.full_title
+        if me.single_series_p:
+          stag = "-"
+          if title is not None: label = title + " " + label
+        else:
+          if title is None: title = me.series_title
+          stag = ep.season.series.name
+          if stag is None: stag = "-"
+          if stag not in series:
+            f.write("SERIES %s %s\n" % (stag, quote(title)))
+            series.add(stag)
+        f.write("ENTRY %s %s %s %d %d %d %g\n" %
+                  (stag, quote(label), quote(ep.source.fn),
+                   ep.tno, ep.start_chapter, ep.end_chapter, ep.duration))
 
   def write_deps(me, f, out):
     deps = set()
@@ -643,7 +677,19 @@ class EpisodeListParser (object):
     except ValueError: opts = None
     else: cmd, opts = cmd[:sep], cmd[sep + 1:]
 
-    if cmd == "series":
+    if cmd == "title":
+      for k, v in me._keyvals(opts): me._bad_keyval("title", k, v)
+      title = ww.rest(); check(title is not None, "missing title")
+      check(me._pl.series_title is None, "already set a title")
+      me._pl.series_title = title
+
+    elif cmd == "single":
+      for k, v in me._keyvals(opts): me._bad_keyval("single", k, v)
+      check(ww.rest() is None, "trailing junk")
+      check(not me._pl.single_series_p, "single-series already set")
+      me._pl.single_series_p = True
+
+    elif cmd == "series":
       name = None
       for k, v in me._keyvals(opts):
         if k is None: name = v
@@ -651,8 +697,17 @@ class EpisodeListParser (object):
       check(name is not None, "missing series name")
       check(name not in me._series, "series `%s' already defined" % name)
       title = ww.rest()
+      if title is None:
+        full = None
+      else:
+        try: sep = title.index("::")
+        except ValueError: full = title
+        else:
+          full = title[sep + 2:].strip()
+          if sep == 0: title = None
+          else: title = title[:sep].strip()
       me._set_mode(MODE_MULTI)
-      me._series[name] = series = Series(me._pl, name, title,
+      me._series[name] = series = Series(me._pl, name, title, full,
                                          me._series_wanted is None or
                                            name in me._series_wanted)
       if series.wantedp: me._pl.nseries += 1
@@ -829,9 +884,15 @@ class EpisodeListParser (object):
     return me._pl
 
 op = OP.OptionParser \
-  (usage = "%prog [-c] [-M DEPS] [-d CACHE] [-o OUT] [-s SERIES] EPLS\n"
-           "%prog -i -d CACHE",
+  (usage = "%prog [-Dc] [-L NAME] [-M DEPS] [-d CACHE] [-o OUT] [-s SERIES] EPLS\n"
+            "%prog -i -d CACHE",
    description = "Generate M3U playlists from an episode list.")
+op.add_option("-D", "--dump",
+              dest = "dump", action = "store_true", default = False,
+              help = "Dump playlist in machine-readable form")
+op.add_option("-L", "--list-name", metavar = "NAME",
+              dest = "list_name", type = "str", default = None,
+              help = "Set the playlist name")
 op.add_option("-M", "--make-deps", metavar = "DEPS",
               dest = "deps", type = "str", default = None,
               help = "Write a `make' fragment for dependencies")
@@ -884,10 +945,15 @@ try:
     ep.parse_file(argv[0])
     pl = ep.done()
 
+    if opts.list_name is None:
+      opts.list_name, _ = OS.path.splitext(OS.path.basename(argv[0]))
+
+    if opts.dump: outfn = pl.dump
+    else: outfn = pl.write
     if opts.output is None or opts.output == "-":
-      pl.write(SYS.stdout)
+      outfn(SYS.stdout)
     else:
-      with open(opts.output, "w") as f: pl.write(f)
+      with open(opts.output, "w") as f: outfn(f)
 
     if opts.deps:
       if opts.deps == "-":