Commit 0cdb8b9e authored by JunHyung An's avatar JunHyung An
Browse files

Make Screenshots_YYMMDD_HHmmss folder / do not need to check exist

parent 6f08f3a7
Loading
Loading
Loading
Loading
+3 −14
Original line number Diff line number Diff line
@@ -92,20 +92,9 @@ class MultipleTest(unittest.TestCase):
            self.max_connections = 50
            self.ramp_up_time = 30 * 60

        # print("\n8. [선택] 스크린샷 저장 위치 (기본값: screenshots):")
        # self.screenshot_base_dir = input().strip() or "screenshots"
        # Check if the "screenshots" folder exists
        self.screenshots_folder = "screenshots"
        if os.path.exists(self.screenshots_folder):

            # Get the current date and time in YYMMDD_HHmm format
        # Make Screenshots_YYMMDD_HHmmss folder
        timestamp = datetime.now().strftime("%y%m%d_%H%M%S")
            new_folder_name = f"{self.screenshots_folder}_{timestamp}"

            # Rename the existing folder
            os.rename(self.screenshots_folder, new_folder_name)

        # Create a new "screenshots" folder and set the base directory for it
        self.screenshots_folder = f"Screenshots_{timestamp}"
        os.makedirs(self.screenshots_folder, exist_ok=True)
        self.screenshot_base_dir = self.screenshots_folder