move OpenBBError class to error model

This commit is contained in:
Diogo Sousa
2023-08-18 11:45:50 +01:00
parent 0d5a7bd1e0
commit 9bb64998e2
7 changed files with 16 additions and 24 deletions

View File

@@ -8,7 +8,7 @@ from openbb_core.api.router.commands import router as router_commands
from openbb_core.api.router.coverage import router as router_coverage
from openbb_core.api.router.settings import router as router_settings
from openbb_core.api.router.system import router as router_system
from openbb_core.app.model.obbject import OpenBBError
from openbb_core.app.model.abstract.error import OpenBBError
from openbb_core.app.model.profile import Profile
from openbb_core.app.model.user_settings import UserSettings

View File

@@ -11,13 +11,14 @@ from pydantic import BaseConfig, Extra, create_model
from openbb_core.app.charting_manager import ChartingManager
from openbb_core.app.logs.logging_manager import LoggingManager
from openbb_core.app.model.abstract.error import OpenBBError
from openbb_core.app.model.abstract.warning import cast_warning
from openbb_core.app.model.charts.chart import Chart
from openbb_core.app.model.command_context import CommandContext
from openbb_core.app.model.journal import Journal
from openbb_core.app.model.journal_entry import JournalEntry
from openbb_core.app.model.journal_query import JournalQuery
from openbb_core.app.model.obbject import Error, OBBject, OpenBBError
from openbb_core.app.model.obbject import Error, OBBject
from openbb_core.app.model.system_settings import SystemSettings
from openbb_core.app.model.user_settings import UserSettings
from openbb_core.app.router import CommandMap

View File

@@ -1,4 +1,4 @@
from typing import Optional
from typing import Optional, Union
from pydantic import BaseModel
@@ -6,3 +6,11 @@ from pydantic import BaseModel
class Error(BaseModel):
message: str
error_kind: Optional[str] = None
class OpenBBError(Exception):
"""OpenBB Error."""
def __init__(self, original: Optional[Union[str, Exception]] = None):
self.original = original
super().__init__(str(original))

View File

@@ -5,7 +5,7 @@ from pydantic import Field
from pydantic.generics import GenericModel
from openbb_core.app.charting_manager import ChartingManager
from openbb_core.app.model.abstract.error import Error
from openbb_core.app.model.abstract.error import Error, OpenBBError
from openbb_core.app.model.abstract.tagged import Tagged
from openbb_core.app.model.abstract.warning import Warning_
from openbb_core.app.model.charts.chart import Chart
@@ -16,14 +16,6 @@ T = TypeVar("T")
PROVIDERS = get_provider_interface().providers_literal
class OpenBBError(Exception):
"""OpenBB Error."""
def __init__(self, original: Optional[Exception] = None):
self.original = original
super().__init__(str(original))
class OBBject(GenericModel, Generic[T], Tagged):
results: Optional[T] = Field(
default=None,

View File

@@ -5,6 +5,7 @@ from fastapi import HTTPException
from jose import JWTError
from jose.exceptions import ExpiredSignatureError
from jose.jwt import decode, get_unverified_header
from openbb_core.app.model.abstract.error import OpenBBError
from openbb_core.app.model.credentials import Credentials
from openbb_core.app.model.hub.features_keys import FeaturesKeys
from openbb_core.app.model.hub.hub_session import HubSession
@@ -14,10 +15,6 @@ from openbb_core.app.model.user_settings import UserSettings
from requests import get, post, put
class OpenBBError(Exception):
"""OpenBB error."""
class HubService:
"""Manage Hub service class."""

View File

@@ -4,6 +4,7 @@ from pathlib import Path
from typing import TYPE_CHECKING, Optional
from openbb_core.app.command_runner import CommandRunnerSession
from openbb_core.app.model.abstract.error import OpenBBError
from openbb_core.app.model.hub.hub_session import HubSession
from openbb_core.app.model.user_settings import UserSettings
from openbb_core.app.service.hub_service import HubService
@@ -13,10 +14,6 @@ if TYPE_CHECKING:
from openbb_core.app.static.container import Container
class OpenBBError(Exception):
pass
class Account:
def __init__(self, container: "Container"):
self._container = container

View File

@@ -6,15 +6,12 @@ from functools import wraps
import pandas as pd
from pydantic import ValidationError
from openbb_core.app.model.abstract.error import OpenBBError
from openbb_core.app.model.abstract.warning import OpenBBWarning
from openbb_core.app.model.obbject import OBBject
from openbb_core.app.utils import df_to_basemodel
class OpenBBError(Exception):
"""A custom exception for OpenBB errors."""
def filter_call(func):
"""Filter command call."""