mirror of
https://github.com/shadoll/moma.git
synced 2026-08-28 11:33:25 +00:00
feat: Enhance FilenameExtractor to support various filename formats and add comprehensive test cases
This commit is contained in:
@@ -0,0 +1,342 @@
|
||||
[
|
||||
{
|
||||
"testname": "test-001",
|
||||
"filename": "Movie Title (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
}
|
||||
},
|
||||
{
|
||||
"testname": "test-002",
|
||||
"filename": "[01] Movie Title (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": "01",
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
}
|
||||
},
|
||||
{
|
||||
"filename": "01. Movie Title (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": "01",
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-003"
|
||||
},
|
||||
{
|
||||
"filename": "1.1. Movie Title (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": "1.1",
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-004"
|
||||
},
|
||||
{
|
||||
"filename": "1.9 Movie Title (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": "1",
|
||||
"title": "9 Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-005"
|
||||
},
|
||||
{
|
||||
"filename": "9 (2009) BDRip [1080p,2ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "9",
|
||||
"year": "2009",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "2ukr,eng"
|
||||
},
|
||||
"testname": "test-006"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title 2020 BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-007"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title.2020.BDRip.[1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-008"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": null,
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-009"
|
||||
},
|
||||
{
|
||||
"filename": "Series Name S01E01 (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Series Name S01E01",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-010"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title (2020) [tmdbid-12345].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": null,
|
||||
"frame_class": null,
|
||||
"hdr": null,
|
||||
"movie_db": [
|
||||
"tmdb",
|
||||
"12345"
|
||||
],
|
||||
"special_info": null,
|
||||
"audio_langs": ""
|
||||
},
|
||||
"testname": "test-011"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title (2020) [Director's Cut] BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": [
|
||||
"Director's Cut"
|
||||
],
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-012"
|
||||
},
|
||||
{
|
||||
"filename": "\u0424\u0456\u043b\u044c\u043c \u041d\u0430\u0437\u0432\u0430 (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "\u0424\u0456\u043b\u044c\u043c \u041d\u0430\u0437\u0432\u0430",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-013"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title (2020) 1080p BDRip [ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-014"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title (2020) BDRip [2160p,HDR,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "2160p",
|
||||
"hdr": "HDR",
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-015"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title (2020) BDRip [1080p,2ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "2ukr,eng"
|
||||
},
|
||||
"testname": "test-016"
|
||||
},
|
||||
{
|
||||
"filename": "Movie.Title (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie.Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-017"
|
||||
},
|
||||
{
|
||||
"filename": "Movie Title (2020) BDRip 1080p ukr eng.mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-018"
|
||||
},
|
||||
{
|
||||
"filename": "[01.1] Movie: Subtitle (2020) [Director's Cut] BDRip [2160p,HDR,2ukr,eng] [tmdbid-12345].mkv",
|
||||
"expected": {
|
||||
"order": "01.1",
|
||||
"title": "Movie: Subtitle",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "2160p",
|
||||
"hdr": "HDR",
|
||||
"movie_db": [
|
||||
"tmdb",
|
||||
"12345"
|
||||
],
|
||||
"special_info": [
|
||||
"Director's Cut"
|
||||
],
|
||||
"audio_langs": "2ukr,eng"
|
||||
},
|
||||
"testname": "test-019"
|
||||
},
|
||||
{
|
||||
"filename": "1.9 (2009) BDRip [1080p,2ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": "1",
|
||||
"title": "9",
|
||||
"year": "2009",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "2ukr,eng"
|
||||
},
|
||||
"testname": "test-020"
|
||||
},
|
||||
{
|
||||
"filename": "1 2009 BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": null,
|
||||
"title": "1",
|
||||
"year": "2009",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-021"
|
||||
},
|
||||
{
|
||||
"filename": "9. Movie Title (2020) BDRip [1080p,ukr,eng].mkv",
|
||||
"expected": {
|
||||
"order": "9",
|
||||
"title": "Movie Title",
|
||||
"year": "2020",
|
||||
"source": "BDRip",
|
||||
"frame_class": "1080p",
|
||||
"hdr": null,
|
||||
"movie_db": null,
|
||||
"special_info": null,
|
||||
"audio_langs": "ukr,eng"
|
||||
},
|
||||
"testname": "test-022"
|
||||
}
|
||||
]
|
||||
@@ -0,0 +1,63 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Test script for filename metadata detection with assertions"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
import json
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
|
||||
|
||||
from renamer.extractors.filename_extractor import FilenameExtractor
|
||||
|
||||
def test_detection():
|
||||
with open('renamer/test/test_cases.json', 'r') as f:
|
||||
test_cases = json.load(f)
|
||||
|
||||
print("Testing filename metadata detection with assertions...\n")
|
||||
|
||||
passed = 0
|
||||
failed = 0
|
||||
|
||||
for i, case in enumerate(test_cases, 1):
|
||||
filename = case['filename']
|
||||
expected = case['expected']
|
||||
testname = case.get('testname', f'Test {i}')
|
||||
|
||||
print(f"{testname}: {filename}")
|
||||
|
||||
extractor = FilenameExtractor(filename)
|
||||
|
||||
actual = {
|
||||
"order": extractor.extract_order(),
|
||||
"title": extractor.extract_title(),
|
||||
"year": extractor.extract_year(),
|
||||
"source": extractor.extract_source(),
|
||||
"frame_class": extractor.extract_frame_class(),
|
||||
"hdr": extractor.extract_hdr(),
|
||||
"movie_db": extractor.extract_movie_db(),
|
||||
"special_info": extractor.extract_special_info(),
|
||||
"audio_langs": extractor.extract_audio_langs()
|
||||
}
|
||||
|
||||
# Check each field
|
||||
test_passed = True
|
||||
for key, exp_value in expected.items():
|
||||
act_value = actual[key]
|
||||
if act_value != exp_value:
|
||||
print(f" ❌ {key}: expected {exp_value!r}, got {act_value!r}")
|
||||
test_passed = False
|
||||
else:
|
||||
print(f" ✅ {key}: {act_value!r}")
|
||||
|
||||
if test_passed:
|
||||
print(" ✅ PASSED\n")
|
||||
passed += 1
|
||||
else:
|
||||
print(" ❌ FAILED\n")
|
||||
failed += 1
|
||||
|
||||
print(f"Results: {passed} passed, {failed} failed")
|
||||
return failed == 0
|
||||
|
||||
if __name__ == '__main__':
|
||||
success = test_detection()
|
||||
sys.exit(0 if success else 1)
|
||||
@@ -92,9 +92,9 @@ def test_extract_movie_db(filename):
|
||||
# Print filename and extracted movie DB clearly
|
||||
print(f"\nFilename: \033[1;36m{filename}\033[0m")
|
||||
print(f"Extracted movie DB: \033[1;32m{movie_db}\033[0m")
|
||||
# Movie DB should be tuple (str, str) or None
|
||||
# Movie DB should be list [str, str] or None
|
||||
if movie_db:
|
||||
assert isinstance(movie_db, tuple) and len(movie_db) == 2
|
||||
assert isinstance(movie_db, list) and len(movie_db) == 2
|
||||
assert isinstance(movie_db[0], str) and isinstance(movie_db[1], str)
|
||||
else:
|
||||
assert movie_db is None
|
||||
|
||||
@@ -0,0 +1,68 @@
|
||||
# Test filenames for data extraction
|
||||
# Each line is a filename to test extraction of: order, title, year, source, frame_class, hdr, movie_db, special_info, audio_langs
|
||||
|
||||
# Standard movie
|
||||
Movie Title (2020) BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# With order in brackets
|
||||
[01] Movie Title (2020) BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# With order dot
|
||||
01. Movie Title (2020) BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# Order with decimal
|
||||
1.1 Movie Title (2020) BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# Order like 1.9 (order 1, title 9...)
|
||||
1.9 Movie Title (2020) BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# Title with number (no order)
|
||||
9 (2009) BDRip [1080p,2ukr,eng].mkv
|
||||
|
||||
# Year not in parentheses
|
||||
Movie Title 2020 BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# Year in dots
|
||||
Movie Title.2020.BDRip.[1080p,ukr,eng].mkv
|
||||
|
||||
# No year
|
||||
Movie Title BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# Series
|
||||
Series Name S01E01 (2020) BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# With TMDB ID
|
||||
Movie Title (2020) [tmdbid-12345].mkv
|
||||
|
||||
# With special edition
|
||||
Movie Title (2020) [Director's Cut] BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# Cyrillic title
|
||||
Фільм Назва (2020) BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# Resolution in name
|
||||
Movie Title (2020) 1080p BDRip [ukr,eng].mkv
|
||||
|
||||
# HDR
|
||||
Movie Title (2020) BDRip [2160p,HDR,ukr,eng].mkv
|
||||
|
||||
# Multiple audio
|
||||
Movie Title (2020) BDRip [1080p,2ukr,eng].mkv
|
||||
|
||||
# Title with dots
|
||||
Movie.Title (2020) BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# No brackets
|
||||
Movie Title (2020) BDRip 1080p ukr eng.mkv
|
||||
|
||||
# Complex
|
||||
[01.1] Movie: Subtitle (2020) [Director's Cut] BDRip [2160p,HDR,2ukr,eng] [tmdbid-12345].mkv
|
||||
|
||||
# Order at start with dot and year
|
||||
1.9 (2009) BDRip [1080p,2ukr,eng].mkv
|
||||
|
||||
# Order at start with space and year
|
||||
1 2009 BDRip [1080p,ukr,eng].mkv
|
||||
|
||||
# Title starting with number dot
|
||||
9. Movie Title (2020) BDRip [1080p,ukr,eng].mkv
|
||||
Reference in New Issue
Block a user