Kim.KANGHEE
193 lines
7.3 KiB
Python
193 lines
7.3 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
통합 서버 정보 수집 스크립트
|
|
프로파일 기반으로 BIOS/iDRAC/RAID 설정 수집
|
|
"""
|
|
import sys
|
|
import yaml
|
|
from pathlib import Path
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
import logging
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|
from core.idrac_client import IDRACClient
|
|
from core.parsers import InfoParser
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s [INFO] root: %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S'
|
|
)
|
|
|
|
|
|
def load_profile(profile_name: str) -> dict:
|
|
"""프로파일 로드 및 상속 처리"""
|
|
profile_file = Path(__file__).parent.parent / "profiles" / "server_info_profiles.yaml"
|
|
|
|
if not profile_file.exists():
|
|
logging.error(f"프로파일 파일 없음: {profile_file}")
|
|
sys.exit(1)
|
|
|
|
with open(profile_file, 'r', encoding='utf-8') as f:
|
|
profiles = yaml.safe_load(f)['profiles']
|
|
|
|
profile = profiles.get(profile_name)
|
|
if not profile:
|
|
logging.error(f"프로파일 '{profile_name}' 없음. 사용 가능: {list(profiles.keys())}")
|
|
sys.exit(1)
|
|
|
|
# extends 처리
|
|
if 'extends' in profile:
|
|
base = profiles[profile['extends']].copy()
|
|
# 리스트 병합
|
|
for key in ['firmware', 'bios_settings', 'idrac_settings', 'raid_settings', 'custom_items']:
|
|
if key in profile:
|
|
base_items = base.get(key, [])
|
|
profile_items = profile[key]
|
|
# 중복 제거하면서 병합
|
|
base[key] = base_items + profile_items
|
|
base.update({k: v for k, v in profile.items() if k not in ['firmware', 'bios_settings', 'idrac_settings', 'raid_settings', 'custom_items']})
|
|
profile = base
|
|
|
|
return profile
|
|
|
|
|
|
def collect_server_info_for_ip(ip: str, profile: dict, output_dir: Path) -> tuple:
|
|
"""단일 IP에 대한 서버 정보 수집"""
|
|
try:
|
|
client = IDRACClient(ip)
|
|
parser = InfoParser()
|
|
|
|
# 서비스 태그 추출
|
|
sysinfo = client.get_sysinfo()
|
|
svc_tag = parser.extract_service_tag(sysinfo)
|
|
if not svc_tag:
|
|
return ip, False, "서비스 태그 추출 실패"
|
|
|
|
# 결과 파일 생성
|
|
output_file = output_dir / f"{svc_tag}.txt"
|
|
|
|
with output_file.open('w', encoding='utf-8', newline='\n') as f:
|
|
f.write(f"Dell EMC Server Bios,iDRAC,R/C Setting (SVC Tag: {svc_tag})\n\n")
|
|
|
|
# 펌웨어 버전 정보
|
|
if 'firmware' in profile:
|
|
f.write("-" * 42 + " Firmware Version 정보 " + "-" * 42 + "\n")
|
|
for idx, item in enumerate(profile['firmware'], 1):
|
|
value = _get_value_from_command(client, item['command'], item['pattern'])
|
|
f.write(f"{idx}. {item['name']} : {value}\n")
|
|
f.write("\n")
|
|
|
|
# BIOS 설정 정보
|
|
if 'bios_settings' in profile:
|
|
f.write("-" * 45 + " Bios 설정 정보 " + "-" * 46 + "\n")
|
|
for idx, item in enumerate(profile['bios_settings'], 1):
|
|
value = _get_value_from_command(client, item['command'], item['pattern'])
|
|
f.write(f"{idx:02d}. {item['name']} : {value}\n")
|
|
f.write("\n")
|
|
|
|
# iDRAC 설정 정보
|
|
if 'idrac_settings' in profile:
|
|
f.write("-" * 45 + " iDRAC 설정 정보 " + "-" * 45 + "\n")
|
|
for idx, item in enumerate(profile['idrac_settings'], 1):
|
|
value = _get_value_from_command(client, item['command'], item['pattern'])
|
|
f.write(f"{idx:02d}. {item['name']} : {value}\n")
|
|
f.write("\n")
|
|
|
|
# RAID 설정 정보
|
|
if 'raid_settings' in profile:
|
|
f.write("-" * 45 + " Raid 설정 정보 " + "-" * 46 + "\n")
|
|
for idx, item in enumerate(profile['raid_settings'], 1):
|
|
value = _get_value_from_command(client, item['command'], item['pattern'])
|
|
f.write(f"{idx:02d}. {item['name']} : {value}\n")
|
|
f.write("\n")
|
|
|
|
# 커스텀 항목 (있을 경우)
|
|
if 'custom_items' in profile and profile['custom_items']:
|
|
f.write("-" * 45 + " 커스텀 항목 " + "-" * 46 + "\n")
|
|
for idx, item in enumerate(profile['custom_items'], 1):
|
|
value = _get_value_from_command(client, item['command'], item['pattern'])
|
|
f.write(f"{idx:02d}. {item['name']} : {value}\n")
|
|
f.write("\n")
|
|
|
|
return ip, True, f"저장: {output_file.name}"
|
|
|
|
except Exception as e:
|
|
logging.error(f"[{ip}] 오류: {e}")
|
|
return ip, False, f"오류: {e}"
|
|
|
|
|
|
def _get_value_from_command(client: IDRACClient, command: str, pattern: str) -> str:
|
|
"""racadm 명령 실행 및 값 추출"""
|
|
# 캐싱된 데이터 사용
|
|
if command == "getsysinfo":
|
|
output = client.get_sysinfo()
|
|
elif command == "hwinventory":
|
|
output = client.get_hwinventory()
|
|
else:
|
|
_, output = client.run_command(command)
|
|
|
|
# 패턴 매칭
|
|
value = InfoParser.extract_value_by_pattern(output, pattern)
|
|
return value if value else "N/A"
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) < 2:
|
|
print("Usage: python collect_server_info.py <ip_file> [profile_name]")
|
|
print("Available profiles: default, intel_server, amd_server, minimal")
|
|
sys.exit(1)
|
|
|
|
ip_file = Path(sys.argv[1])
|
|
profile_name = sys.argv[2] if len(sys.argv) > 2 else "default"
|
|
|
|
if not ip_file.exists():
|
|
logging.error(f"IP 파일 없음: {ip_file}")
|
|
sys.exit(1)
|
|
|
|
# 프로파일 로드
|
|
profile = load_profile(profile_name)
|
|
logging.info(f"프로파일 사용: {profile_name} - {profile.get('description', '')}")
|
|
|
|
# 출력 디렉토리 (data/temp/staging)
|
|
script_dir = Path(__file__).resolve().parent
|
|
# unified 디렉토리에 있으므로, data 디렉토리는 3단계 상위 (unified -> scripts -> data)
|
|
data_dir = script_dir.parent.parent
|
|
|
|
# 만약 구조가 다를 경우를 대비한 안전장치 (scripts가 아닐 경우 등)
|
|
if data_dir.name != "data":
|
|
# fallback: 현재 작업 디렉토리 기준 data 찾기 시도
|
|
data_dir = Path("data").resolve()
|
|
|
|
output_dir = data_dir / "temp" / "staging"
|
|
output_dir.mkdir(parents=True, exist_ok=True)
|
|
|
|
# IP 목록 읽기
|
|
ips = [line.strip() for line in ip_file.read_text(encoding='utf-8').splitlines() if line.strip()]
|
|
|
|
if not ips:
|
|
logging.error("IP 목록이 비어 있습니다.")
|
|
sys.exit(1)
|
|
|
|
logging.info(f"총 {len(ips)}대 처리 시작")
|
|
|
|
# 병렬 처리 (서버 정보 수집은 racadm 호출이 많아 worker 수 줄임)
|
|
ok, fail = 0, 0
|
|
with ThreadPoolExecutor(max_workers=10) as executor:
|
|
futures = {executor.submit(collect_server_info_for_ip, ip, profile, output_dir): ip for ip in ips}
|
|
|
|
for future in futures:
|
|
ip, success, msg = future.result()
|
|
if success:
|
|
logging.info(f"[OK] {ip} - {msg}")
|
|
ok += 1
|
|
else:
|
|
logging.error(f"[FAIL] {ip} - {msg}")
|
|
fail += 1
|
|
|
|
logging.info(f"완료: 성공 {ok}대 / 실패 {fail}대")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|