mirror of
https://github.com/Arrowar/StreamingCommunity.git
synced 2025-06-07 20:15:24 +00:00
72 lines
2.2 KiB
Python
72 lines
2.2 KiB
Python
# 19.06.24
|
|
|
|
import logging
|
|
|
|
from typing import List
|
|
|
|
|
|
# Internal utilities
|
|
from Src.Util._jsonConfig import config_manager
|
|
from Src.Util.os import remove_special_characters
|
|
|
|
|
|
# Config
|
|
MAP_EPISODE = config_manager.get('DEFAULT', 'map_episode_name')
|
|
|
|
|
|
def manage_selection(cmd_insert: str, max_count: int) -> List[int]:
|
|
"""
|
|
Manage user selection for seasons to download.
|
|
|
|
Args:
|
|
- cmd_insert (str): User input for season selection.
|
|
- max_count (int): Maximum count of seasons available.
|
|
|
|
Returns:
|
|
list_season_select (List[int]): List of selected seasons.
|
|
"""
|
|
list_season_select = []
|
|
logging.info(f"Command insert: {cmd_insert}, end index: {max_count + 1}")
|
|
|
|
# For a single number (e.g., '5')
|
|
if cmd_insert.isnumeric():
|
|
list_season_select.append(int(cmd_insert))
|
|
|
|
# For a range (e.g., '[5-12]')
|
|
elif "[" in cmd_insert:
|
|
start, end = map(int, cmd_insert[1:-1].split('-'))
|
|
list_season_select = list(range(start, end + 1))
|
|
|
|
# For all seasons
|
|
elif cmd_insert == "*":
|
|
list_season_select = list(range(1, max_count+1))
|
|
|
|
# Return list of selected seasons)
|
|
logging.info(f"List return: {list_season_select}")
|
|
return list_season_select
|
|
|
|
def map_episode_title(tv_name: str, number_season: int, episode_number: int, episode_name: str) -> str:
|
|
"""
|
|
Maps the episode title to a specific format.
|
|
|
|
Args:
|
|
tv_name (str): The name of the TV show.
|
|
number_season (int): The season number.
|
|
episode_number (int): The episode number.
|
|
episode_name (str): The original name of the episode.
|
|
|
|
Returns:
|
|
str: The mapped episode title.
|
|
"""
|
|
map_episode_temp = MAP_EPISODE
|
|
map_episode_temp = map_episode_temp.replace("%(tv_name)", remove_special_characters(tv_name))
|
|
map_episode_temp = map_episode_temp.replace("%(season)", str(number_season))
|
|
map_episode_temp = map_episode_temp.replace("%(episode)", str(episode_number))
|
|
map_episode_temp = map_episode_temp.replace("%(episode_name)", remove_special_characters(episode_name))
|
|
|
|
# Additional fix
|
|
map_episode_temp = map_episode_temp.replace(".", "_")
|
|
|
|
logging.info(f"Map episode string return: {map_episode_temp}")
|
|
return map_episode_temp
|