Adding checks to make sure stress-test doesn't run during game

This commit is contained in:
Vishal 2024-09-28 03:17:33 +05:30 committed by GitHub
parent 918988f5cd
commit 073013c778
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -8,7 +8,7 @@ import logging
from typing import cast, override
import bauiv1 as bui
import bascenev1 as bs
class BenchmarksAndStressTestsWindow(bui.MainWindow):
"""Window for launching benchmarks or stress tests."""
@ -372,16 +372,22 @@ class BenchmarksAndStressTestsWindow(bui.MainWindow):
bui.app.classic.run_media_reload_benchmark()
def _stress_test_pressed(self) -> None:
from bascenev1lib.mainmenu import MainMenuActivity
if bui.app.classic is None:
logging.warning('stress-test requires classic')
return
bui.app.classic.run_stress_test(
playlist_type=self._stress_test_game_type,
playlist_name=cast(
str, bui.textwidget(query=self._stress_test_playlist_name_field)
),
player_count=self._stress_test_player_count,
round_duration=self._stress_test_round_duration,
)
bui.containerwidget(edit=self._root_widget, transition='out_right')
activity = bs.get_foreground_host_activity()
if isinstance(activity, MainMenuActivity):
bui.app.classic.run_stress_test(
playlist_type=self._stress_test_game_type,
playlist_name=cast(
str, bui.textwidget(query=self._stress_test_playlist_name_field)
),
player_count=self._stress_test_player_count,
round_duration=self._stress_test_round_duration,
)
bui.containerwidget(edit=self._root_widget, transition='out_right')
else:
bui.screenmessage(bui.Lstr(value='Already present in another activity.'))