enter-timeout.py

class EnterTimeout:
    def __init__(self, async_cm, timeout):
        self._async_cm = async_cm
        self._timeout = timeout
        
    async def __aenter__(self):
        with fail_after(self.timeout):
            return await self._async_cm.__aenter__()
        
    async def __aexit__(self, *args):
        return await self._async_cm.__aexit__(*args)
    
    
async with EnterTimeout(some_cm, 10):
    ...