Fix #191: 改进对于stage 2 media系列番号(S2M/S2MBD)的识别

Yuukiy · Jan 8, 2024 · 79e05ee · 79e05ee
1 parent 342ed34
commit 79e05ee
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/core/avid.py b/core/avid.py
@@ -46,7 +46,7 @@ def get_id(filepath: str) -> str:
         if match:
             return 'heydouga-' + '-'.join(match.groups())
         # 匹配片商 MUGEN 的奇怪番号。由于MK3D2DBD的模式，要放在普通番号模式之前进行匹配
-        match = re.search(r'(MKB?D)[-_]*(S\d{2,3})|(MK3D2DBD)[-_]*(\d\d)', filename, re.I)
+        match = re.search(r'(MKB?D)[-_]*(S\d{2,3})|(MK3D2DBD|S2M|S2MBD)[-_]*(\d{2,3})', filename, re.I)
         if match:
             if match.group(1) is not None:
                 avid = match.group(1) + '-' + match.group(2)

diff --git a/unittest/testdata_avid.txt b/unittest/testdata_avid.txt
@@ -647,3 +647,7 @@ MK3D2DBD-01 枢木みかん【MKV左右7.90G】.mkv	MK3D2DBD-01
 MKD-S148.mp4	MKD-S148
 (KIRARI)(MKBD-S143)KIRARI 143 週末モデル 神田るな_1.mp4	MKBD-S143
 (KIRARI)(MKBD-S143)KIRARI 143 週末モデル 神田るな_2.mp4	MKBD-S143
+S2MBD007.mp4	S2MBD-007
+Encore Vol.7 (S2MBD-007) Yui Hatano [1080p].mp4	S2MBD-007
+Encore Vol.07(アンコール Vol.07) 波多野結衣(Yui Hatano) S2MBD-007.iso	S2MBD-007
+S2M-007 Encore Vol.7 波多野結衣~絶品美女再現絕代.mp4	S2M-007