sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-12 00:28:20 +08:00
Code Activity

Fix: Research reset save research_projects in stage folder not base folder

Browse Source
This commit is contained in:
LmeSzinc
2020-12-11 22:19:12 +08:00
parent 1f6948d121
commit 16ea207cde
2 changed files with 10 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
module/device/screenshot.py
View File

@@ -73,12 +73,13 @@ class Screenshot(AScreenCap):
return self.image
def save_screenshot(self, genre='items', interval=None):
def save_screenshot(self, genre='items', interval=None, to_base_folder=False):
"""Save a screenshot. Use millisecond timestamp as file name.
Args:
genre (str, optional): Screenshot type.
interval (int, float): Seconds between two save. Saves in the interval will be dropped.
to_base_folder (bool): If save to base folder.
Returns:
bool: True if save succeed.
@@ -91,7 +92,8 @@ class Screenshot(AScreenCap):
fmt = 'png'
file = '%s.%s' % (int(now * 1000), fmt)
folder = os.path.join(self.config.SCREEN_SHOT_SAVE_FOLDER, genre)
folder = self.config.SCREEN_SHOT_SAVE_FOLDER_BASE if to_base_folder else self.config.SCREEN_SHOT_SAVE_FOLDER
folder = os.path.join(folder, genre)
if not os.path.exists(folder):
os.mkdir(folder)