Patch invoke for Python 3.11 support
parent
9c65919070
commit
673baf0d7f
32
tasks.py
32
tasks.py
|
@ -2,17 +2,49 @@ import asyncio
|
||||||
import io
|
import io
|
||||||
import shutil
|
import shutil
|
||||||
import tarfile
|
import tarfile
|
||||||
|
from collections import namedtuple
|
||||||
from contextlib import contextmanager
|
from contextlib import contextmanager
|
||||||
|
from inspect import getfullargspec
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Generator
|
from typing import Generator
|
||||||
from typing import Optional
|
from typing import Optional
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
import httpx
|
import httpx
|
||||||
|
import invoke # type: ignore
|
||||||
from invoke import Context # type: ignore
|
from invoke import Context # type: ignore
|
||||||
from invoke import run # type: ignore
|
from invoke import run # type: ignore
|
||||||
from invoke import task # type: ignore
|
from invoke import task # type: ignore
|
||||||
|
|
||||||
|
|
||||||
|
def fix_annotations():
|
||||||
|
"""
|
||||||
|
Pyinvoke doesn't accept annotations by default, this fix that
|
||||||
|
Based on: @zelo's fix in https://github.com/pyinvoke/invoke/pull/606
|
||||||
|
Context in: https://github.com/pyinvoke/invoke/issues/357
|
||||||
|
Python 3.11 https://github.com/pyinvoke/invoke/issues/833
|
||||||
|
"""
|
||||||
|
|
||||||
|
ArgSpec = namedtuple("ArgSpec", ["args", "defaults"])
|
||||||
|
|
||||||
|
def patched_inspect_getargspec(func):
|
||||||
|
spec = getfullargspec(func)
|
||||||
|
return ArgSpec(spec.args, spec.defaults)
|
||||||
|
|
||||||
|
org_task_argspec = invoke.tasks.Task.argspec
|
||||||
|
|
||||||
|
def patched_task_argspec(*args, **kwargs):
|
||||||
|
with patch(
|
||||||
|
target="inspect.getargspec", new=patched_inspect_getargspec, create=True
|
||||||
|
):
|
||||||
|
return org_task_argspec(*args, **kwargs)
|
||||||
|
|
||||||
|
invoke.tasks.Task.argspec = patched_task_argspec
|
||||||
|
|
||||||
|
|
||||||
|
fix_annotations()
|
||||||
|
|
||||||
|
|
||||||
@task
|
@task
|
||||||
def generate_db_migration(ctx, message):
|
def generate_db_migration(ctx, message):
|
||||||
# type: (Context, str) -> None
|
# type: (Context, str) -> None
|
||||||
|
|
Loading…
Reference in New Issue