Update 2025-12-19 20:23:59

data/scripts/03-tsr_log.py

@@ -4,6 +4,14 @@ import time
 from dotenv import load_dotenv
 import sys
 from multiprocessing import Pool
+import logging
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s [INFO] root: %(message)s',
+    datefmt='%Y-%m-%d %H:%M:%S'
+)
 
 # 환경 변수 로드
 load_dotenv()  # .env 파일에서 환경 변수 로드
@@ -14,26 +22,27 @@ IDRAC_USER, IDRAC_PASS = os.getenv("IDRAC_USER"), os.getenv("IDRAC_PASS")
 # IP 주소 파일 로드 함수
 def load_ip_file(ip_file_path):
     if not os.path.isfile(ip_file_path):
-        sys.exit(f"IP file {ip_file_path} does not exist.")
+        logging.error(f"IP file {ip_file_path} does not exist.")
+        sys.exit(1)
     with open(ip_file_path, "r") as file:
         return [line.strip() for line in file if line.strip()]
 
 # iDRAC 정보를 가져오는 함수 정의
 def fetch_idrac_info(idrac_ip):
-    print(f"Collecting TSR report for iDRAC IP: {idrac_ip}")
+    logging.info(f"Collecting TSR report for iDRAC IP: {idrac_ip}")
     try:
         result = subprocess.run(
             ["racadm", "-r", idrac_ip, "-u", IDRAC_USER, "-p", IDRAC_PASS, "techsupreport", "collect"],
             capture_output=True,
             text=True
         )
-        print(
-            f"Successfully collected TSR report for {idrac_ip}"
-            if result.returncode == 0
-            else f"Failed to collect TSR report for {idrac_ip}: {result.stderr.strip()}"
-        )
+        if result.returncode == 0:
+            logging.info(f"Successfully collected TSR report for {idrac_ip}")
+        else:
+            logging.error(f"Failed to collect TSR report for {idrac_ip}: {result.stderr.strip()}")
+
     except Exception as e:
-        print(f"Exception occurred for {idrac_ip}: {e}")
+        logging.error(f"Exception occurred for {idrac_ip}: {e}")
 
 # 메인 함수
 if __name__ == "__main__":
@@ -48,4 +57,4 @@ if __name__ == "__main__":
         pool.map(fetch_idrac_info, ip_list)
 
     elapsed_time = time.time() - start_time
-    print(f"설정 완료. 수집 완료 시간: {int(elapsed_time // 3600)} 시간, {int((elapsed_time % 3600) // 60)} 분, {int(elapsed_time % 60)} 초.")
+    logging.info(f"설정 완료. 수집 완료 시간: {int(elapsed_time // 3600)} 시간, {int((elapsed_time % 3600) // 60)} 분, {int(elapsed_time % 60)} 초.")