[python] 테이블 생성 및 관리

이 기술 블로그에서는 Python을 사용하여 데이터베이스에서 테이블을 생성하고 관리하는 방법에 대해 다룹니다.

목차

  1. 데이터베이스 연결
  2. 테이블 생성
  3. 데이터 추가
  4. 데이터 조회
  5. 데이터 수정
  6. 데이터 삭제

데이터베이스 연결

Python에서 데이터베이스를 사용하려면 sqlite3 모듈을 사용하여 데이터베이스에 연결해야 합니다.

import sqlite3

# 데이터베이스에 연결
conn = sqlite3.connect('example.db')

테이블 생성

이어서, 테이블을 생성하기 위해 SQL 쿼리를 실행할 수 있습니다.

# 커서 생성
c = conn.cursor()

# 테이블 생성
c.execute('''CREATE TABLE stocks
             (date text, trans text, symbol text, qty real, price real)''')

# 연결 종료
conn.close()

데이터 추가

테이블을 생성한 후에는 데이터를 추가할 수 있습니다.

# 데이터베이스에 연결
conn = sqlite3.connect('example.db')
c = conn.cursor()

# 데이터 추가
c.execute("INSERT INTO stocks VALUES ('2022-01-05', 'BUY', 'RHAT', 100, 35.14)")

# 변경 사항 저장
conn.commit()

# 연결 종료
conn.close()

데이터 조회

추가한 데이터를 조회할 때는 다음과 같이 쿼리를 실행합니다.

# 데이터베이스에 연결
conn = sqlite3.connect('example.db')
c = conn.cursor()

# 데이터 조회
c.execute("SELECT * FROM stocks")
print(c.fetchall())

# 연결 종료
conn.close()

데이터 수정

기존 데이터를 수정하려면 UPDATE 쿼리를 사용할 수 있습니다.

# 데이터베이스에 연결
conn = sqlite3.connect('example.db')
c = conn.cursor()

# 데이터 수정
c.execute("UPDATE stocks SET qty = 150 WHERE symbol = 'RHAT'")

# 변경 사항 저장
conn.commit()

# 연결 종료
conn.close()

데이터 삭제

특정 데이터를 삭제하고 싶을 경우 DELETE 쿼리를 사용할 수 있습니다.

# 데이터베이스에 연결
conn = sqlite3.connect('example.db')
c = conn.cursor()

# 데이터 삭제
c.execute("DELETE FROM stocks WHERE symbol = 'RHAT'")

# 변경 사항 저장
conn.commit()

# 연결 종료
conn.close()

이로써 Python을 사용하여 테이블을 생성하고 데이터를 추가, 조회, 수정, 삭제하는 방법에 대해 알아보았습니다.

참고문헌: