From: Vincent Le Gallic <legallic@crans.org>
Date: Thu, 11 Oct 2018 13:17:44 +0000 (+0200)
Subject: [rename_series] Possibilité de choisir le nombre de 0 pour les épisodes
X-Git-Url: http://gitweb.pimeys.fr/?a=commitdiff_plain;h=2457eeebfbb461cdce46b7eb8b7399e29c1eb84d;p=scripts-20-100.git

[rename_series] Possibilité de choisir le nombre de 0 pour les épisodes

Pratique pour les saisons de moins de 10 épisodes
---

diff --git a/rename_series.py b/rename_series.py
index 60e6d5e..8b04919 100755
--- a/rename_series.py
+++ b/rename_series.py
@@ -63,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()])
@@ -87,7 +87,7 @@ 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 = []
@@ -120,6 +120,7 @@ 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)
     
@@ -127,6 +128,6 @@ if __name__ == "__main__":
     
     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)