added %await_ line magic; renamed %%asycnio cell magic to %%async_

Author: Gr1N

README.md

@@ -32,14 +32,19 @@ Enjoy!
        ...:         await asyncio.sleep(2)
        ...:
 
-    In [5]: %%asyncio
+    In [5]: %%async_
        ...: await foo()
        ...:
     time = 1478985421.307329
     time = 1478985423.309606
     time = 1478985425.31514
 
-    In [6]:
+    In [6]: %await_ foo()
+    time = 1487097377.700184
+    time = 1487097379.705614
+    time = 1487097381.707186
+
+    In [7]:
 
 ## Testing
 

asynciomagic.py

@@ -1,10 +1,20 @@
 # -*- coding: utf-8 -*-
 
 import ast
-from ast import Call, Attribute, Name, Load
+from ast import (
+    Call,
+    Attribute,
+    Name,
+    Load,
+)
 import asyncio as _asyncio
 
-from IPython.core.magic import Magics, magics_class, cell_magic
+from IPython.core.magic import (
+    Magics,
+    magics_class,
+    cell_magic,
+    line_magic,
+)
 from IPython.utils.text import indent
 
 
@@ -44,13 +54,23 @@ def visit_Return(self, node):
 
 @magics_class
 class AsyncIOMagics(Magics):
+    @line_magic
+    def await_(self, line):
+        expr = 'async def __f(): await {line}'.format(line=indent(line))
+
+        self._exec(expr)
+
     @cell_magic
-    def asyncio(self, line, cell):
-        coro_wrapper = 'async def __f():\n{cell}'.format(cell=indent(cell))
-        coro_wrapper = ast.parse(coro_wrapper)
-        coro_wrapper = coro_wrapper.body[0].body
+    def async_(self, line, cell):
+        expr = 'async def __f():\n{cell}'.format(cell=indent(cell))
+
+        self._exec(expr)
+
+    def _exec(self, expr):
+        expr = ast.parse(expr)
+        expr = expr.body[0].body
 
-        nodes = [RewriteAwait().visit(node) for node in coro_wrapper]
+        nodes = [RewriteAwait().visit(node) for node in expr]
         module = ast.Module(nodes)
         ast.fix_missing_locations(module)
 

setup.py

@@ -26,6 +26,7 @@
         'Programming Language :: Python',
         'Programming Language :: Python :: 3',
         'Programming Language :: Python :: 3.5',
+        'Programming Language :: Python :: 3.6',
         'Topic :: Software Development :: Libraries :: Python Modules',
     ]
 )