파이썬 데코레이터 decorater

파이썬 데코레이터

assignment 데코레이터를 사용하여 데이터베이스(DB) 접속

• 반복적인 DB 연결/해제 로직을 깔끔하게 감싸는 데 유용합니다.

기본 구조

1. 데코레이터 함수: DB에 연결하고, wrapped 함수 실행 후 연결 해제
2. 커넥션 객체: sqlite3, psycopg2, cx_Oracle, pymysql 등 사용 가능

   • 활용 예시: 쿼리 실행 시 자동으로 연결, 종료를 처리

---

예제 (SQLite 기준)

import sqlite3
from functools import wraps

# 데코레이터 정의
def with_db_connection(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        conn = sqlite3.connect("example.db")  # 연결
        try:
            result = func(conn, *args, **kwargs)  # 함수에 conn 전달
            conn.commit()
            return result
        except Exception as e:
            conn.rollback()
            raise e
        finally:
            conn.close()  # 연결 종료
    return wrapper

# 실제 사용
@with_db_connection
def create_table(conn):
    cursor = conn.cursor()
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS users (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT
        )
    """)

@with_db_connection
def insert_user(conn, name):
    cursor = conn.cursor()
    cursor.execute("INSERT INTO users (name) VALUES (?)", (name,))

@with_db_connection
def get_users(conn):
    cursor = conn.cursor()
    cursor.execute("SELECT * FROM users")
    return cursor.fetchall()

# 함수 실행
create_table()
insert_user("Alice")
insert_user("Bob")
print(get_users())  # [('1', 'Alice'), ('2', 'Bob')]

장점

• 중복되는 connect() / close() 코드 제거
• 트랜잭션 자동 처리 (commit, rollback)
• 커넥션 전달을 깔끔하게 통제 가능

---

다른 DB 예시 (psycopg2, cx_Oracle, pymysql)

• 필요에 따라 아래처럼 응용 가능:

def with_pg_connection(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        conn = psycopg2.connect(
            host="localhost",
            database="mydb",
            user="myuser",
            password="mypassword"
        )
        try:
            return func(conn, *args, **kwargs)
        finally:
            conn.close()
    return wrapper

• 추가적으로 필요하신 구조
  • 커넥션 풀 연동 (cx_Oracle.SessionPool, psycopg2.pool)
  • 비동기 데코레이터 (async with)
  • JSON 설정 기반 자동 커넥션 선택

관련글

• 다중디비 커넥션풀캐시 데코레이터