34 lines
1.6 KiB
Python
34 lines
1.6 KiB
Python
from .text_formatter import TextFormatter
|
|
from .date_formatter import DateFormatter
|
|
from .special_info_formatter import SpecialInfoFormatter
|
|
|
|
|
|
class ProposedNameFormatter:
|
|
"""Class for formatting proposed filenames"""
|
|
|
|
def __init__(self, extractor):
|
|
"""Initialize with media extractor data"""
|
|
|
|
self.__order = f"[{extractor.get('order')}] " if extractor.get("order") else ""
|
|
self.__title = extractor.get("title") or "Unknown Title"
|
|
self.__year = DateFormatter.format_year(extractor.get("year"))
|
|
self.__source = f" {extractor.get('source')}" if extractor.get("source") else ""
|
|
self.__frame_class = extractor.get("frame_class") or None
|
|
self.__hdr = f",{extractor.get('hdr')}" if extractor.get("hdr") else ""
|
|
self.__audio_langs = extractor.get("audio_langs") or None
|
|
self.__special_info = f" \[{SpecialInfoFormatter.format_special_info(extractor.get('special_info'))}]" if extractor.get("special_info") else ""
|
|
self.__extension = extractor.get("extension") or "ext"
|
|
|
|
def __str__(self) -> str:
|
|
"""Convert the proposed name to string"""
|
|
return self.rename_line()
|
|
|
|
def rename_line(self) -> str:
|
|
return f"{self.__order}{self.__title} {self.__year}{self.__special_info}{self.__source} [{self.__frame_class}{self.__hdr},{self.__audio_langs}].{self.__extension}"
|
|
|
|
def rename_line_formatted(self, file_path) -> str:
|
|
"""Format the proposed name for display with color"""
|
|
if file_path.name == str(self):
|
|
return f">> {TextFormatter.green(str(self))} <<"
|
|
return f">> {TextFormatter.bold_yellow(str(self))} <<"
|