REDROOM

Preview: mixins.py Size: 2.78 KB

/proc/thread-self/root/lib/python3.6/site-packages/acme/mixins.py

"""Useful mixins for Challenge and Resource objects"""
from typing import Any
from typing import Dict


class VersionedLEACMEMixin:
    """This mixin stores the version of Let's Encrypt's endpoint being used."""
    @property
    def le_acme_version(self) -> int:
        """Define the version of ACME protocol to use"""
        return getattr(self, '_le_acme_version', 1)

    @le_acme_version.setter
    def le_acme_version(self, version: int) -> None:
        # We need to use object.__setattr__ to not depend on the specific implementation of
        # __setattr__  in current class (eg. jose.TypedJSONObjectWithFields raises AttributeError
        # for any attempt to set an attribute to make objects immutable).
        object.__setattr__(self, '_le_acme_version', version)

    def __setattr__(self, key: str, value: Any) -> None:
        if key == 'le_acme_version':
            # Required for @property to operate properly. See comment above.
            object.__setattr__(self, key, value)
        else:
            super().__setattr__(key, value)  # pragma: no cover


class ResourceMixin(VersionedLEACMEMixin):
    """
    This mixin generates a RFC8555 compliant JWS payload
    by removing the `resource` field if needed (eg. ACME v2 protocol).
    """
    def to_partial_json(self) -> Dict[str, Any]:
        """See josepy.JSONDeserializable.to_partial_json()"""
        return _safe_jobj_compliance(super(),
                                     'to_partial_json', 'resource')

    def fields_to_partial_json(self) -> Dict[str, Any]:
        """See josepy.JSONObjectWithFields.fields_to_partial_json()"""
        return _safe_jobj_compliance(super(),
                                     'fields_to_partial_json', 'resource')


class TypeMixin(VersionedLEACMEMixin):
    """
    This mixin allows generation of a RFC8555 compliant JWS payload
    by removing the `type` field if needed (eg. ACME v2 protocol).
    """
    def to_partial_json(self) -> Dict[str, Any]:
        """See josepy.JSONDeserializable.to_partial_json()"""
        return _safe_jobj_compliance(super(),
                                     'to_partial_json', 'type')

    def fields_to_partial_json(self) -> Dict[str, Any]:
        """See josepy.JSONObjectWithFields.fields_to_partial_json()"""
        return _safe_jobj_compliance(super(),
                                     'fields_to_partial_json', 'type')


def _safe_jobj_compliance(instance: Any, jobj_method: str,
                          uncompliant_field: str) -> Dict[str, Any]:
    if hasattr(instance, jobj_method):
        jobj: Dict[str, Any] = getattr(instance, jobj_method)()
        if instance.le_acme_version == 2:
            jobj.pop(uncompliant_field, None)
        return jobj

    raise AttributeError('Method {0}() is not implemented.'.format(jobj_method))  # pragma: no cover

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2025-04-13 08:21:05	Edit Download recursive recursive
challenges.py	21.19 KB	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
client.py	52.86 KB	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
crypto_util.py	16.34 KB	lrw-r--r--	2022-11-09 23:17:36	Edit Download recursive recursive
errors.py	4.21 KB	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
fields.py	1.84 KB	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
jws.py	2.33 KB	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
magic_typing.py	595 B	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
messages.py	24.74 KB	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
mixins.py	2.78 KB	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
standalone.py	12.83 KB	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
util.py	303 B	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive
__init__.py	726 B	lrw-r--r--	2021-12-07 22:02:45	Edit Download recursive recursive

Directory Contents