diff options
| author | Federico Caselli <cfederico87@gmail.com> | 2020-08-16 10:48:57 +0200 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2020-12-30 15:49:09 -0500 |
| commit | 2581655c545a0cf705e0347e81cd092896d3207c (patch) | |
| tree | 70a9528216c3de54797b4197978c494ae819d39a /lib/sqlalchemy/util/queue.py | |
| parent | 102b91d8950926f1215dd7c59c5b7f200b5c0f8b (diff) | |
| download | sqlalchemy-2581655c545a0cf705e0347e81cd092896d3207c.tar.gz | |
Support testing of async drivers without fallback mode
Change-Id: I4940d184a4dc790782fcddfb9873af3cca844398
Diffstat (limited to 'lib/sqlalchemy/util/queue.py')
| -rw-r--r-- | lib/sqlalchemy/util/queue.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/lib/sqlalchemy/util/queue.py b/lib/sqlalchemy/util/queue.py index 3687dc8dc..a92d6b862 100644 --- a/lib/sqlalchemy/util/queue.py +++ b/lib/sqlalchemy/util/queue.py @@ -25,6 +25,7 @@ from . import compat from .compat import threading from .concurrency import asyncio from .concurrency import await_fallback +from .concurrency import await_only __all__ = ["Empty", "Full", "Queue"] @@ -202,7 +203,7 @@ class Queue: class AsyncAdaptedQueue: - await_ = staticmethod(await_fallback) + await_ = staticmethod(await_only) def __init__(self, maxsize=0, use_lifo=False): if use_lifo: @@ -265,3 +266,7 @@ class AsyncAdaptedQueue: Empty(), replace_context=err, ) + + +class FallbackAsyncAdaptedQueue(AsyncAdaptedQueue): + await_ = staticmethod(await_fallback) |