Source code for whispertrades.common

import traceback
from typing import Any, Callable, Union

from pydantic import BaseModel



[docs]
class BaseResponse(BaseModel):
    success: bool
    message: str
    data: Union[list[dict], dict] = []
    pages: list = None




[docs]
class APIError(Exception):
    def __init__(self, message: str):
        super().__init__(message)
        if message == 'Invalid token permissions':
            raise TokenPermissionError(message)
        elif message == 'Not authenticated':
            raise InvalidTokenError(message)




[docs]
class TokenPermissionError(Exception):
    pass




[docs]
class InvalidTokenError(Exception):
    pass




[docs]
class ReportRunningWarning(UserWarning):
    pass




[docs]
class ReportUninitializedWarning(UserWarning):
    pass




[docs]
class UpdatingDict(dict):
    def __init__(self, update_fn: Callable[[str], Any] = None, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self._update_fn = update_fn

    def __getitem__(self, key: str) -> Any:
        if key in self and self._update_fn:
            if not "IPython\\lib\\pretty.py" in traceback.extract_stack()[-2].filename:  # do not trigger update if called from IPython/Jupyter
                self[key] = self._update_fn(key)
        return super().__getitem__(key)