본문 바로가기
■ 칼퇴를 위한 파이썬 : 사례

파이썬으로 데이터베이스 연결하기: SQLite 사용법

by 포탈메이커 2023. 7. 11.

파이썬으로 데이터베이스 연결하기: SQLite 사용법

SQLite는 서버가 필요 없는 경량 데이터베이스로, 파일 시스템에 저장되는 싱글 파일 형태의 데이터베이스입니다. 이 블로그 포스팅에서는 파이썬을 통해 SQLite 데이터베이스에 연결하는 방법을 알아보겠습니다.

1. SQLite 설치하기

SQLite는 파이썬에 기본적으로 내장되어 있기 때문에 별도의 설치가 필요하지 않습니다. 그러므로 파이썬 환경을 설정한 후에 바로 사용할 수 있습니다.

2. SQLite 데이터베이스 생성하기

SQLite 데이터베이스를 생성하기 위해서는 SQLite에 있는 sqlite3 모듈을 import해야 합니다. 아래의 코드를 통해 데이터베이스를 생성할 수 있습니다.

```python import sqlite3

conn = sqlite3.connect('example.db') ```

위의 코드는 example.db라는 이름의 SQLite 데이터베이스를 생성하고, conn 객체를 통해 데이터베이스에 연결합니다. 만약 example.db 파일이 존재하지 않는다면, 새로운 데이터베이스 파일이 생성됩니다.

3. 데이터베이스에 테이블 생성하기

SQLite 데이터베이스 내에서 작업을 하려면 테이블을 생성해야 합니다. 테이블 생성을 위해서는 SQL 질의어인 CREATE TABLE을 사용합니다. 아래의 코드를 참고하여 테이블을 생성할 수 있습니다.

```python import sqlite3

conn = sqlite3.connect('example.db')

c = conn.cursor() c.execute('''CREATE TABLE stocks (date text, trans text, symbol text, qty real, price real)''') ```

위의 코드는 stocks라는 이름의 테이블을 생성하며, 각 열에는 date, trans, symbol, qty, price라는 이름의 컬럼을 가지고 있습니다.

4. 데이터 삽입하기

테이블이 생성되었다면 데이터를 삽입할 수 있습니다. 데이터를 삽입하기 위해서는 SQL 질의어인 INSERT INTO를 사용합니다. 아래의 코드를 참고하여 데이터를 삽입할 수 있습니다.

```python import sqlite3

conn = sqlite3.connect('example.db')

c = conn.cursor() c.execute("INSERT INTO stocks VALUES ('2006-01-05', 'BUY', 'RHAT', 100, 35.14)") ```

위의 코드는 stocks 테이블에 데이터를 삽입하는 예시입니다. 첫 번째 인자는 날짜, 두 번째 인자는 거래 유형, 세 번째 인자는 종목 코드, 네 번째 인자는 수량, 다섯 번째 인자는 가격입니다.

5. 데이터 조회하기

데이터베이스에 삽입된 데이터를 조회하려면 SQL 질의어인 SELECT를 사용해야 합니다. 아래의 코드를 참고하여 데이터를 조회할 수 있습니다.

```python import sqlite3

conn = sqlite3.connect('example.db')

c = conn.cursor() c.execute("SELECT * FROM stocks")

rows = c.fetchall()

for row in rows: print(row) ```

위의 코드는 stocks 테이블에 존재하는 모든 데이터를 조회하여 출력하는 예시입니다. fetchall() 메소드를 사용하여 모든 데이터를 한 번에 가져옵니다.

6. 데이터베이스 연결 종료하기

데이터베이스 작업이 끝났다면 데이터베이스 연결을 종료해야 합니다. 아래의 코드를 통해 연결을 종료할 수 있습니다.

```python import sqlite3

conn = sqlite3.connect('example.db')

데이터베이스 작업 수행

conn.close() ```

위의 코드는 close() 메소드를 사용하여 데이터베이스 연결을 종료합니다.

마무리

이제 파이썬을 사용하여 SQLite 데이터베이스에 연결하는 방법에 대해 알아보았습니다. 데이터베이스 생성, 테이블 생성, 데이터 삽입, 데이터 조회, 데이터베이스 연결 종료 등을 다루는 기초적인 내용을 다뤘으니 이를 기반으로 더 다양한 데이터베이스 작업을 수행해볼 수 있을 것입니다.