Update test

重写rename，更容易修改

src/module/parser/analyser/__init__.py

@@ -1,4 +1,4 @@
 from .raw_parser import raw_parser
-from .rename_parser import DownloadParser
+from .rename_parser import torrent_parser
 from .tmdb_parser import TMDBMatcher
 

src/module/parser/analyser/rename_parser.py

@@ -2,10 +2,8 @@ import re
 import logging
 from dataclasses import dataclass
 
-
 logger = logging.getLogger(__name__)
 
-
 @dataclass
 class DownloadInfo:
     name: str
@@ -26,91 +24,91 @@ RULES = [
 ]
 
 
-class DownloadParser:
-    def __init__(self):
-        self.method_dict = {
-            "normal": self.rename_normal,
-            "pn": self.rename_pn,
-            "advance": self.rename_advance,
-            "no_season_pn": self.rename_no_season_pn,
-            "none": self.rename_none
-        }
+def rename_init(name, folder_name, season, suffix) -> DownloadInfo:
+    n = re.split(r"[\[\]()【】（）]", name)
+    suffix = suffix if suffix else n[-1]
+    if len(n) > 1:
+        file_name = name.replace(f"[{n[1]}]", "")
+    else:
+        file_name = name
+    if season < 10:
+        season = f"0{season}"
+    return DownloadInfo(name, season, suffix, file_name, folder_name)
 
 
-    @staticmethod
-    def rename_init(name, folder_name, season, suffix) -> DownloadInfo:
-        n = re.split(r"[\[\]()【】（）]", name)
-        suffix = suffix if suffix else n[-1]
-        if len(n) > 1:
-            file_name = name.replace(f"[{n[1]}]", "")
-        else:
-            file_name = name
-        if season < 10:
-            season = f"0{season}"
-        return DownloadInfo(name, season, suffix, file_name, folder_name)
+def rename_normal(info: DownloadInfo):
+    for rule in RULES:
+        match_obj = re.match(rule, info.name, re.I)
+        if match_obj is not None:
+            title = re.sub(r"([Ss]|Season )\d{1,3}", "", match_obj.group(1)).strip()
+            new_name = f"{title} S{info.season}E{match_obj.group(2)}{match_obj.group(3)}"
+            return new_name
 
-    def rename_normal(self, info: DownloadInfo):
-        for rule in RULES:
-            match_obj = re.match(rule, info.name, re.I)
-            if match_obj is not None:
-                title = re.sub(r"([Ss]|Season )\d{1,3}", "", match_obj.group(1)).strip()
-                new_name = f"{title} S{info.season}E{match_obj.group(2)}{match_obj.group(3)}"
-                return new_name
 
-    def rename_pn(self, info: DownloadInfo):
-        for rule in RULES:
-            match_obj = re.match(rule, info.file_name, re.I)
-            if match_obj is not None:
-                title = re.sub(r"([Ss]|Season )\d{1,3}", "", match_obj.group(1)).strip()
-                title = title if title != "" else info.folder_name
-                new_name = re.sub(
-                    r"[\[\]]",
-                    "",
-                    f"{title} S{info.season}E{match_obj.group(2)}{info.suffix}",
-                )
-                return new_name
+def rename_pn(info: DownloadInfo):
+    for rule in RULES:
+        match_obj = re.match(rule, info.file_name, re.I)
+        if match_obj is not None:
+            title = re.sub(r"([Ss]|Season )\d{1,3}", "", match_obj.group(1)).strip()
+            title = title if title != "" else info.folder_name
+            new_name = re.sub(
+                r"[\[\]]",
+                "",
+                f"{title} S{info.season}E{match_obj.group(2)}{info.suffix}",
+            )
+            return new_name
 
-    def rename_advance(self, info: DownloadInfo):
-        for rule in RULES:
-            match_obj = re.match(rule, info.file_name, re.I)
-            if match_obj is not None:
-                new_name = re.sub(
-                    r"[\[\]]",
-                    "",
-                    f"{info.folder_name} S{info.season}E{match_obj.group(2)}{info.suffix}",
-                )
-                return new_name
 
-    def rename_no_season_pn(self, info: DownloadInfo):
-        for rule in RULES:
-            match_obj = re.match(rule, info.file_name, re.I)
-            if match_obj is not None:
-                title = match_obj.group(1).strip()
-                new_name = re.sub(
-                    r"[\[\]]",
-                    "",
-                    f"{title} E{match_obj.group(2)}{info.suffix}",
-                )
-                return new_name
+def rename_advance(info: DownloadInfo):
+    for rule in RULES:
+        match_obj = re.match(rule, info.file_name, re.I)
+        if match_obj is not None:
+            new_name = re.sub(
+                r"[\[\]]",
+                "",
+                f"{info.folder_name} S{info.season}E{match_obj.group(2)}{info.suffix}",
+            )
+            return new_name
 
-    @staticmethod
-    def rename_none(info: DownloadInfo):
-        return info.name
 
-    def download_rename(
-            self,
-            name: str,
-            folder_name,
-            season,
-            suffix,
-            method
-    ):
-        rename_info = self.rename_init(name, folder_name, season, suffix)
-        return self.method_dict[method.lower()](rename_info)
+def rename_no_season_pn(info: DownloadInfo):
+    for rule in RULES:
+        match_obj = re.match(rule, info.file_name, re.I)
+        if match_obj is not None:
+            title = match_obj.group(1).strip()
+            new_name = re.sub(
+                r"[\[\]]",
+                "",
+                f"{title} E{match_obj.group(2)}{info.suffix}",
+            )
+            return new_name
+
+
+def rename_none(info: DownloadInfo):
+    return info.name
+
+
+METHODS = {
+    "normal": rename_normal,
+    "pn": rename_pn,
+    "advance": rename_advance,
+    "no_season_pn": rename_no_season_pn,
+    "none": rename_none
+}
+
+
+def torrent_parser(
+        file_name: str,
+        folder_name: str,
+        season: int,
+        suffix: str,
+        method: str = "pn",
+):
+    info = rename_init(file_name, folder_name, season, suffix)
+    return METHODS[method.lower()](info)
 
 
 if __name__ == "__main__":
     name = "[Lilith-Raws] Tate no Yuusha no Nariagari S02 - 02 [Baha][WEB-DL][1080p][AVC AAC][CHT][MP4]"
-    rename = DownloadParser()
-    new_name = rename.download_rename(name, "异世界舅舅（2022）", 1, ".mp4", "normal")
+    new_name = torrent_parser(name, "异世界舅舅（2022）", 2, ".mp4", "advance")
     print(new_name)