🚀 Move python code to separate module

Author: uriyyo

.gitignore

@@ -7,6 +7,7 @@
 .idea/**/usage.statistics.xml
 .idea/**/dictionaries
 .idea/**/shelf
+.idea/
 
 # Sensitive or high-churn files
 .idea/**/dataSources/
@@ -90,4 +91,143 @@ hs_err_pid*
 output/
 
 .gradle/
-build/
+build/
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# 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/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/

async-pydevd/__init__.py

@@ -0,0 +1,20 @@
+from pathlib import Path
+
+import nest_asyncio
+
+ROOT: Path = Path(__file__).parent
+
+FILES: tuple[Path, ...] = (
+    Path(nest_asyncio.__file__),
+    ROOT / "asyncio_patch.py",
+    ROOT / "async_eval.py",
+    ROOT / "pydevd_patch.py",
+    ROOT / "pydevd_main.py",
+)
+
+
+def generate() -> str:
+    return "\n".join(p.read_text("utf-8") for p in FILES).replace('"""', "'''").strip()
+
+
+__all__ = ["generate"]

async-pydevd/async_pydevd/__init__.py

@@ -0,0 +1,65 @@
+import ast
+import sys
+import textwrap
+
+
+# Async equivalent of builtin eval function
+def async_eval(expr: str, _globals: dict = None, _locals: dict = None):
+    if _locals is None:
+        _locals = {}
+
+    if _globals is None:
+        _globals = {}
+
+    expr = textwrap.indent(expr, "    ")
+    expr = f"async def _():\n{expr}"
+
+    parsed_stmts = ast.parse(expr).body[0]
+    for node in parsed_stmts.body:
+        ast.increment_lineno(node)
+
+    last_stmt = parsed_stmts.body[-1]
+
+    if isinstance(last_stmt, ast.Expr):
+        return_expr = ast.copy_location(ast.Return(last_stmt), last_stmt)
+        return_expr.value = return_expr.value.value
+        parsed_stmts.body[-1] = return_expr
+
+    parsed_fn = ast.parse(
+        f"""\
+async def __async_exec_func__(__locals__=__locals__):
+    try:
+        pass
+    finally:
+        __locals__.update(locals())
+        del __locals__['__locals__']
+
+import asyncio
+
+__async_exec_func_result__ = asyncio.get_event_loop().run_until_complete(__async_exec_func__())
+    """
+    )
+
+    parsed_fn.body[0].body[0].body = parsed_stmts.body
+
+    try:
+        code = compile(parsed_fn, filename="<ast>", mode="exec")
+    except (SyntaxError, TypeError):
+        parsed_stmts.body[-1] = last_stmt
+        parsed_fn.body[0].body[0].body = parsed_stmts.body
+        code = compile(parsed_fn, filename="<ast>", mode="exec")
+
+    _updated_locals = {
+        **_locals,
+        "__locals__": _locals,
+    }
+    _updated_globals = {
+        **_globals,
+        **_updated_locals,
+    }
+
+    exec(code, _updated_globals, _updated_locals)
+    return _updated_locals["__async_exec_func_result__"]
+
+
+sys.__async_eval__ = async_eval

async-pydevd/async_pydevd/async_eval.py

@@ -0,0 +1,38 @@
+import asyncio
+import functools
+import sys
+
+
+def _patch_asyncio_set_get_new():
+    if sys.platform.lower().startswith("win"):
+        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
+    apply()
+
+    def _patch_loop_if_not_patched(loop):
+        if not hasattr(loop, "_nest_patched"):
+            _patch_loop(loop)
+
+    def _patch_asyncio_api(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            loop = func(*args, **kwargs)
+            _patch_loop_if_not_patched(loop)
+            return loop
+
+        return wrapper
+
+    asyncio.get_event_loop = _patch_asyncio_api(asyncio.get_event_loop)
+    asyncio.new_event_loop = _patch_asyncio_api(asyncio.new_event_loop)
+
+    _set_event_loop = asyncio.set_event_loop
+
+    @functools.wraps(asyncio.set_event_loop)
+    def set_loop_wrapper(loop):
+        _patch_loop_if_not_patched(loop)
+        _set_event_loop(loop)
+
+    asyncio.set_event_loop = set_loop_wrapper
+
+
+_patch_asyncio_set_get_new()

async-pydevd/async_pydevd/asyncio_patch.py

@@ -0,0 +1,4 @@
+import sys
+from runpy import run_path
+
+run_path(sys.argv.pop(1), {}, "__main__")

async-pydevd/async_pydevd/pydevd_main.py

@@ -0,0 +1,73 @@
+def is_async_code(code: str) -> bool:
+    return "__async_eval__" not in code and ("await" in code or "async" in code)
+
+
+def make_code_async(code: str) -> str:
+    if not code:
+        return code
+
+    if is_async_code(code):
+        code = code.replace("@" + "LINE" + "@", "\n")
+        return f"__import__('sys').__async_eval__({code!r}, globals(), locals())"
+
+    return code
+
+
+# 1. Add ability to evaluate async expression
+from _pydevd_bundle import pydevd_vars
+
+original_evaluate = pydevd_vars.evaluate_expression
+
+
+def evaluate_expression(thread_id: int, frame_id: int, expression: str, doExec: bool):
+    if is_async_code(expression):
+        doExec = False
+
+    return original_evaluate(thread_id, frame_id, make_code_async(expression), doExec)
+
+
+pydevd_vars.evaluate_expression = evaluate_expression
+
+# 2. Add ability to use async breakpoint conditions
+from _pydevd_bundle.pydevd_breakpoints import LineBreakpoint
+
+
+def normalize_line_breakpoint(line_breakpoint: LineBreakpoint) -> None:
+    line_breakpoint.expression = make_code_async(line_breakpoint.expression)
+    line_breakpoint.condition = make_code_async(line_breakpoint.condition)
+
+
+original_init = LineBreakpoint.__init__
+
+
+def line_breakpoint_init(self: LineBreakpoint, *args, **kwargs):
+    original_init(self, *args, **kwargs)
+    normalize_line_breakpoint(self)
+
+
+LineBreakpoint.__init__ = line_breakpoint_init
+
+# 3. Add ability to use async code in console
+from _pydevd_bundle import pydevd_console_integration
+
+original_console_exec = pydevd_console_integration.console_exec
+
+
+def console_exec(thread_id: int, frame_id: int, expression: str, dbg):
+    return original_console_exec(thread_id, frame_id, make_code_async(expression), dbg)
+
+
+pydevd_console_integration.console_exec = console_exec
+
+# 4. Add ability to use async code
+from _pydev_bundle.pydev_console_types import Command
+
+
+def command_run(self):
+    text = make_code_async(self.code_fragment.text)
+    symbol = self.symbol_for_fragment(self.code_fragment)
+
+    self.more = self.interpreter.runsource(text, "<input>", symbol)
+
+
+Command.run = command_run

async-pydevd/async_pydevd/pydevd_patch.py

@@ -0,0 +1,16 @@
+[tool.poetry]
+name = "async-pydevd"
+version = "0.1.0"
+description = "pydevd with async/await support"
+authors = ["Yurii Karabas <[email protected]>"]
+license = "MIT"
+
+[tool.poetry.dependencies]
+python = "^3.9"
+nest-asyncio = "^1.4.3"
+
+[tool.poetry.dev-dependencies]
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"