데이터베이스와 관련된 작업은 프로그래밍에서 매우 중요합니다. 많은 프로그래밍 언어들이 데이터베이스와의 상호작용을 위한 다양한 방법을 제공합니다. 그 중에서도 SQLite는 경량 데이터베이스 관리 시스템으로, 파이썬과 함께 사용되기에 매우 적합합니다. 본 기사에서는 SQLite와 파이썬을 사용하여 데이터베이스와 연동하는 방법에 대해 살펴보겠습니다.
SQLite는 서버가 필요 없는 관계형 데이터베이스 관리 시스템입니다. 이것은 파일 기반 데이터베이스로, 전체 데이터베이스를 하나의 파일로 저장합니다. SQLite는 다음과 같은 특징을 가지고 있습니다:
파이썬에서 SQLite를 사용하려면 sqlite3 모듈을 사용해야 합니다. 이 모듈은 파이썬 표준 라이브러리에 포함되어 있어, 추가적인 설치 없이 사용할 수 있습니다.
먼저, 파이썬이 설치되어 있는지 확인합니다. 다음 단계로, SQLite를 테스트할 간단한 스크립트를 작성해 보겠습니다.
python --version
설치된 버전이 출력된다면, 파이썬이 정상적으로 설치된 것입니다. 이제 파이썬 인터프리터를 방문하여 SQLite 데이터베이스 파일을 생성하는 스크립트를 작성합니다.
SQLite를 사용하여 데이터베이스를 생성하기 위해서는 먼저 sqlite3 모듈을 임포트해야 합니다. 다음은 데이터베이스를 생성하는 파이썬 코드입니다.
import sqlite3
conn = sqlite3.connect('example.db')
c = conn.cursor()
c.execute('''CREATE TABLE IF NOT EXISTS users
(id INTEGER PRIMARY KEY, name TEXT, age INTEGER)''')
conn.commit()
conn.close()
위 코드에서는 example.db라는 이름의 데이터베이스 파일을 생성하고, users라는 테이블을 정의합니다. 각 사용자에 대한 id, name, age 칼럼이 포함되어 있습니다.
테이블을 생성한 후, 데이터베이스에 데이터를 추가하는 방법을 살펴보겠습니다. 다음은 데이터를 추가하는 코드입니다.
def add_user(name, age):
conn = sqlite3.connect('example.db')
c = conn.cursor()
c.execute("INSERT INTO users (name, age) VALUES (?, ?)", (name, age))
conn.commit()
conn.close()
add_user('John Doe', 30)
add_user('Jane Smith', 25)
위 함수 add_user를 이용하여 사용자를 추가할 수 있습니다. 함수는 이름과 나이를 인자로 받아 데이터베이스에 추가합니다.
데이터를 추가한 후, 데이터베이스에서 데이터를 조회하는 방법을 알아보겠습니다. 다음은 사용자 리스트를 조회하는 코드입니다.
def get_users():
conn = sqlite3.connect('example.db')
c = conn.cursor()
c.execute("SELECT * FROM users")
users = c.fetchall()
conn.close()
return users
for user in get_users():
print(user)
위 get_users 함수는 users 테이블에서 모든 데이터를 조회하여 리스트로 반환합니다. 이 데이터를 순회하여 각 사용자의 정보를 출력할 수 있습니다.
데이터베이스에 있는 데이터를 수정하는 방법도 필요합니다. 다음은 특정 사용자의 나이를 수정하는 코드 예시입니다.
def updateuserage(userid, newage):
conn = sqlite3.connect('example.db')
c = conn.cursor()
c.execute("UPDATE users SET age = ? WHERE id = ?", (newage, userid))
conn.commit()
conn.close()
updateuserage(1, 31)
updateuserage 함수는 특정 id에 해당하는 사용자의 나이를 업데이트합니다. 이와 같은 방식으로 원하는 데이터를 수정할 수 있습니다.
데이터베이스에서 데이터를 삭제하는 방법 또한 잘 알고 있어야 합니다. 다음은 사용자를 삭제하는 코드입니다.
def deleteuser(userid):
conn = sqlite3.connect('example.db')
c = conn.cursor()
c.execute("DELETE FROM users WHERE id = ?", (user_id,))
conn.commit()
conn.close()
delete_user(2)
delete_user 함수는 특정 id에 해당하는 사용자를 삭제합니다. 데이터 삭제는 주의해서 수행해야 합니다.
SQLite와 파이썬을 활용하여 더 복잡한 애플리케이션을 구축할 수 있습니다. 예를 들어, 간단한 사용자 관리 시스템을 만들어 볼 수 있습니다. 아래는 기본적인 사용자 관리 시스템의 구조입니다.
def main():
while True:
print("1. 사용자 추가")
print("2. 사용자 조회")
print("3. 사용자 수정")
print("4. 사용자 삭제")
print("5. 종료")
choice = input("선택하세요: ")
if choice == '1':
name = input("이름을 입력하세요: ")
age = int(input("나이를 입력하세요: "))
add_user(name, age)
elif choice == '2':
users = get_users()
for user in users:
print(user)
elif choice == '3':
user_id = int(input("수정할 사용자 ID를 입력하세요: "))
new_age = int(input("새 나이를 입력하세요: "))
updateuserage(userid, newage)
elif choice == '4':
user_id = int(input("삭제할 사용자 ID를 입력하세요: "))
deleteuser(userid)
elif choice == '5':
break
else:
print("잘못된 선택입니다.")
if name == 'main':
main()
위 코드는 사용자 관리 시스템의 기본 구조를 보여줍니다. 사용자가 원할 때마다 원하는 작업을 선택하여 수행할 수 있습니다.
SQLite와 파이썬을 통해 데이터베이스 연동의 기초를 익혔습니다. SQLite의 특징과 함께, 파이썬을 이용하여 데이터베이스를 생성, 조회, 수정, 삭제하는 방법을 배웠습니다. 이러한 기초적인 지식을 바탕으로, 더 복잡한 데이터베이스 시스템을 구축할 수 있는 발판을 마련했습니다.
SQLite는 여러 프로젝트에서 활용될 수 있는 유용한 도구입니다. 이 기사를 통해 얻은 지식을 바탕으로, 본격적인 데이터베이스 프로그래밍에 도전해 보시기 바랍니다.