mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 04:52:07 +00:00
009699928f
* Relink sdk/sdk -> platform/platform * Relink sdk -> platform * Rename folders * Fix dev_install script * Fix test generators * Fix test generators * Fix test generators * Organize imports
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
from inspect import getmembers, isfunction
|
|
from typing import List
|
|
|
|
from openbb_charting import charting_router
|
|
from openbb_core.app.router import RouterLoader
|
|
|
|
# refers to the root of the repo, not the root of the extension
|
|
# this script should be run from the root of the repo
|
|
CHARTING_ROUTER = (
|
|
"openbb_platform/extensions/charting/openbb_charting/charting_router.py"
|
|
)
|
|
|
|
|
|
def get_routes() -> List[str]:
|
|
router = RouterLoader.from_extensions()
|
|
return [route.path for route in router.api_router.routes] # type: ignore
|
|
|
|
|
|
def create_charting_functions_w_routes(routes_to_add: List[str]):
|
|
with open(CHARTING_ROUTER, "a") as f:
|
|
for route in routes_to_add:
|
|
original_route = route.replace("_", "/")
|
|
f.write(
|
|
f"""
|
|
|
|
def {route}(**kwargs) -> Tuple[OpenBBFigure, Dict[str, Any]]:
|
|
raise NotImplementedError("Command `{original_route}` does not have a charting function")
|
|
"""
|
|
)
|
|
|
|
|
|
def get_charting_functions() -> List[str]:
|
|
functions = getmembers(charting_router, isfunction)
|
|
if functions:
|
|
return [function[0] for function in functions]
|
|
return []
|
|
|
|
|
|
if __name__ == "__main__":
|
|
routes = get_routes()
|
|
# format routes to be valid function names
|
|
routes = [route.replace("/", "_")[1:] for route in routes]
|
|
# get functions already in charting_router.py
|
|
known_functions = get_charting_functions()
|
|
# remove known functions from routes
|
|
routes = [route for route in routes if route not in known_functions]
|
|
# create new functions in charting_router.py
|
|
create_charting_functions_w_routes(routes)
|