def __init__()

in evalbench/databases/postgres.py [0:0]

• 26 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, db_config):
        super().__init__(db_config)
        self.connector = Connector()

        def get_conn():
            conn = self.connector.connect(
                self.db_path,
                "pg8000",
                user=self.username,
                password=self.password,
                db=self.db_name,
            )
            return conn

        def get_engine_args():
            common_args = {
                "creator": get_conn,
                "connect_args": {"command_timeout": 60},
            }
            if "is_tmp_db" in db_config:
                common_args["poolclass"] = NullPool
            else:
                common_args["pool_size"] = 50
                common_args["pool_recycle"] = 300
            return common_args

        self.engine = sqlalchemy.create_engine(
            "postgresql+pg8000://", **get_engine_args()
        )