(∩｀-´)⊃━☆ﾟ.*･｡ﾟ

Author: Gr1N

.editorconfig

@@ -0,0 +1,19 @@
+; This file is for unifying the coding style for different editors and IDEs.
+; More information at http://EditorConfig.org
+
+; Editors and IDEs plugins:
+;  — SublimeText — https://github.com/sindresorhus/editorconfig-sublime
+;  — Vim — https://github.com/editorconfig/editorconfig-vim
+;  — JetBrains IDEs — https://github.com/editorconfig/editorconfig-jetbrains
+
+root = true
+
+[*]
+charset = utf-8
+end_of_line = lf
+insert_final_newline = true
+trim_trailing_whitespace = true
+
+[*.{py,md,ini}]
+indent_style = space
+indent_size = 4

.gitignore

@@ -0,0 +1,87 @@
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*,cover
+.hypothesis/
+
+# Translations
+*.mo
+*.pot
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+
+### Django ###
+*.log
+*.pot
+*.pyc
+__pycache__/
+local_settings.py
+db.sqlite3
+media
+
+
+### SublimeText ###
+# cache files for sublime text
+*.tmlanguage.cache
+*.tmPreferences.cache
+*.stTheme.cache
+
+# workspace files are user-specific
+*.sublime-workspace
+
+# project files should be checked into the repository, unless a significant
+# proportion of contributors will probably not be using SublimeText
+*.sublime-project
+
+# sftp configuration file
+sftp-config.json
+
+
+### Other ###

MANIFEST.in

@@ -0,0 +1,4 @@
+include MANIFEST.in
+include *.md
+global-exclude __pycache__
+global-exclude *.py[co]

README.md

@@ -0,0 +1,50 @@
+# asyncio-ipython-magic [![Requirements Status](https://requires.io/github/Gr1N/asyncio-ipython-magic/requirements.svg?branch=master)](https://requires.io/github/Gr1N/asyncio-ipython-magic/requirements/?branch=master) [![PyPI](https://img.shields.io/pypi/v/asyncio-ipython-magic.svg)](https://pypi.python.org/pypi/asyncio-ipython-magic) [![Supported Python versions](https://img.shields.io/pypi/pyversions/asyncio-ipython-magic.svg)](https://pypi.python.org/pypi/asyncio-ipython-magic)
+
+An extension for [IPython](https://ipython.org) that help to run AsyncIO code in your interactive session.
+
+Based on [Gist](https://gist.github.com/takluyver/b9663b08ac9a4472afa6).
+
+## Installation
+
+Install `asyncio-ipython-magic` using [pip](http://www.pip-installer.org/):
+
+    $ pip install asyncio-ipython-magic
+
+...or directly from the repository using the `%install_ext` magic command:
+
+    $ In[1]: %install_ext https://raw.githubusercontent.com/Gr1N/asyncio-ipython-magic/master/asynciomagic.py
+
+Enjoy!
+
+## Usage
+
+    In [1]: %load_ext asynciomagic
+
+    In [2]: import asyncio
+
+    In [3]: import time
+
+    In [4]: async def foo():
+       ...:     i = 0
+       ...:     while i < 3:
+       ...:         print('time =', time.time())
+       ...:         i += 1
+       ...:         await asyncio.sleep(2)
+       ...:
+
+    In [5]: %%asyncio
+       ...: await foo()
+       ...:
+    time = 1478985421.307329
+    time = 1478985423.309606
+    time = 1478985425.31514
+
+    In [6]:
+
+## Testing
+
+It just works, I hope.
+
+## License
+
+*asyncio-ipython-magic* is licensed under the MIT license. See the license file for details.

asynciomagic.py

@@ -0,0 +1,68 @@
+# -*- coding: utf-8 -*-
+
+import ast
+from ast import Call, Attribute, Name, Load
+import asyncio as _asyncio
+
+from IPython.core.magic import Magics, magics_class, cell_magic
+from IPython.utils.text import indent
+
+
+class RewriteAwait(ast.NodeTransformer):
+    def visit_Await(self, node):
+        self.generic_visit(node)
+
+        new = Call(
+            func=Attribute(
+                value=Call(
+                    func=Attribute(
+                        value=Name(
+                            id='_asyncio',
+                            ctx=Load()
+                        ),
+                        attr='get_event_loop',
+                        ctx=Load()),
+                    args=[],
+                    keywords=[]
+                ),
+                attr='run_until_complete',
+                ctx=Load()
+            ),
+            args=[node.value],
+            keywords=[]
+        )
+
+        return ast.copy_location(new, node)
+
+    def visit_AsyncFunctionDef(self, node):
+        # Don't transform awaits inside an 'async def' function
+        return node
+
+    def visit_Return(self, node):
+        raise SyntaxError('Return outside function definition')
+
+
+@magics_class
+class AsyncIOMagics(Magics):
+    @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
+
+        nodes = [RewriteAwait().visit(node) for node in coro_wrapper]
+        module = ast.Module(nodes)
+        ast.fix_missing_locations(module)
+
+        coro = compile(module, filename='<asynciomagic>', mode='exec')
+
+        self.shell.ex(coro)
+
+
+def load_ipython_extension(ipython):
+    ipython.user_global_ns['_asyncio'] = _asyncio
+    ipython.register_magics(AsyncIOMagics)
+
+
+def unload_ipython_extension(ipython):
+    ipython.user_global_ns.pop('_asyncio', None)

setup.cfg

@@ -0,0 +1,2 @@
+[bdist_wheel]
+universal = 1

setup.py

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+
+from setuptools import setup
+
+
+setup(
+    name='asyncio-ipython-magic',
+    version='0.0.1',
+    description='An extension for IPython that help to run AsyncIO code in '
+                'your interactive session.',
+    author='Nikita Grishko',
+    author_email='[email protected]',
+    url='https://github.com/Gr1N/asyncio-ipython-magic',
+    py_modules=(
+        'asynciomagic',
+    ),
+    install_requires=(
+        'ipython',
+    ),
+    include_package_data=True,
+    classifiers=(
+        'Development Status :: 3 - Alpha',
+        'Framework :: IPython',
+        'Intended Audience :: Developers',
+        'Operating System :: OS Independent',
+        'Programming Language :: Python',
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.5',
+        'Topic :: Software Development :: Libraries :: Python Modules',
+    ),
+)