Commit 6f08f3a7 authored by JunHyung An's avatar JunHyung An
Browse files

Prevent overwrite screenshots

parent 73c1543b
Loading
Loading
Loading
Loading
+15 −3
Original line number Diff line number Diff line
@@ -6,6 +6,7 @@ import time
import unittest
import random
import os
from datetime import datetime

from selenium import webdriver
from selenium.webdriver.common.by import By
@@ -93,9 +94,20 @@ class MultipleTest(unittest.TestCase):

        # print("\n8. [선택] 스크린샷 저장 위치 (기본값: screenshots):")
        # self.screenshot_base_dir = input().strip() or "screenshots"
        self.screenshot_base_dir = "screenshots"
        if not os.path.exists(self.screenshot_base_dir):
            os.makedirs(self.screenshot_base_dir)
        # 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
            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
        os.makedirs(self.screenshots_folder, exist_ok=True)
        self.screenshot_base_dir = self.screenshots_folder

    def setUp(self):
        self._get_user_inputs()