sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-15 21:37:22 +08:00
Code Activity

Dev: coin_statistics

Browse Source
This commit is contained in:
sui-feng-cb
2025-04-25 20:41:40 +08:00
parent 4b2cb5a1e1
commit b8c3772376
6 changed files with 122 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
module/statistics/utils.py
View File

@@ -23,7 +23,7 @@ def load_folder(folder, ext='.png'):
Image shape: width=96, height=96, channel=3, format=png.
Image name: Camel-Case, such as 'PlateGeneralT3'. Suffix in name will be ignore.
For example, 'Javelin' and 'Javelin_2' are different templates, but have same output name 'Javelin'.
ext (str): File extension.
ext (str|list[str]): File extension.
Returns:
dict: Key: str, image file base name. Value: full filepath.
@@ -34,7 +34,8 @@ def load_folder(folder, ext='.png'):
out = {}
for file in os.listdir(folder):
name, extension = os.path.splitext(file)
if extension == ext:
if (isinstance(ext, str) and extension == ext) \
or (isinstance(ext, list) and extension in ext):
out[name] = os.path.join(folder, file)
return out