OpenBB/website/content/platform/development/developer-guidelines/obbject_extensions.md

title, sidebar_position, description, keywords

title	sidebar_position	description	keywords
OBBject Extensions	8	This page provides information about how to write extensions for the OpenBB OBBject class.	OBBject Python Development OpenBB Platform

import HeadTitle from '@site/src/components/General/HeadTitle.tsx';

As mentioned, OpenBB provides some basic methods for interacting with common data structures that will be seen in the results attribute. If you are working with custom data, or adding new endpoints, it is possible that you will want to have your own methods for interacting with the data. The OpenBB Platform provides a way to do this by extending the OBBject class. The architecture for extensions was designed similar to how extensions and accessors are done in pandas, and relies on plugins through the poetry dependency management package.

This page will go through the steps of developing a simple, custom extension for the OBBject class.

Folder structure

example
├── README.md
├── openbb_example
│   └── __init__.py
├── poetry.lock
└── pyproject.toml

Writing the extension

In this example the extension code all lives inside __init__.py.

First we create an Extension class instance to tell openbb-core our extension name and any required credentials to be available at obb.user.credentials.

from openbb_core.app.model.extension import Extension

ext = Extension(name="example", credentials=["some_api_key"])

The credentials are required if the extension uses a key for data or a service, or it can alternatively be connection information for a database, or any other information that needs to be passed to the extension.

Now we define the extension functionality. The decorator @ext.obbject_accessor registers an accessor in each OBBject that is returned when a command is executed. This accessor is just a namespace that will contain the methods defined in the decorated class.

Here, we just add a method to say hi:

@ext.obbject_accessor
class Example:
    def __init__(self, obbject):
        self._obbject = obbject

    def hello(self):
        api_key = self._obbject._credentials.some_api_key.get_secret_value()
        print(f"Hello, this is my credential: {api_key}!")

Entry Point

The pyproject.toml file will specify the usual setup and dependency information. Additionally, we need to include the following block:

[tool.poetry.plugins."openbb_obbject_extension"]
example = "openbb_example:ext"

With this in the file, we can install the extension by running poetry install from the extension folder.

Using the extension

Now that the extension is installed and built, we can use it! Because we are extending the OBBject, this will be available on any function:

>>> from openbb import obb
>>> obb.user.credentials.some_api_key = "test"
>>> obbject = obb.equity.price.historical("AAPL")
>>> obbject.example.hello()
Hello, this is my credential: test!

In this example, we have added obbject.example as the extension and can use the .hello() functionality right from our OBBject.