Python에서 SQL 사용 (1)

배경

회사의 일과 개인의 경력이 일치하지 않을 때가 있다.

2021년까지 인공지능, 데이터 사이언스 업무를 하였으나 2022년부터 다시 웹 개발을 하게 되었다.

 

어렵게 익힌 파이썬 기술은 점점 잊혀지고 백엔드 기술인 SQL은 이제 눈에 들어오는 정도이다.

두 기술 모두 중요하므로 함께 사용하기로 결정하니 마음이 편안해진다.

 

개요

먼저, 파이썬에서 SQL을 사용하기 위해 필수적으로 쓸 패키지는 SQLite이다.

SQLite는 서버 프로세스가 필요 없고 비표준 SQL 변형으로 데이터베이스에 접근할 수 있는 경량 디스크 기반 데이터베이스를 제공하는 C 라이브러리이다.

sql ite3 모듈은 DB-API 2.0 사양과 호환 되는 SQL 인터페이스 를 제공한다.

 

사용법

1. 데이터베이스 객체 생성

먼저 sqlite3 모듈을 사용할 데이터베이스를 나타내는 Connection 객체를 생성한다.111

import sqlite3
con = sqlite3.connect('ev_models.db')

2. Cursor 객체를 생성

Cursor 객체를 생성하고 execute 메서드를 호출하여 SQL 명령을 수행한다.

cur = con.cursor()

# 테이블 생성
cur.execute('''CREATE TABLE ev_sbsd
               (id INTEGER PRIMARY KEY AUTOINCREMENT,
               	model_name text, top_speed text, range text, subsidy integer)''')

# 데이터 삽입
cur.execute("INSERT INTO ev_sbsd VALUES ('Model 3 Long Range AWD','233','527.9','315')")

# 변경을 저장 (commit)
con.commit()

# 작업이 끝나면 연결을 종료한다.
# 변경을 commit 했는지 확인하지 않으면 변경은 사라진다.
con.close()

3. 삽입한 데이터를 확인

import sqlite3
con = sqlite3.connect('example.db')
cur = con.cursor()

for row in cur.execute('SELECT * FROM ev_sbsd'):
	print(row)

출력 결과는 다음과 같다.

(1, 'Model 3 Long Range AWD', '233', '527.9', 315)

 

실행 결과는 입력한 데이터가 정상적으로 표시된다.