pyUmbral/umbral/config.py

from typing import Optional, Type
from warnings import warn

from umbral.curve import Curve, SECP256K1
from umbral.params import UmbralParameters


class _CONFIG:
    __curve = None
    __params = None
    __CURVE_TO_USE_IF_NO_DEFAULT_IS_SET_BY_USER = SECP256K1
    __WARNING_IF_NO_DEFAULT_SET = "No default curve has been set.  Using SECP256K1.  A slight performance penalty has been incurred for only this call.  Set a default curve with umbral.config.set_default_curve()."

    class UmbralConfigurationError(RuntimeError):
        """Raised when somebody does something dumb re: configuration."""

    @classmethod
    def __set_curve_by_default(cls):
        warn(cls.__WARNING_IF_NO_DEFAULT_SET, RuntimeWarning)
        cls.set_curve(cls.__CURVE_TO_USE_IF_NO_DEFAULT_IS_SET_BY_USER)

    @classmethod
    def params(cls) -> UmbralParameters:
        if not cls.__params:
            cls.__set_curve_by_default()
        return cls.__params

    @classmethod
    def curve(cls) -> Type[Curve]:
        if not cls.__curve:
            cls.__set_curve_by_default()
        return cls.__curve

    @classmethod
    def set_curve(cls, curve: Optional[Curve] = None) -> None:
        if cls.__curve:
            raise cls.UmbralConfigurationError(
                "You can only set the default curve once.  Do it once and then leave it alone.")
        else:
            from umbral.params import UmbralParameters
            if curve is None:
                curve = _CONFIG.__CURVE_TO_USE_IF_NO_DEFAULT_IS_SET_BY_USER
            cls.__curve = curve
            cls.__params = UmbralParameters(curve)


def set_default_curve(curve: Optional[Curve] = None) -> None:
    return _CONFIG.set_curve(curve)


def default_curve() -> Type[Curve]:
    return _CONFIG.curve()


def default_params() -> UmbralParameters:
    return _CONFIG.params()