[python] PyYAML을 이용하여 YAML 파일에서 특정 필드 값을 변경하기
YAML은 사람과 기계가 모두 이해하기 쉬운 데이터 직렬화 양식으로 많이 사용됩니다. 때때로 우리는 YAML 파일의 특정 필드 값을 찾아서 수정해야 할 때가 있습니다. 이를 위해 Python의 PyYAML 라이브러리를 사용할 수 있습니다.
PyYAML 설치하기
PyYAML을 사용하기 위해 먼저 설치해야 합니다. 다음 명령어를 사용하여 설치하세요.
pip install PyYAML
YAML 파일 읽기 및 특정 필드 값 변경하기
다음과 같은 YAML 파일이 있다고 가정해봅시다.
person:
name: John
age: 30
job: Engineer
여기서 “age” 필드 값을 30에서 35로 변경하려고 합니다.
import yaml
# YAML 파일 읽기
with open('example.yaml', 'r') as file:
data = yaml.safe_load(file)
# 특정 필드 값 변경하기
data['person']['age'] = 35
# 변경된 데이터를 YAML 파일로 저장하기
with open('example.yaml', 'w') as file:
yaml.dump(data, file)
위 예제에서는 yaml.safe_load() 함수를 사용하여 YAML 파일을 읽어옵니다. 그런 다음, 필요한 필드 값에 접근하여 값을 변경하고 yaml.dump() 함수를 사용하여 변경된 데이터를 다시 YAML 파일에 저장합니다.
주의사항
- PyYAML을 사용하여 YAML 파일을 읽을 때는
safe_load()를 사용해야 합니다.load()함수는 보안 취약점을 가지고 있을 수 있으므로 사용을 지양해야 합니다.