[python] Psycopg2를 사용하여 데이터베이스 테이블의 제약 조건을 설정하는 방법은?

22 Nov 2023

python

Psycopg2는 Python에서 PostgreSQL 데이터베이스에 연결하고 상호 작용하기 위한 강력한 라이브러리입니다. 이 라이브러리를 사용하여 PostgreSQL 데이터베이스 테이블에 제약 조건을 설정하는 방법을 알아보겠습니다.

먼저, Psycopg2를 설치해야 합니다. 다음 명령을 사용하여 설치할 수 있습니다:

pip install psycopg2

이제 Psycopg2를 사용하여 PostgreSQL 데이터베이스에 연결할 준비가 되었습니다. 다음 예제 코드에서는 ‘books’라는 테이블에 ‘title’과 ‘author’라는 두 개의 열을 생성하고, ‘title’ 열을 고유하도록 제약 조건을 설정하는 방법을 보여줍니다:

import psycopg2

# 연결 설정
conn = psycopg2.connect(host="localhost", database="mydatabase", user="myuser", password="mypassword")
cur = conn.cursor()

# 테이블 생성
cur.execute("""
    CREATE TABLE books (
        id SERIAL PRIMARY KEY,
        title VARCHAR(255) UNIQUE NOT NULL,
        author VARCHAR(255) NOT NULL
    )
""")

# 변경 사항 커밋
conn.commit()

# 연결 닫기
cur.close()
conn.close()

위의 예제 코드에서 ‘title’ 열에 ‘UNIQUE NOT NULL’ 제약 조건을 설정했습니다. 이를 통해 ‘title’ 열에는 중복되지 않는 고유한 값을 가지도록 제약이 적용됩니다.

다른 유형의 제약 조건을 설정하기 위해서는 PostgreSQL 문서에서 더 자세한 정보를 찾아볼 수 있습니다. Psycopg2 문서에도 더 많은 예제와 정보가 있으니 참고하기 바랍니다.

참고 문서:

Psycopg2 문서: https://www.psycopg.org/docs/
PostgreSQL 문서: https://www.postgresql.org/docs/