[bind_ftp] Dossier spécifique pour l'Upload sur zeldren

[scripts-20-100.git] / rename_series.py

diff --git a/rename_series.py b/rename_series.py
index 4bc4c81769f131f5309124d85a42f0078bb3eb89..8b049197a2456ce4c735cc4c3180d7f4596f2a61 100755 (executable)
--- a/rename_series.py
+++ b/rename_series.py
@@ -12,16 +12,24 @@ import pprint
 
 FORMAT = "%%(series_name)s - [%%(season)0%(logmaxseason)sdx%%(episode)0%(logmaxepisode)sd] - %%(title)s.%%(extension)s"
 
-SEASONEPISODE_REGEXP = ur"^(?P<series_name>([A-Za-z0-9_-]+(\.| ))+)(S|- \[)(?P<season>\d+)(E|x)(?P<episode>\d+).*\.(?P<extension>[a-z0-9]*)$"
+SEASONEPISODE_REGEXP = ur"^(?P<series_name>([A-Za-z0-9_()-]+(\.| ))+)(S|- \[?)(?P<season>\d+)(E|x)(?P<episode>\d+).*\.(?P<extension>[a-z0-9]*)$"
 sep_re = re.compile(SEASONEPISODE_REGEXP)
 
+def _tryint(s):
+    """Renvoie int(s) ou s si échec."""
+    try:
+        return int(s)
+    except:
+        return s
+
 def get_titles(filename, debug=False, oneintwo=False):
     """Récupère les titres des épisodes dans le fichier"""
     text = open(filename).read()
     if oneintwo:
         # On enlève une ligne sur deux
         text = re.sub("(.*\n).*\n", r'\1', text)
-    titles = re.findall('^[0-9\s]+"(.*?)"', text, flags=re.MULTILINE)
+    titles = re.findall('^([0-9\s]+)"(.*?)"', text, flags=re.MULTILINE)
+    titles = {_tryint(n) : title for (n, title) in titles}
     if debug:
         print "Titres :"
         pprint.pprint(titles)
@@ -55,7 +63,7 @@ def get_files(titlesfilename, debug=False):
         pprint.pprint(dir)
     return dir
 
-def process(dir, titles, maxseason, seriesname, args):
+def process(dir, titles, maxseason, maxepisode, seriesname, args):
     """Prépare tous les renommages.
        Renvoie une map before -> after."""
     series_names = set([proper_name(d["series_name"]) for (_, d) in dir.values()])
@@ -67,11 +75,11 @@ def process(dir, titles, maxseason, seriesname, args):
         series_name = raw_input("Spécifier titre : ")
     else:
         series_name = list(series_names)[0]
-    l = range(1, len(dir) + 1)
+    l = range(args.starting, args.starting + len(dir))
     eps = dir.keys()
     eps.sort()
     if eps != l:
-        raise ValueError("Il manque des épisodes : %s" % eps)
+        raise ValueError("N° titres ≠ N° épisodes : %s ≠ %s" % (l, eps))
     if len(titles) != len(eps):
         print "Pas autant de titres que d'épisodes : on retente en lisant une ligne sur deux."
         titles = get_titles(args.titles, oneintwo=True)
@@ -79,14 +87,14 @@ def process(dir, titles, maxseason, seriesname, args):
             raise ValueError("Pas autant de titres que d'épisodes !")
     formatdir = {
         "logmaxseason" : len(str(maxseason)),
-        "logmaxepisode" : len(str(max(eps))),
+        "logmaxepisode" : max(len(str(max(eps))), len(str(maxepisode))),
         "series_name" : series_name
     }
     map = []
     for k in eps:
         (before, d) = dir[k]
         d.update(formatdir)
-        d["title"] = titles[k-1]
+        d["title"] = titles[k]
         newname = FORMAT % d
         newname = newname % d
         map.append((before, newname))
@@ -112,12 +120,14 @@ if __name__ == "__main__":
     parser.add_argument('-q', '--quiet', help="Ne pas afficher les warnings", action="store_true")
     parser.add_argument('-t', '--titles', help="Fichier contenant les titres des épisodes", action="store", type=str, default="titles")
     parser.add_argument('-m', '--maxseason', help="Nombre total de saisons (pour le nombre de 0)", action="store", type=str, default=1)
+    parser.add_argument('-M', '--maxepisode', help="Nombre total d'épisodes (pour le nombre de 0)", action="store", type=str, default=10)
     parser.add_argument('-s', '--seriesname', help="Override le nom de la série", action="store", type=str)
+    parser.add_argument('-n', '--starting', help="Numéro du 1er épisode", action="store", type=int, default=1)
     
     args = parser.parse_args()
     
     titles = get_titles(args.titles, args.verbose)
     files = get_files(args.titles, args.verbose)
-    map = process(files, titles, args.maxseason, args.seriesname, args)
+    map = process(files, titles, args.maxseason, args.maxepisode, args.seriesname, args)
     if confirm(map):
         do(map)