[python] PyYAML을 사용하여 YAML 파일을 데이터베이스에 저장하기
YAML은 인간이 쉽게 읽고 쓸 수 있는 데이터 직렬화 형식입니다. 이 형식은 데이터를 계층적으로 구조화하기에 매우 편리합니다. Python에서는 PyYAML이라는 라이브러리를 사용하여 YAML 파일을 데이터베이스에 저장할 수 있습니다.
1. PyYAML 설치하기
먼저, PyYAML을 설치해야 합니다. 다음 명령을 터미널에서 실행하여 PyYAML을 설치할 수 있습니다.
pip install PyYAML
2. YAML 파일 읽기
데이터베이스에 저장할 YAML 파일을 읽어와야 합니다. YAML 파일은 텍스트 파일이기 때문에 일반적인 파일 읽기 방법을 사용하면 됩니다. 예를 들어, 다음과 같은 YAML 파일이 있다고 가정해 봅시다.
---
- name: John Doe
age: 25
- name: Jane Smith
age: 30
다음 코드는 “data.yml” 파일을 읽어와 YAML 데이터를 파싱하는 예제입니다.
import yaml
with open('data.yml', 'r') as file:
data = yaml.safe_load(file)
3. 데이터베이스에 저장하기
이제 YAML 데이터를 데이터베이스에 저장할 차례입니다. 데이터베이스 종류에 따라 저장하는 방법이 다를 수 있습니다. 여기서는 간단한 예시로 SQLite 데이터베이스에 저장하는 방법을 알아보겠습니다.
import sqlite3
# SQLite 데이터베이스에 연결
conn = sqlite3.connect('database.db')
cursor = conn.cursor()
# 데이터베이스에 YAML 데이터 저장
for item in data:
cursor.execute("INSERT INTO my_table (name, age) VALUES (?, ?)", (item['name'], item['age']))
# 변경사항 커밋
conn.commit()
# 연결 종료
conn.close()
위의 코드에서 my_table에는 필요한 테이블 이름을 사용하시면 됩니다. SQLite 외의 다른 데이터베이스를 사용한다면 해당 데이터베이스의 라이브러리와 문법에 맞게 코드를 수정해야 합니다.