From 6fe5871d96af328d3b7f7136054039c97be5770b Mon Sep 17 00:00:00 2001 From: mhchia Date: Tue, 24 Dec 2019 14:44:28 +0800 Subject: [PATCH] Use `async-exit-stack` over contextlib For `AsyncExitStack` --- libp2p/tools/factories.py | 3 +-- libp2p/tools/pubsub/dummy_account_node.py | 3 ++- setup.py | 1 + 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/libp2p/tools/factories.py b/libp2p/tools/factories.py index 74fbb5f..144c8bc 100644 --- a/libp2p/tools/factories.py +++ b/libp2p/tools/factories.py @@ -1,7 +1,6 @@ -from contextlib import AsyncExitStack from typing import Any, AsyncIterator, Dict, List, Sequence, Tuple, cast -# NOTE: import ``asynccontextmanager`` from ``contextlib`` when support for python 3.6 is dropped. +from async_exit_stack import AsyncExitStack from async_generator import asynccontextmanager from async_service import background_trio_service import factory diff --git a/libp2p/tools/pubsub/dummy_account_node.py b/libp2p/tools/pubsub/dummy_account_node.py index 9079ac2..f29a9fd 100644 --- a/libp2p/tools/pubsub/dummy_account_node.py +++ b/libp2p/tools/pubsub/dummy_account_node.py @@ -1,6 +1,7 @@ -from contextlib import AsyncExitStack, asynccontextmanager from typing import AsyncIterator, Dict, Tuple +from async_exit_stack import AsyncExitStack +from async_generator import asynccontextmanager from async_service import Service, background_trio_service from libp2p.host.host_interface import IHost diff --git a/setup.py b/setup.py index aaebd30..435f72e 100644 --- a/setup.py +++ b/setup.py @@ -76,6 +76,7 @@ install_requires = [ "async_generator==1.10", "trio>=0.13.0", "async-service>=0.1.0a2,<0.2.0", + "async-exit-stack==1.0.1", ]