[python] Peewee에서 데이터 삽입하기

Peewee는 파이썬에서 사용할 수 있는 간단하고 가벼운 ORM(Object-Relational Mapping) 라이브러리입니다. 이를 통해 데이터베이스와의 상호작용을 쉽게 할 수 있습니다. 이번 글에서는 Peewee를 사용하여 데이터를 삽입하는 방법에 대해 알아보겠습니다.

Peewee 설치하기

Peewee를 사용하기 위해 먼저 설치해야 합니다. 다음 명령을 사용하여 Peewee를 설치할 수 있습니다.

pip install peewee

데이터베이스 연결하기

Peewee를 사용하여 데이터를 삽입하기 전에, 먼저 데이터베이스에 연결해야 합니다. 다음은 SQLite 데이터베이스에 연결하는 예시입니다.

from peewee import SqliteDatabase, Model, CharField

db = SqliteDatabase('mydatabase.db')

class BaseModel(Model):
    class Meta:
        database = db

# 모델 정의 및 테이블 생성을 위한 코드 작성

db.connect()

위의 코드에서 SqliteDatabase 클래스는 사용할 데이터베이스의 종류를 지정하고, Model 클래스는 모델을 정의할 때 상속받아 사용합니다. 데이터베이스에 접속하기 위해 db.connect()를 호출해야 합니다.

데이터 삽입하기

Peewee를 사용하여 데이터를 삽입하기 위해서는 먼저 모델을 정의해야 합니다. 모델은 데이터베이스 안의 테이블과 매핑되는 클래스입니다. 다음은 User 모델을 정의하는 예시입니다.

class User(BaseModel):
    username = CharField()
    email = CharField()

위의 예시에서 User 모델은 BaseModel을 상속하고 있습니다. usernameemail은 데이터베이스 테이블의 칼럼과 매칭되는 필드입니다.

데이터를 삽입하기 위해서는 다음과 같이 create() 메소드를 사용합니다.

# User 데이터 생성
user = User.create(username='Alice', email='alice@example.com')

위의 코드에서는 User 모델의 인스턴스를 생성하고, create() 메소드를 호출하여 데이터를 삽입했습니다. usernameemail은 모델의 필드와 매칭되는 파라미터입니다.

데이터 삽입 결과 확인하기

데이터를 삽입한 후에는 삽입된 데이터를 확인할 수 있습니다. 다음은 User 모델의 모든 데이터를 조회하는 예시입니다.

# 모든 User 데이터 조회
users = User.select()
for user in users:
    print(user.username, user.email)

위의 코드에서는 User.select()를 호출하여 User 테이블의 모든 데이터를 가져옵니다. 각 데이터는 user 변수에 할당되고, usernameemail 필드에 접근하여 값을 출력합니다.

정리

Peewee를 사용하여 데이터를 삽입하는 방법에 대해 알아보았습니다. 데이터를 삽입하기 위해서는 먼저 데이터베이스에 연결하고, 모델을 정의한 후 create() 메소드를 사용하여 데이터를 삽입할 수 있습니다. 데이터 삽입 후에는 삽입된 데이터를 확인할 수 있습니다. Peewee의 자세한 사용법은 공식 문서를 참고하시기 바랍니다.

참고: Peewee 공식 문서