jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-03 16:02:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

change importlib_metadata import statement

Browse Source
This commit is contained in:
Diogo Sousa
2023-08-17 10:34:01 +01:00
parent 925373eb33
commit bd0ed33fd6
3 changed files with 8 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
openbb_sdk/sdk/core/openbb_core/app/charting_manager.py
View File

@@ -1,7 +1,7 @@
from importlib import import_module
from typing import Callable, Generic, Optional, Tuple, TypeVar
import importlib_metadata
from importlib_metadata import entry_points
from openbb_core.app.model.abstract.singleton import SingletonMeta
from openbb_core.app.model.charts.chart import Chart, ChartFormat
@@ -87,8 +87,7 @@ class ChartingManager(metaclass=SingletonMeta):
bool
Either charting extension is installed or not.
"""
entry_points = importlib_metadata.entry_points(group=plugin)
extensions = [ext.name for ext in entry_points]
extensions = [ext.name for ext in entry_points(group=plugin)]
return charting_extension in extensions
@@ -99,7 +98,7 @@ class ChartingManager(metaclass=SingletonMeta):
"""
Get the module of the given extension.
"""
for entry_point in importlib_metadata.entry_points(group=plugin):
for entry_point in entry_points(group=plugin):
if entry_point.name == extension_name:
return import_module(entry_point.module)

6
openbb_sdk/sdk/core/openbb_core/app/router.py
View File

@@ -16,8 +16,8 @@ from typing import (
overload,
)
import importlib_metadata
from fastapi import APIRouter, Depends
from importlib_metadata import entry_points
from pydantic import BaseModel
from pydantic.config import BaseConfig
from pydantic.validators import find_validators
@@ -441,9 +441,7 @@ class RouterLoader:
def from_extensions() -> Router:
router = Router()
for entry_point in importlib_metadata.entry_points(
group="openbb_core_extension"
):
for entry_point in entry_points(group="openbb_core_extension"):
try:
router.include_router(
router=entry_point.load(), prefix=f"/{entry_point.name}"

8
openbb_sdk/sdk/provider/openbb_provider/registry.py
View File

@@ -2,7 +2,7 @@
from typing import Dict
import importlib_metadata
from importlib_metadata import entry_points
from openbb_provider.abstract.provider import Provider
@@ -37,12 +37,10 @@ class RegistryLoader:
def from_extensions() -> Registry:
"""Load providers from entry points."""
registry = Registry()
for entry_point in importlib_metadata.entry_points(
group="openbb_provider_extension"
):
for entry_point in sorted(entry_points(group="openbb_provider_extension")):
try:
registry.include_provider(
provider=entry_point.load(),
provider=entry_point.load()
)
except Exception as e:
raise LoadingError(f"Invalid provider '{entry_point.name}': {e}") from e