diff options
author | Victor Stinner <victor.stinner@gmail.com> | 2014-11-19 03:01:08 +0100 |
---|---|---|
committer | Victor Stinner <victor.stinner@gmail.com> | 2014-11-19 03:01:08 +0100 |
commit | 88a06d3c608a7be32df0435ddc96e43de37829be (patch) | |
tree | 7c882988a77774bbd8b06bc93837be9ec721838e | |
parent | 63f4872647b652ba7ae120ba89f73b5c991f59bd (diff) | |
download | aioeventlet-88a06d3c608a7be32df0435ddc96e43de37829be.tar.gz |
Fix the event loop policy
-rw-r--r-- | aiogreen/__init__.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/aiogreen/__init__.py b/aiogreen/__init__.py index df48a12..d346f94 100644 --- a/aiogreen/__init__.py +++ b/aiogreen/__init__.py @@ -32,13 +32,12 @@ class EventLoopPolicy(trollius.AbstractEventLoopPolicy): def get_event_loop(self): if not _is_main_thread(): raise NotImplementedError("currently aiogreen can only run in the main thread") - self._loop = EventLoop() + if self._loop is None: + self._loop = EventLoop() return self._loop def new_event_loop(self): - if self._loop is not None: - raise NotImplementedError("cannot run two event loops in the same thread") - return self.get_event_loop() + return EventLoop() def set_event_loop(self, loop): if self._loop is not None: |