diff --git a/scripts/remerge_movies_roman_studio.py b/scripts/remerge_movies_roman_studio.py index a8c9274..a8b64b2 100644 --- a/scripts/remerge_movies_roman_studio.py +++ b/scripts/remerge_movies_roman_studio.py @@ -22,7 +22,12 @@ from app.models.merge_candidate import MergeCandidate, MergeKind, MergeStatus from app.models.movie import Movie, MoviePerformer from app.resolve.movie_match import find_movie_candidates from app.resolve.movie_merge import MovieMergeError, merge_movies -from app.resolve.movie_score import AUTO_THRESHOLD, _roman_to_arabic, score_movie_candidate +from app.resolve.movie_score import ( + AUTO_THRESHOLD, + _roman_to_arabic, + _same_studio_name, + score_movie_candidate, +) def _affected_ids(session) -> list[uuid.UUID]: @@ -100,7 +105,10 @@ def main() -> None: roman = _roman_to_arabic(c.title_normalized or "") != (c.title_normalized or "") or ( _roman_to_arabic(m.title_normalized) != m.title_normalized ) - twin_studio = bool(c.studio_id and m.studio_id and c.studio_id != m.studio_id) + twin_studio = bool( + c.studio_id and m.studio_id and c.studio_id != m.studio_id + and _same_studio_name(s, c.studio_id, m.studio_id) + ) if not (roman or twin_studio): continue # ta para wyglada dla oceny tak samo jak przy ingescie b = score_movie_candidate(