OpenBB/openbb_terminal/stocks/options/barchart_model.py

"""Barchart Model"""
__docformat__ = "numpy"

import logging

import pandas as pd
import requests
from bs4 import BeautifulSoup

from openbb_terminal.decorators import log_start_end
from openbb_terminal.helper_funcs import get_user_agent

logger = logging.getLogger(__name__)


@log_start_end(log=logger)
def get_options_info(symbol: str) -> pd.DataFrame:
    """Scrape barchart for options info

    Parameters
    ----------
    symbol: str
        Stock ticker symbol

    Returns
    -------
    df: pd.DataFrame
        Dataframe of options information
    """
    page = f"https://www.barchart.com/stocks/quotes/{symbol}/overview"

    r = requests.get(page, headers={"User-Agent": get_user_agent()})
    soup = BeautifulSoup(r.text, "html.parser")
    tags = soup.find(
        "div",
        attrs={
            "class": "barchart-content-block symbol-fundamentals bc-cot-table-wrapper"
        },
    )
    data = tags.find_all("li")
    labels = []
    values = []
    for row in data:
        labels.append(row.find_all("span")[0].getText())
        values.append(row.find_all("span")[1].getText())

    df = pd.DataFrame(data=[labels, values]).T

    return df