[Python] SQLite와 파이썬 연동하기

파이썬에서 SQLite를 연동하여 사용하는 방법에 대해 자세히 알아보도록 하겠습니다. SQLite는 가볍고 간단한 데이터베이스 시스템으로 파이썬과 함께 사용하기 아주 좋습니다.

SQLite란?

SQLite는 서버가 필요 없는 파일 기반의 경량 데이터베이스 입니다. 단일 파일에 모든 데이터를 저장하기 때문에 관리가 쉽고, 특히 소규모 프로젝트나 모바일 애플리케이션에 많이 사용 됩니다.

파이썬에서 SQLite 사용 준비하기

파이썬에선 SQLite를 사용하기 위해 별도의 설치가 필요 없다는 점이 매우 좋아요! 파이썬에는 'sqlite3' 모듈이 기본으로 내장되어 있기 때문 입니다. 그래서 우리는 바로 사용을 시작할 수 있답니다.

 

SQLite 데이터베이스 연결하기

먼저 SQLite 데이터베이스에 연결해보도록 하겠습니다. 아래 코드를 작성해보세요.

실행 결과

import sqlite3

# 데이터베이스 연결 (없으면 새로 생성됩니다)
conn = sqlite3.connect('my_database.db')

# 커서 객체 생성
cursor = conn.cursor()

print("데이터베이스에 성공적으로 연결되었습니다!")

이 코드를 실행하면 'my_database.db'라는 파일이 생성되고, 데이터베이스에 연결됩니다. 만약 이미 파일이 있다면, 그 파일에 연결됩니다.

 

테이블 생성하기

이제 데이터베이스에 테이블을 만들어 볼까요? 예를 들어, 간단한 사용자 정보를 저장하는 테이블을 만들어 보겠습니다.

# 테이블 생성
cursor.execute('''CREATE TABLE IF NOT EXISTS users
                  (id INTEGER PRIMARY KEY,
                   name TEXT NOT NULL,
                   email TEXT UNIQUE NOT NULL,
                   age INTEGER)''')

print("users 테이블이 성공적으로 생성되었습니다!")

 

데이터 삽입하기

테이블을 만들었으니, 이제 데이터를 넣어볼 차례 입니다.

# 데이터 삽입
cursor.execute("INSERT INTO users (name, email, age) VALUES (?, ?, ?)",
               ('홍길동', 'hong@example.com', 30))
cursor.execute("INSERT INTO users (name, email, age) VALUES (?, ?, ?)",
               ('김철수', 'kim@example.com', 25))

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

print("데이터가 성공적으로 삽입되었습니다!")

여기서 '?'는 파라미터의 위치를 나타 냅니다. 이렇게 하면 SQL 인젝션 공격을 방지할 수 있어서 안전 합니다.

 

데이터 조회하기

이제 우리가 넣은 데이터를 조회해 볼까요?

# 데이터 조회
cursor.execute("SELECT * FROM users")
users = cursor.fetchall()

print("users 테이블의 모든 데이터:")
for user in users:
    print(user)

 

데이터 수정하기

데이터를 수정하는 것도 어렵지 않아요. 예를 들어, 홍길동의 나이를 31로 바꿔볼까요?

# 데이터 수정
cursor.execute("UPDATE users SET age = ? WHERE name = ?", (31, '홍길동'))
conn.commit()

print("홍길동의 나이가 성공적으로 수정되었습니다!")

 

데이터 삭제하기

마지막으로, 데이터 삭제하는 방법도 알아보겠습니다.

# 데이터 삭제
cursor.execute("DELETE FROM users WHERE name = ?", ('김철수',))
conn.commit()

print("김철수의 데이터가 성공적으로 삭제되었습니다!")

 

연결 종료하기

작업이 모두 끝났다면, 반드시 데이터베이스 연결을 종료해야 합니다.

# 연결 종료
conn.close()

print("데이터베이스 연결이 종료되었습니다.")

 

이렇게 해서 우리는 파이썬에서 SQLite를 사용하는 기본적인 방법들을 모두 알아보았습니다. 정말 간단하죠? SQLite는 복잡한 설정 없이도 빠르게 데이터베이스 작업을 할 수 있어 매우 유용합니다.

 

여러분도 이제 파이썬 프로젝트에서 간단한 데이터베이스 작업이 필요할때 SQLite를 사용해 보세요. 작은 프로젝트부터 시작해서 점점 더 복잡한 데이터베이스 작업으로 나아갈 수 있을 거에요.