개발노트: 파이썬 데코레이터 decorater 편집

== 파이썬 데코레이터 == {{알림 |내용=데코레이터를 사용하여 데이터베이스(DB) 접속 * 반복적인 DB 연결/해제 로직을 깔끔하게 감싸는 데 유용합니다. }} === 기본 구조 === # 데코레이터 함수: DB에 연결하고, wrapped 함수 실행 후 연결 해제 # 커넥션 객체: sqlite3, psycopg2, cx_Oracle, pymysql 등 사용 가능 #:* 활용 예시: 쿼리 실행 시 자동으로 연결, 종료를 처리 ---- === 예제 (SQLite 기준) === <source lang=python> 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')] </source> ==== 장점 ==== * 중복되는 connect() / close() 코드 제거 * 트랜잭션 자동 처리 (commit, rollback) * 커넥션 전달을 깔끔하게 통제 가능 ---- === 다른 DB 예시 (psycopg2, cx_Oracle, pymysql) === * 필요에 따라 아래처럼 응용 가능: <source lang=python> 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 </source> * 추가적으로 필요하신 구조 ** 커넥션 풀 연동 (cx_Oracle.SessionPool, psycopg2.pool) ** 비동기 데코레이터 (async with) ** JSON 설정 기반 자동 커넥션 선택 == 관련글 == * [[다중DB + 커넥션풀캐시 + 데코레이터]] [[category:python]]