Python code snippets / template / live templates (raw, pycharm and vscode)

decorator_pycharm.py

import functools


def $NAME$(func):
    @functools.wraps(func)
    async def _$NAME$(*args, **kwargs):
        # optionally do something before
        result = await func(*args, **kwargs)
        # optionally do something after
        return result

    return _$NAME$

decorator_raw.py

import functools


def decorator_name(func):
    @functools.wraps(func)
    async def _decorator_name(*args, **kwargs):
        # optionally do something before
        result = await func(*args, **kwargs)
        # optionally do something after
        return result

    return _decorator_name

decorator_with_params_pycharm.py

import functools
from typing import Any


def $NAME$(func=None, *, param: Any):
    def $NAME$_decorator(func):

        @functools.wraps(func)
        async def _$NAME$(*args, **kwargs):
            # optionally do something before
            result = await func(*args, **kwargs)
            # optionally do something after
            return result

        return _$NAME$

    if func is None:
        return $NAME$_decorator
    else:
        return $NAME$_decorator(func)
$END$

decorator_with_params_raw.py

import functools
from typing import Any


def decorator_name(func=None, *, param: Any):
    def decorator_name_decorator(func):

        @functools.wraps(func)
        async def _decorator_name(*args, **kwargs):
            # optionally do something before
            result = await func(*args, **kwargs)
            # optionally do something after
            return result

        return _decorator_name

    if func is None:
        return decorator_name_decorator
    else:
        return decorator_name_decorator(func)