mirror of
https://github.com/hect0x7/JMComic-Crawler-Python.git
synced 2025-11-04 14:49:43 +08:00
39 lines
962 B
Python
39 lines
962 B
Python
from test_jmcomic import *
|
|
|
|
|
|
class Test_Custom(JmTestConfigurable):
|
|
|
|
def test_custom_entity(self):
|
|
"""
|
|
测试自定义属性
|
|
"""
|
|
dic = {1: 'd', 2: 'e'}
|
|
|
|
class MyAlbum(JmAlbumDetail):
|
|
|
|
@property
|
|
def aname(self):
|
|
return dic[int(self.album_id)]
|
|
|
|
class MyPhoto(JmPhotoDetail):
|
|
|
|
@property
|
|
def pname(self):
|
|
return dic[int(self.photo_id)]
|
|
|
|
JmModuleConfig.CLASS_ALBUM = MyAlbum
|
|
JmModuleConfig.CLASS_PHOTO = MyPhoto
|
|
|
|
base_dir: str = workspace()
|
|
dir_rule = DirRule('Bd_Aaname_Ppname', base_dir)
|
|
# noinspection PyTypeChecker
|
|
save_dir = dir_rule.deside_image_save_dir(
|
|
MyAlbum('1', *['0'] * 13),
|
|
MyPhoto('2', *['0'] * 7)
|
|
)
|
|
|
|
self.assertEqual(
|
|
os.path.abspath(save_dir),
|
|
os.path.abspath(base_dir + dic[1] + '/' + dic[2]),
|
|
)
|