Add return type to PoolAcquireContext.aenter and rename attribute #1270

Original file line number	Diff line number	Diff line change
Expand Up		@@ -1042,19 +1042,19 @@ async def __aexit__(self, *exc):

		class PoolAcquireContext:

	__slots__ = ('timeout', 'connection', 'done', 'pool')
	__slots__ = ('timeout', '_conn', 'done', 'pool')

		def __init__(self, pool: Pool, timeout: Optional[float]) -> None:
		self.pool = pool
		self.timeout = timeout
	self.connection = None
	self._conn = None
		self.done = False

	async def __aenter__(self):
	if self.connection is not None or self.done:
	async def __aenter__(self) -> connection.Connection:
	if self._conn is not None or self.done:
		raise exceptions.InterfaceError('a connection is already acquired')
	self.connection = await self.pool._acquire(self.timeout)
	return self.connection
	self._conn = await self.pool._acquire(self.timeout)
	return self._conn

		async def __aexit__(
		self,
Expand All		@@ -1063,8 +1063,8 @@ async def __aexit__(
		exc_tb: Optional[TracebackType] = None,
		) -> None:
		self.done = True
	con = self.connection
	self.connection = None
	con = self._conn
	self._conn = None
		await self.pool.release(con)

		def __await__(self):
Expand Down

Navigation Menu