Initial commit

data/scripts/PortGUID.py

@@ -0,0 +1,92 @@
+import os
+import re
+import subprocess
+from dotenv import load_dotenv
+from concurrent.futures import ThreadPoolExecutor, as_completed
+
+# .env 파일에서 사용자 이름 및 비밀번호 설정
+load_dotenv()
+IDRAC_USER = os.getenv("IDRAC_USER")
+IDRAC_PASS = os.getenv("IDRAC_PASS")
+
+def fetch_idrac_info(idrac_ip, output_dir):
+    try:
+        # 서비스 태그 가져오기 (get 제외)
+        cmd_getsysinfo = f"racadm -r {idrac_ip} -u {IDRAC_USER} -p {IDRAC_PASS} getsysinfo"
+        getsysinfo = subprocess.getoutput(cmd_getsysinfo)
+        svc_tag_match = re.search(r"SVC Tag\s*=\s*(\S+)", getsysinfo)
+        svc_tag = svc_tag_match.group(1) if svc_tag_match else None
+
+        if not svc_tag:
+            print(f"Failed to retrieve SVC Tag for IP: {idrac_ip}")
+            return
+
+        # InfiniBand.VndrConfigPage 목록 가져오기
+        cmd_list = f"racadm -r {idrac_ip} -u {IDRAC_USER} -p {IDRAC_PASS} get InfiniBand.VndrConfigPage"
+        output_list = subprocess.getoutput(cmd_list)
+
+        # InfiniBand.VndrConfigPage.<숫자> 및 Key 값 추출
+        matches = re.findall(r"InfiniBand\.VndrConfigPage\.(\d+)\s+\[Key=InfiniBand\.Slot\.(\d+)-\d+#VndrConfigPage]", output_list)
+
+        # 결과를 저장할 파일 생성
+        output_file = os.path.join(output_dir, f"{svc_tag}.txt")
+        os.makedirs(output_dir, exist_ok=True)
+
+        with open(output_file, "w") as f:
+            # 서비스 태그 저장
+            f.write(f"{svc_tag}\n")
+
+            # 모든 PortGUID를 저장할 리스트
+            hex_guid_list = []
+
+            # 각 InfiniBand.VndrConfigPage.<숫자> 처리
+            for number, slot in matches:
+                cmd_detail = f"racadm -r {idrac_ip} -u {IDRAC_USER} -p {IDRAC_PASS} get InfiniBand.VndrConfigPage.{number}"
+                output_detail = subprocess.getoutput(cmd_detail)
+
+                # PortGUID 값 추출
+                match_guid = re.search(r"PortGUID=(\S+)", output_detail)
+                port_guid = match_guid.group(1) if match_guid else "Not Found"
+
+                # Slot.<숫자>: <PortGUID> 형식으로 저장
+                f.write(f"Slot.{slot}: {port_guid}\n")
+
+                # PortGUID를 0x 형식으로 변환하여 리스트에 추가
+                if port_guid != "Not Found":
+                    hex_guid_list.append(f"0x{port_guid.replace(':', '').upper()}")
+
+            # 모든 PortGUID를 "GUID: 0x<GUID1>;0x<GUID2>" 형식으로 저장
+            if hex_guid_list:
+                f.write(f"GUID: {';'.join(hex_guid_list)}\n")
+
+    except Exception as e:
+        print(f"Error processing IP {idrac_ip}: {e}")
+
+def main(ip_file):
+    if not os.path.isfile(ip_file):
+        print(f"IP file {ip_file} does not exist.")
+        return
+
+    output_dir = "/app/idrac_info/idrac_info"
+    os.makedirs(output_dir, exist_ok=True)
+
+    with open(ip_file, "r") as file:
+        ip_addresses = [line.strip() for line in file.readlines()]
+
+    with ThreadPoolExecutor(max_workers=100) as executor:
+        future_to_ip = {executor.submit(fetch_idrac_info, ip, output_dir): ip for ip in ip_addresses}
+
+        for future in as_completed(future_to_ip):
+            try:
+                future.result()
+            except Exception as e:
+                print(f"Error processing task: {e}")
+
+if __name__ == "__main__":
+    import sys
+    if len(sys.argv) != 2:
+        print("Usage: python script.py <ip_file>")
+        sys.exit(1)
+
+    ip_file = sys.argv[1]
+    main(ip_file)