Add uvloop.install() helper function.

MagicStack · Aug 2, 2018 · a3d8d40 · a3d8d40
1 parent f24c822
commit a3d8d40
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/uvloop/__init__.py b/uvloop/__init__.py
@@ -1,4 +1,4 @@
-import asyncio
+import asyncio as __asyncio
 
 from asyncio.events import BaseDefaultEventLoopPolicy as __BasePolicy
 
@@ -8,10 +8,10 @@
 
 
 __version__ = '0.12.0.dev0'
-__all__ = ('new_event_loop', 'EventLoopPolicy')
+__all__ = ('new_event_loop', 'install', 'EventLoopPolicy')
 
 
-class Loop(__BaseLoop, asyncio.AbstractEventLoop):
+class Loop(__BaseLoop, __asyncio.AbstractEventLoop):
     pass
 
 
@@ -20,6 +20,11 @@ def new_event_loop():
     return Loop()
 
 
+def install():
+    """A helper function to install uvloop policy."""
+    __asyncio.set_event_loop_policy(EventLoopPolicy())
+
+
 class EventLoopPolicy(__BasePolicy):
     """Event loop policy.