From: Mark Wooding <mdw@distorted.org.uk>
Date: Thu, 17 Mar 2022 16:15:04 +0000 (+0000)
Subject: mkm3u: Pull out the episode automatic naming logic.
X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~mdw/git/epls/commitdiff_plain/4a25b86c5923349432cee14ed0c49a458c245a94

mkm3u: Pull out the episode automatic naming logic.
---

diff --git a/mkm3u b/mkm3u
index 2e2f5cd..5fd2196 100755
--- a/mkm3u
+++ b/mkm3u
@@ -299,16 +299,18 @@ class BaseSeason (object):
     me.episodes.append(ep)
     src.nuses += neps; me.ep_i += neps
     return ep
+  def _epnames(me, i, neps):
+    playlist = me.series.playlist
+    if neps == 1: return playlist.epname, "%d" % i
+    elif neps == 2: return playlist.epnames, "%d, %d" % (i, i + 1)
+    else: return playlist.epnames, "%d–%d" % (i, i + neps - 1)
 
 class Season (BaseSeason):
   def __init__(me, series, title, i, *args, **kw):
     super().__init__(series, *args, **kw)
     me.title, me.i = title, i
   def _eplabel(me, i, neps, title):
-    playlist = me.series.playlist
-    if neps == 1: epname = playlist.epname; epn = "%d" % i
-    elif neps == 2: epname = playlist.epnames; epn = "%d, %d" % (i, i + 1)
-    else: epname = playlist.epnames; epn = "%d–%d" % (i, i + neps - 1)
+    epname, epn = me._epnames(i, neps)
     if title is None:
       if me.implicitp: label = "%s %s" % (epname, epn)
       elif me.title is None: label = "%s %d.%s" % (epname, me.i, epn)