Support for asynchronous calls

[pySilver]

Looks like project is missing async support, similar to what was introduced at AmbitionEng/django-pghistory#220

[pySilver]

"""
Async-compatible utilities for pgtrigger.

pgtrigger.ignore() uses thread-local storage, which doesn't work with async
code because Django's async ORM (asave, adelete, etc.) uses sync_to_async
internally, running DB operations in a different thread than where the context
manager was entered.

This module provides an async-compatible wrapper that ensures the ignore state
is set in the same thread where the DB operation executes.

Inspired by django-pghistory's approach:
https://github.com/AmbitionEng/django-pghistory/blob/main/pghistory/runtime.py
"""

from contextlib import ContextDecorator
from typing import Any

import pgtrigger
from asgiref.sync import sync_to_async


class pgtrigger_ignore(ContextDecorator):  # noqa: N801
    """
    Async-compatible wrapper for pgtrigger.ignore().

    Works as both sync and async context manager, and as a decorator.

    Usage as async context manager:
        async with pgtrigger_ignore("app.Model:trigger_name"):
            await instance.asave()

    Usage as sync context manager:
        with pgtrigger_ignore("app.Model:trigger_name"):
            instance.save()

    Usage as decorator (sync functions only):
        @pgtrigger_ignore("app.Model:trigger_name")
        def my_func():
            instance.save()
    """

    def __init__(self, *uris: str) -> None:
        self._uris = uris
        self._context: Any = None

    def __enter__(self) -> None:
        self._context = pgtrigger.ignore(*self._uris)
        return self._context.__enter__()

    def __exit__(self, *exc: object) -> bool | None:
        if self._context is not None:
            return self._context.__exit__(*exc)
        return None

    async def __aenter__(self) -> None:
        return await sync_to_async(self.__enter__)()

    async def __aexit__(self, *exc: object) -> bool | None:
        return await sync_to_async(self.__exit__)(*exc)

should work