PyMySQL은 Python에서 MySQL 데이터베이스에 접근할 수 있는 모듈이다. 이 모듈을 사용하여 데이터베이스의 테이블에 제약 조건을 설정하는 방법을 알아보자.

1. PyMySQL 설치

먼저, PyMySQL 모듈을 설치해야 한다. 아래 명령을 사용하여 PyMySQL을 설치한다.

pip install PyMySQL

2. 데이터베이스 연결

데이터베이스에 접속하기 위해 PyMySQL을 사용하여 연결해야 한다. 연결을 위해 호스트, 사용자명, 비밀번호, 데이터베이스 이름 등을 알고 있어야 한다.

import pymysql

conn = pymysql.connect(
    host='localhost',
    user='root',
    password='password',
    database='mydb'
)

3. 테이블 생성하기

데이터베이스에 테이블을 생성한다. 생성된 테이블에 제약 조건을 설정할 수 있다.

아래는 ‘users’라는 테이블을 생성하고, ‘id’라는 필드에 PRIMARY KEY 제약 조건을 설정하는 예제이다.

cursor = conn.cursor()

create_table_query = '''
    CREATE TABLE users (
        id INT PRIMARY KEY,
        name VARCHAR(50),
        email VARCHAR(50)
    )
'''

cursor.execute(create_table_query)

4. 제약 조건 설정하기

테이블이 생성되면 제약 조건을 설정할 수 있다. 아래는 ‘users’ 테이블에 FOREIGN KEY 제약 조건을 설정하는 예제이다.

alter_table_query = '''
    ALTER TABLE users
    ADD CONSTRAINT fk_user FOREIGN KEY (department_id) REFERENCES departments(id)
'''

cursor.execute(alter_table_query)

5. 데이터베이스 연결 종료

작업이 끝나면 데이터베이스 연결을 종료해야 한다.

conn.close()

위의 예제를 참고하여 PyMySQL을 사용하여 데이터베이스의 테이블에 제약 조건을 설정할 수 있다. PyMySQL을 사용하면 데이터베이스 작업을 손쉽게 처리할 수 있다.

참고: PyMySQL 공식 문서