Initial project upload
This commit is contained in:
Kim.KANGHEE
@@ -0,0 +1,192 @@
|
||||
#!/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()
|
||||
Reference in New Issue
Block a user