20 Sat

[HEAD FIRST PYTHON] 9강 컨텍스트 관리 프로토콜 : 파이썬의 with문과 연결하기

• 7장에서는 컨텍스트 관리 프로토콜을 적용한 with문을 이용하는 것이 좋다는 결론을 얻었고 8장에서는 클래스를 배웠다. 9장에서는 클래스와 컨텍스트 관리자를 통해 데이터베이스 코드를 공유하는 작업을 시작할 것임

메서드로 컨텍스트 관리하기

컨텍스트 관리 프로토콜을 준수하는 모든 클래스는 __enter__ 와 __exit__ 두 개의 요술 메서드를 반드시 정의해야 한다.

• enter는 설정을 담당한다.

  • 객체가 with문의 스위트를 시작하기 전에 enter 메서드를 먼저 호출한다. 프로토콜은 enter가 with문에 반환값을 제공할 수 있음을 명시한다.

• exit는 마무리를 담당한다.

  • 객체가 with문의 스위트를 마무리 하기전에 exit 메서드를 호출한다.

  • 예외 발생 등으로 인해 정상적으로 동작하지 않을 경우를 대비해 exit 메서드를 구현해야 한다.

• init은 초기화를 담당한다.

  • enter 메서드가 호출되기 전에 먼저 실행된다.

  • 컨텍스트 관리자는 init을 정의할 필요는 없지만 설정 코드와 초기화 코드를 분리할 수 있으므로 정의하면 유용하다.

    def __init__(self, config: dict):
        """Add the database configuration parameters to the object.

        This class expects a single dictionary argument which needs to assign
        the appropriate values to (at least) the following keys:

            host - the IP address of the host running MySQL/MariaDB.
            user - the MySQL/MariaDB username to use.
            password - the user's password.
            database - the name of the database to use.

        For more options, refer to the mysql-connector-python documentation.
        """
        self.configuration = config

    def __enter__(self) -> 'cursor':
        """Connect to database and create a DB cursor.

        Return the database cursor to the context manager.
        """
        self.conn = mysql.connector.connect(**self.configuration)
        self.cursor = self.conn.cursor()
        return self.cursor

    def __exit__(self, exc_type, exc_value, exc_traceback):
        """Destroy the cursor as well as the connection (after committing).
        """
        self.conn.commit()
        self.cursor.close()
        self.conn.close()

• 1 : init하는 과정

• 14 : dbconfig를 self.configuration으로 초기화한다.

• 16 : enter의 반환값으로 cursor라는 것을 알려주기 위해 어노테이션을 추가 한다.

• 21 : dbconfig가 아니라 self.configuration을 써준다.

• 23 : cursor를 반환

• 25 : 인자값에 대해서는 추후에 설명