renamer.py

#!/usr/local/bin/python3

import logging
import os
import re
import sys

logging.basicConfig(level=logging.INFO)


class NoMatch(Exception):
    pass


class NotAnEpisode(Exception):
    pass


class MissingArguments(Exception):
    pass


def construct_subs_name(episode_name: str):
    name = os.path.splitext(episode_name)[0] + ".srt"
    return name


def find_matching_subs(episode_name, subs_path: str) -> str:
    """looks in a given path for a match of
    standard tv show enumeration mark like 'S02E12'
    and returns corresponding subtitiles file"""
    ep_mark = re.match(r".*(\d{2}.\d{2})", episode_name, re.IGNORECASE)
    if not ep_mark:
        raise NotAnEpisode(f"No episode pattern found for {episode_name}")
    ep_mark = ep_mark.group(1)
    for subs in os.listdir(subs_path):
        if ep_mark.lower() in subs.lower():
            return subs
    raise NoMatch(f"no matching subtitles for {episode_name}")


def rename(episodes_path, subtitles_path: str) -> None:
    files = map(
        lambda file: os.path.join(episodes_path, file), os.listdir(episodes_path)
    )
    for episode in filter(os.path.isfile, files):
        try:
            sub_name = os.path.join(
                subtitles_path, find_matching_subs(episode, subtitles_path)
            )
            sub_new_name = construct_subs_name(episode)
            logging.info(f"{sub_name}\nrenamed to\n{sub_new_name}")
            os.rename(sub_name, sub_new_name)
        except (NotAnEpisode, NoMatch) as e:
            logging.warning(f"something is wrong:\n{e.args[0]}")


if __name__ == "__main__":
    try:
        EPISODES = sys.argv[1]
        SUBS = sys.argv[2]
    except IndexError:
        sys.exit("Please provide both season's and subtitles' locations")
    rename(EPISODES, SUBS)