Python, SQLAlchemy를 사용하여 PostgreSQL에 연결하는 방법
2024-07-27
이 글에서는 Python, SQLAlchemy를 사용하여 PostgreSQL 데이터베이스에 연결하는 방법에 대해 설명합니다.
필수 조건
- Python 3.x 이상 설치
- PostgreSQL 설치 및 실행
- SQLAlchemy 설치 (pip install sqlalchemy)
단계
엔진 생성하기:
from sqlalchemy import create_engine # 데이터베이스 연결 정보 설정 engine = create_engine("postgresql://user:password@host:port/database")위 코드에서:
user: PostgreSQL 사용자 이름password: PostgreSQL 사용자 비밀번호host: PostgreSQL 서버 호스트 이름 (보통 'localhost')port: PostgreSQL 서버 포트 번호 (보통 5432)database: 사용하려는 PostgreSQL 데이터베이스 이름
연결 확인하기:
# 연결 확인 with engine.connect() as connection: connection.execute("SELECT 1")with engine.connect()문장은connection이라는 변수에 데이터베이스 연결을 할당합니다.connection.execute("SELECT 1")문장은 간단한 SQL 쿼리를 실행하여 연결이 제대로 작동하는지 확인합니다.
데이터베이스 조작하기:
연결을 생성하고 확인한 후, SQLAlchemy를 사용하여 데이터베이스를 자유롭게 조작할 수 있습니다.
예를 들어, 테이블 생성, 데이터 삽입, 조회, 삭제 등을 수행할 수 있습니다.
주의 사항
- 민감한 데이터(비밀번호 등)는 코드에 직접 작성하지 않고, 환경 변수 등을 사용하여 안전하게 저장하는 것이 좋습니다.
- 실제 코드 작성 시에는 에러 처리 및 예외 상황 고려 등을 추가적으로 수행해야 합니다.
Python, SQLAlchemy를 사용하여 PostgreSQL에 연결하는 예제 코드
from sqlalchemy import create_engine, Table, Column, Integer, String
# 엔진 생성
engine = create_engine("postgresql://user:password@host:port/database")
# 메타데이터 생성
metadata = MetaData()
# 테이블 정의
users_table = Table("users", metadata,
Column("id", Integer, primary_key=True),
Column("name", String(255)),
Column("email", String(255), unique=True))
# 테이블 생성
metadata.create_all(engine)
데이터 삽입
from sqlalchemy import insert
# 사용자 데이터
user_data = [
{"name": "홍길동", "email": "[email protected]"},
{"name": "김철수", "email": "[email protected]"},
]
# 데이터 삽입
with engine.connect() as connection:
for user in user_data:
stmt = insert(users_table).values(name=user["name"], email=user["email"])
connection.execute(stmt)
데이터 조회
from sqlalchemy import select
# 모든 사용자 조회
with engine.connect() as connection:
stmt = select(users_table)
result = connection.execute(stmt).fetchall()
for row in result:
print(f"id: {row[0]}, name: {row[1]}, email: {row[2]}")
데이터 수정
from sqlalchemy import update
# 특정 사용자의 이름 및 이메일 수정
with engine.connect() as connection:
stmt = update(users_table).where(users_table.c.id == 1).values(name="이순신", email="[email protected]")
connection.execute(stmt)
데이터 삭제
from sqlalchemy import delete
# 특정 사용자 삭제
with engine.connect() as connection:
stmt = delete(users_table).where(users_table.c.id == 2)
connection.execute(stmt)
참고:
- 이 코드는 기본적인 예시이며, 실제 상황에 맞게 수정해야 합니다.
- 에러 처리, 트랜잭션 처리 등을 추가적으로 고려해야 합니다.
- SQLAlchemy 공식 문서에서 더 많은 정보를 확인할 수 있습니다.
Python, SQLAlchemy를 사용하지 않고 PostgreSQL에 연결하는 대체 방법
psycopg2는 Python에서 PostgreSQL과 상호 작용하는 가장 인기 있는 드라이버입니다. SQLAlchemy보다 간단하지만, 객체 관계 매핑(ORM) 기능과 같은 고급 기능은 제공하지 않습니다.
예제 코드:
import psycopg2
# 데이터베이스 연결 정보 설정
conn = psycopg2.connect(
host="localhost",
port="5432",
database="mydatabase",
user="postgres",
password="mypassword"
)
# 커서 생성
cursor = conn.cursor()
# 데이터 삽입
cursor.execute("INSERT INTO users (name, email) VALUES (%s, %s)", ("이순신", "[email protected]"))
# 커밋
conn.commit()
# 커서 및 연결 종료
cursor.close()
conn.close()
cx_Oracle 사용하기
cx_Oracle은 Python에서 Oracle 데이터베이스와 상호 작용하는 데 사용되는 드라이버입니다. PostgreSQL 대신 Oracle 데이터베이스를 사용하는 경우 유용합니다.
import cx_Oracle
# 데이터베이스 연결 정보 설정
conn = cx_Oracle.connect(
user="scott",
password="tiger",
dsn="localhost/orcl"
)
# 커서 생성
cursor = conn.cursor()
# 데이터 삽입
cursor.execute("INSERT INTO employees (name, email) VALUES (:name, :email)", {"name": "홍길동", "email": "[email protected]"})
# 커밋
conn.commit()
# 커서 및 연결 종료
cursor.close()
conn.close()
aiopg 사용하기
aiopg는 비동기 방식으로 PostgreSQL과 상호 작용하는 데 사용되는 드라이버입니다. asyncio 기반 애플리케이션에서 유용합니다.
import asyncio
import aiopg
async def connect_db():
async with aiopg.connect(host="localhost", port="5432", database="mydatabase", user="postgres", password="mypassword") as conn:
cursor = await conn.cursor()
await cursor.execute("INSERT INTO users (name, email) VALUES ($1, $2)", ("김철수", "[email protected]"))
await conn.commit()
await cursor.close()
asyncio.run(connect_db())
- 이 외에도 다양한 PostgreSQL 연결 방법이 존재합니다.
- 특정 상황에 가장 적합한 방법을 선택하는 것이 중요합니다.
- 각 드라이버의 공식 문서를 참고하여 자세한 사용 방법을 확인해야 합니다.
python database postgresql
