Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-07-08 10:44:20 -04:00
Code Issues Packages Projects Releases Wiki Activity

Create the xem fixup rule (always runs for now)

Browse Source
This commit is contained in:
Zoe Roux 2024-02-05 01:00:55 +01:00
parent 0439e1f37a
commit ed9c4ebb68
1 changed files with 76 additions and 56 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

132
scanner/scanner/parser/rules.py
View File

@ -1,6 +1,6 @@
# Read that for examples/rules: https://github.com/pymedusa/Medusa/blob/master/medusa/name_parser/rules/rules.py # Read that for examples/rules: https://github.com/pymedusa/Medusa/blob/master/medusa/name_parser/rules/rules.py
from typing import Any, Generator, Iterable, List, Optional, cast from typing import Any, List, Optional, cast
from rebulk import Rule, RemoveMatch, AppendMatch, POST_PROCESS from rebulk import Rule, RemoveMatch, AppendMatch, POST_PROCESS
from rebulk.match import Matches, Match from rebulk.match import Matches, Match
from copy import copy from copy import copy
@ -10,7 +10,6 @@ class EpisodeTitlePromotion(Rule):
"""Promote "episode_title" to "episode" when the title is in fact the episode number """Promote "episode_title" to "episode" when the title is in fact the episode number
Example: '[Erai-raws] Youkoso Jitsuryoku Shijou Shugi no Kyoushitsu e S3 - 05 [1080p][Multiple Subtitle][0DDEAFCD].mkv' Example: '[Erai-raws] Youkoso Jitsuryoku Shijou Shugi no Kyoushitsu e S3 - 05 [1080p][Multiple Subtitle][0DDEAFCD].mkv'
Default: Default:
```json ```json
{ {
@ -18,15 +17,8 @@ class EpisodeTitlePromotion(Rule):
"title": "Youkoso Jitsuryoku Shijou Shugi no Kyoushitsu e", "title": "Youkoso Jitsuryoku Shijou Shugi no Kyoushitsu e",
"season": 3, "season": 3,
"episode_title": "05", "episode_title": "05",
"screen_size": "1080p",
"subtitle_language": "Multiple languages",
"crc32": "0DDEAFCD",
"container": "mkv",
"mimetype": "video/x-matroska",
"type": "episode"
} }
``` ```
Expected: Expected:
```json ```json
{ {
@ -34,12 +26,6 @@ class EpisodeTitlePromotion(Rule):
"title": "Youkoso Jitsuryoku Shijou Shugi no Kyoushitsu e", "title": "Youkoso Jitsuryoku Shijou Shugi no Kyoushitsu e",
"season": 3, "season": 3,
"episode": 5, "episode": 5,
"screen_size": "1080p",
"subtitle_language": "Multiple languages",
"crc32": "0DDEAFCD",
"container": "mkv",
"mimetype": "video/x-matroska",
"type": "episode"
} }
``` ```
""" """
@ -67,7 +53,6 @@ class TitleNumberFixup(Rule):
Example: '[Erai-raws] Zom 100 - Zombie ni Naru made ni Shitai 100 no Koto - 01 [1080p][Multiple Subtitle][8AFBB298].mkv' Example: '[Erai-raws] Zom 100 - Zombie ni Naru made ni Shitai 100 no Koto - 01 [1080p][Multiple Subtitle][8AFBB298].mkv'
(or '[SubsPlease] Mob Psycho 100 Season 3 - 12 (1080p) [E5058D7B].mkv') (or '[SubsPlease] Mob Psycho 100 Season 3 - 12 (1080p) [E5058D7B].mkv')
Default: Default:
```json ```json
{ {
@ -78,15 +63,8 @@ class TitleNumberFixup(Rule):
1 1
], ],
"episode_title": "Zombie ni Naru made ni Shitai", "episode_title": "Zombie ni Naru made ni Shitai",
"screen_size": "1080p",
"subtitle_language": "Multiple languages",
"crc32": "8AFBB298",
"container": "mkv",
"mimetype": "video/x-matroska",
"type": "episode"
} }
``` ```
Expected: Expected:
```json ```json
{ {
@ -94,12 +72,6 @@ class TitleNumberFixup(Rule):
"title": "Zom 100", "title": "Zom 100",
"episode": 1, "episode": 1,
"episode_title": "Zombie ni Naru made ni Shitai 100 no Koto", "episode_title": "Zombie ni Naru made ni Shitai 100 no Koto",
"screen_size": "1080p",
"subtitle_language": "Multiple languages",
"crc32": "8AFBB298",
"container": "mkv",
"mimetype": "video/x-matroska",
"type": "episode"
} }
``` ```
""" """
@ -110,14 +82,14 @@ class TitleNumberFixup(Rule):
def when(self, matches: Matches, context) -> Any: def when(self, matches: Matches, context) -> Any:
episodes: List[Match] = matches.named("episode") # type: ignore episodes: List[Match] = matches.named("episode") # type: ignore
if not episodes or len(episodes) < 2: if len(episodes) < 2:
return return
to_remove = [] to_remove = []
to_add = [] to_add = []
for episode in episodes: for episode in episodes:
prevs: List[Match] = matches.previous(episode) # type: ignore prevs: List[Match] = matches.previous(episode) # type: ignore
title = prevs[0] if len(prevs) > 0 and prevs[0].tagged("title") else None title = prevs[0] if prevs and prevs[0].tagged("title") else None
if not title: if not title:
continue continue
@ -127,9 +99,9 @@ class TitleNumberFixup(Rule):
continue continue
to_remove.extend([title, episode]) to_remove.extend([title, episode])
newTitle = copy(title) new_title = copy(title)
newTitle.end = episode.end new_title.end = episode.end
newTitle.value = f"{title.value} {episode.value}" new_title.value = f"{title.value} {episode.value}"
# If an hole was created to parse the episode at the current pos, merge it back into the title # If an hole was created to parse the episode at the current pos, merge it back into the title
holes = matches.holes(episode.end) holes = matches.holes(episode.end)
@ -138,10 +110,10 @@ class TitleNumberFixup(Rule):
if "-" in val: if "-" in val:
val, *_ = val.split("-") val, *_ = val.split("-")
val = val.rstrip() val = val.rstrip()
newTitle.value = f"{newTitle.value}{val}" new_title.value = f"{new_title.value}{val}"
newTitle.end += len(val) new_title.end += len(val)
to_add.append(newTitle) to_add.append(new_title)
return [to_remove, to_add] return [to_remove, to_add]
@ -149,7 +121,6 @@ class MultipleSeasonRule(Rule):
"""Understand `abcd Season 2 - 5.mkv` as S2E5 """Understand `abcd Season 2 - 5.mkv` as S2E5
Example: '[Erai-raws] Spy x Family Season 2 - 08 [1080p][Multiple Subtitle][00C44E2F].mkv' Example: '[Erai-raws] Spy x Family Season 2 - 08 [1080p][Multiple Subtitle][00C44E2F].mkv'
Default: Default:
```json ```json
{ {
@ -163,27 +134,14 @@ class MultipleSeasonRule(Rule):
7, 7,
8 8
], ],
"screen_size": "1080p",
"release_group": "Multiple Subtitle",
"crc32": "00C44E2F",
"container": "mkv",
"mimetype": "video/x-matroska",
"type": "episode"
} }
``` ```
Expected: Expected:
```json ```json
{ {
"title": "Spy x Family", "title": "Spy x Family",
"season": 2, "season": 2,
"episode": 8, "episode": 8,
"screen_size": "1080p",
"release_group": "Multiple Subtitle",
"crc32": "00C44E2F",
"container": "mkv",
"mimetype": "video/x-matroska",
"type": "episode"
} }
``` ```
""" """
@ -194,7 +152,7 @@ class MultipleSeasonRule(Rule):
def when(self, matches: Matches, context) -> Any: def when(self, matches: Matches, context) -> Any:
seasons: List[Match] = matches.named("season") # type: ignore seasons: List[Match] = matches.named("season") # type: ignore
if not seasons or len(seasons) < 1: if not seasons:
return return
# Only apply this rule if all seasons are due to the same match # Only apply this rule if all seasons are due to the same match
@ -208,12 +166,12 @@ class MultipleSeasonRule(Rule):
if "-" not in value: if "-" not in value:
return return
newSeason, *newEpisodes = (x.strip() for x in value.split("-")) new_season, *new_episodes = (x.strip() for x in value.split("-"))
to_remove = [x for x in seasons if cast(Match, x.parent).value != newSeason] to_remove = [x for x in seasons if cast(Match, x.parent).value != new_season]
to_add = [] to_add = []
try: try:
episodes = [int(x) for x in newEpisodes] episodes = [int(x) for x in new_episodes]
parents: List[Match] = [match.parent for match in to_remove] # type: ignore parents: List[Match] = [match.parent for match in to_remove] # type: ignore
for episode in episodes: for episode in episodes:
smatch = next( smatch = next(
@ -231,4 +189,66 @@ class MultipleSeasonRule(Rule):
return [to_remove, to_add] return [to_remove, to_add]
# TODO: xem fixup? like for "Somthing Season 2" was a xem override but we converted that to "Something", season: 2 then convert it back here? class XemFixup(Rule):
"""Fix both alternate names and seasons that are known on the xem but parsed differently by guessit
Example: "JoJo's Bizarre Adventure - Diamond is Unbreakable - 12.mkv"
Default:
```json
{
"title": "JoJo's Bizarre Adventure",
"alternative_title": "Diamond is Unbreakable",
"episode": 12,
}
```
Expected:
```json
{
"title": "JoJo's Bizarre Adventure - Diamond is Unbreakable",
"episode": 12,
}
```
Or
Example: 'Owarimonogatari S2 E15.mkv'
Default:
```json
{
"title": "Owarimonogatari",
"season": 2,
"episode": 15
}
```
Expected:
```json
{
"title": "Owarimonogatari S2",
"episode": 15
}
```
"""
priority = POST_PROCESS
consequence = [RemoveMatch, AppendMatch]
def when(self, matches: Matches, context) -> Any:
titles: List[Match] = matches.named("title", lambda m: m.tagged("title")) # type: ignore
if not titles:
return
title = titles[0]
nmatch: List[Match] = matches.next(title) # type: ignore
if not nmatch or not (nmatch[0].tagged("title") or nmatch[0].named("season")):
return
holes: List[Match] = matches.holes(title.end, nmatch[0].start) # type: ignore
hole = " ".join(f" {h.value}" if h.value != "-" else " - " for h in holes)
new_title = copy(title)
new_title.end = nmatch[0].end
new_title.value = f"{title.value}{hole}{nmatch[0].value}"
# TODO: check if new_title exists on thexem, if not early return
return [[title, nmatch[0]], [new_title]]