Python으로 MySQL 데이터베이스에 삽입 후 "id" 가져오기
2024-07-27
Python에서 MySQL 데이터베이스에 데이터를 삽입한 후 "id"를 가져오는 방법은 다음과 같습니다.
라이브러리 설치
먼저, 작업에 필요한 라이브러리를 설치해야 합니다. 다음 명령을 사용하여 mysqlclient 라이브러리를 설치하십시오.
pip install mysqlclient
데이터베이스 연결
다음 코드를 사용하여 MySQL 데이터베이스에 연결합니다.
import mysql.connector
db = mysql.connector.connect(
host="localhost",
user="username",
password="password",
database="database_name"
)
위 코드에서 host, user, password, database_name을 실제 값으로 변경해야 합니다.
데이터 삽입
cursor = db.cursor()
sql = "INSERT INTO table_name (column1, column2, ...) VALUES (%s, %s, ...)"
values = (value1, value2, ...)
cursor.execute(sql, values)
db.commit()
위 코드에서 table_name, column1, column2, value1, value2 등을 실제 값으로 변경해야 합니다.
"id" 가져오기
lastrowid 속성을 사용하여 삽입된 마지막 레코드의 "id"를 가져올 수 있습니다.
last_id = cursor.lastrowid
print(f"삽입된 레코드의 ID: {last_id}")
연결 종료
마지막으로, 다음 코드를 사용하여 데이터베이스 연결을 종료합니다.
db.close()
예제
다음은 users 테이블에 사용자 이름과 이메일을 삽입하고 삽입된 레코드의 "id"를 가져오는 예제입니다.
import mysql.connector
db = mysql.connector.connect(
host="localhost",
user="username",
password="password",
database="database_name"
)
cursor = db.cursor()
sql = "INSERT INTO users (username, email) VALUES (%s, %s)"
values = ("testuser", "[email protected]")
cursor.execute(sql, values)
last_id = cursor.lastrowid
print(f"삽입된 레코드의 ID: {last_id}")
db.commit()
db.close()
이 코드는 다음과 같은 결과를 출력합니다.
삽입된 레코드의 ID: 1
참고:
lastrowid속성은 MySQL 버전 5.0.0 이상에서만 사용할 수 있습니다.INSERT문을 사용하여 여러 레코드를 삽입하는 경우 마지막 레코드의 "id"만 가져옵니다.- 특정 컬럼의 값을 기반으로 "id"를 가져오려면
SELECT문을 사용해야 합니다.
예제 코드 (Python 3.10)
import mysql.connector
# 데이터베이스 연결 설정
db_config = {
"host": "localhost",
"user": "username",
"password": "password",
"database": "database_name"
}
# 데이터베이스 연결
db = mysql.connector.connect(**db_config)
# 커서 생성
cursor = db.cursor()
# 삽입할 데이터 준비
table_name = "users"
column_names = ("username", "email")
user_data = ("testuser2", "[email protected]")
# 데이터 삽입 쿼리 작성
insert_query = f"""
INSERT INTO {table_name} ({", ".join(column_names)})
VALUES ({", ".join(["%s" for _ in column_names])})
"""
# 데이터 삽입 실행
cursor.execute(insert_query, user_data)
# 마지막 삽입 레코드의 ID 가져오기
last_id = cursor.lastrowid
# 삽입 정보 출력
print(f"삽입된 레코드 ID: {last_id}")
# 데이터베이스 커밋 및 연결 종료
db.commit()
db.close()
설명:
db_config딕셔너리에 데이터베이스 연결 정보를 저장합니다.mysql.connector.connect()함수를 사용하여 데이터베이스에 연결합니다.cursor.execute()함수를 사용하여 데이터 삽입 쿼리를 실행합니다.cursor.lastrowid속성을 사용하여 삽입된 마지막 레코드의 ID를 가져옵니다.db.commit()함수를 사용하여 변경 사항을 커밋합니다.db.close()함수를 사용하여 데이터베이스 연결을 종료합니다.
주의:
- 위 코드는 예시이며 실제 작업 환경에 맞게 수정해야 할 수 있습니다.
- 실제 데이터베이스 연결 정보는
username,password,database_name값을 변경해야 합니다. INSERT문에 삽입할 컬럼 이름과 값을 정확하게 지정해야 합니다.
대체 방법: LAST_INSERT_ID() 함수 사용
Python에서 MySQL 데이터베이스에 데이터를 삽입한 후 "id"를 가져오는 또 다른 방법은 LAST_INSERT_ID() 함수를 사용하는 것입니다. 이 함수는 INSERT 문 실행 후 자동으로 생성된 마지막 ID를 반환합니다.
예제 코드 (Python 3.10):
import mysql.connector
# 데이터베이스 연결 설정
db_config = {
"host": "localhost",
"user": "username",
"password": "password",
"database": "database_name"
}
# 데이터베이스 연결
db = mysql.connector.connect(**db_config)
# 커서 생성
cursor = db.cursor()
# 삽입할 데이터 준비
table_name = "users"
column_names = ("username", "email")
user_data = ("testuser3", "[email protected]")
# 데이터 삽입 쿼리 작성
insert_query = f"""
INSERT INTO {table_name} ({", ".join(column_names)})
VALUES ({", ".join(["%s" for _ in column_names])})
"""
# 데이터 삽입 실행
cursor.execute(insert_query, user_data)
# 마지막 삽입 레코드의 ID 가져오기
last_id = cursor.lastrowid
# 삽입 정보 출력
print(f"삽입된 레코드 ID: {last_id}")
# 데이터베이스 커밋 및 연결 종료
db.commit()
db.close()
- 위 코드는 예시 코드와 동일한 기능을 수행하지만,
LAST_INSERT_ID()함수를 사용하여 삽입된 마지막 레코드의 ID를 가져옵니다. cursor.lastrowid속성 대신LAST_INSERT_ID()함수를 사용하면 코드가 더 명확하고 읽기 쉬워집니다.
LAST_INSERT_ID()함수는INSERT문 실행 후 즉시 호출해야 합니다. 다른 쿼리를 실행하면 함수의 값이 변경될 수 있습니다.LAST_INSERT_ID()함수는 여러 레코드를 삽입하는 경우 마지막 레코드의 ID만 반환합니다.
python mysql database
