[python] PyMySQL을 이용하여 데이터베이스의 테이블의 외래 키를 설정하는 방법을 알아본다.

PyMySQL은 Python에서 MySQL 데이터베이스를 다룰 수 있는 모듈로, 외래 키를 설정하는 기능도 제공합니다. 이번 글에서는 PyMySQL을 사용하여 데이터베이스의 테이블에 외래 키를 설정하는 방법을 알아보겠습니다.

1. PyMySQL 설치하기

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

pip install PyMySQL

2. 데이터베이스 연결하기

PyMySQL을 사용하기 위해 데이터베이스에 연결해야 합니다. 다음과 같은 코드를 사용하여 데이터베이스에 연결할 수 있습니다.

import pymysql

# 데이터베이스 연결
connection = pymysql.connect(
    host='localhost',
    user='root',
    password='password',
    db='database_name'
)

위 코드에서 host, user, password, db 매개변수를 실제 데이터베이스 정보에 맞게 수정해야 합니다.

3. 외래 키 설정하기

테이블에 외래 키를 설정하려면 SQL 문을 사용해야 합니다. PyMySQL을 이용하여 SQL 문을 실행하는 방법은 다음과 같습니다.

import pymysql

# 데이터베이스 연결
connection = pymysql.connect(
    host='localhost',
    user='root',
    password='password',
    db='database_name'
)

# 커서 생성
cursor = connection.cursor()

# 외래 키 설정 SQL 문 실행
sql = "ALTER TABLE 테이블명 ADD FOREIGN KEY (컬럼명) REFERENCES 참조할_테이블(참조할_컬럼)"
cursor.execute(sql)

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

# 연결 종료
connection.close()

위 코드에서 테이블명, 컬럼명, 참조할_테이블, 참조할_컬럼은 실제 데이터베이스 테이블 및 컬럼 이름에 맞게 수정해야 합니다.

4. 외래 키 제약 조건 설정하기

PyMySQL을 사용하여 외래 키 제약 조건을 설정할 수도 있습니다. 다음과 같은 코드를 사용하여 외래 키 제약 조건을 설정할 수 있습니다.

import pymysql

# 데이터베이스 연결
connection = pymysql.connect(
    host='localhost',
    user='root',
    password='password',
    db='database_name'
)

# 커서 생성
cursor = connection.cursor()

# 외래 키 제약 조건 설정 SQL 문 실행
sql = "ALTER TABLE 테이블명 ADD CONSTRAINT 제약조건명 FOREIGN KEY (컬럼명) REFERENCES 참조할_테이블(참조할_컬럼)"
cursor.execute(sql)

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

# 연결 종료
connection.close()

위 코드에서 테이블명, 컬럼명, 참조할_테이블, 참조할_컬럼, 제약조건명은 실제 데이터베이스 테이블, 컬럼 및 제약 조건에 맞게 수정해야 합니다.

참고 자료