feat: customize season detection with variable setting it to constant

src/movies_organizer.py

@@ -26,9 +26,10 @@ def movies(ctx, path, force):
 @main.command(help="Rename TV Series")
 @click.pass_context
 @click.option("--path", "-p", help="Path", default=".")
+@click.option("--season", "-s", help="Which season use if its not detected", type=click.INT, default=None)
 @click.option("--force", "-f", "-y", help="Automatically rename", is_flag=True, default=False)
-def series(ctx, path, force):
-    rename_series(ctx, path, force)
+def series(ctx, path, season, force):
+    rename_series(ctx, path, season, force)
 
 
 if __name__ == "__main__":

src/rename_series.py

@@ -53,16 +53,18 @@ def removeIllegal(str):
 RE_EA = re.compile("E?P?(\d+).*", re.IGNORECASE)
 
 
-def get_season_episode(file_name: str):
+def get_season_episode(file_name: str, season):
     if re.search(RE_SE, file_name):
         season, episode = re.search(RE_SE, file_name).groups()
         return AddZero(season), AddZero(episode)
-    elif re.search(RE_X, file_name):
+
+    if re.search(RE_X, file_name):
         season, episode = re.search(RE_X, file_name).groups()
         return AddZero(season), AddZero(episode)
-    elif re.search(RE_E, file_name):
+
+    if re.search(RE_E, file_name):
         episode = re.search(RE_E, file_name).groups()[0]
-        return AddZero(1), AddZero(episode)
+        return AddZero(1 if season is None else season), AddZero(episode)
 
     return "", ""
 
@@ -80,7 +82,7 @@ def AddZero(input):
     return input
 
 
-def rename_series(ctx, path, force):
+def rename_series(ctx, path, season, force):
     click.echo("Reading Files....")
 
     for file in sorted(os.listdir(path)):
@@ -108,7 +110,7 @@ def rename_series(ctx, path, force):
             file_name = file_name.replace(stuff, "")
         file_name = file_name.replace(".", " ")
 
-        season, episode = get_season_episode(file_name)
+        season, episode = get_season_episode(file_name, season)
         if not (season and episode):
             click.secho(f'No Season/Episode found in "{file_name}"', fg="red")
             continue